Binary Classifier Optimization for Large Language Model Alignment

本文是LLM系列文章,针对《Binary Classifier Optimization for Large Language Model Alignment》的翻译。

大型语言模型对齐的二元分类器优化

摘要

通过偏好优化使大型语言模型(LLM)与人类偏好保持一致是至关重要的,但也是劳动密集型的,需要评估人员对每个提示进行选择和拒绝的文本完成情况的比较。最近,Kahneman Tversky Optimization(KTO)已经证明,LLM可以在每个提示完成对上仅使用二进制“竖起大拇指”或“向下大拇指”信号进行对齐。在本文中,我们提出了理论基础来解释通过这些二进制信号实现的成功对准。我们的分析揭示了一个新的视角:优化一个二元分类器,其logit是一个奖励,隐式地诱导最小化直接偏好优化(DPO)损失。在这一发现过程中,我们确定了两种有效对齐的技术:奖励转移和底层分布匹配。因此,我们提出了一种新的算法,二进制分类器优化,它集成了这些技术。我们在两种情况下验证了我们的方法:第一,在配对偏好数据集上,我们的方法与DPO和KTO的性能相当;第二,在模拟真实世界条件的二进制信号数据集上,竖起大拇指和竖起大拇指的数据之间存在不同的潜在分布。我们的模型始终如一地证明了两个基本LLM和三个不同的二进制信号数据集之间的有效和稳健的对齐,展示了我们从二进制反馈中学习的方法的优势。

1 引言

2 相关工作

3 前言

4 二进制分类器优化

5 实验

6 结论

本文为使用二进制信号(如“竖起大拇指”或“拇指向下”)成功对齐大型语言模型(LLM)提供了理论基础。我们的分析表明,训练一个

### Binary Classifier Loss 增大的含义 Binary classifier loss增大通常意味着模型预测的结果与实际标签之间的差异增加。对于生成对抗网络(GAN),判别器(Discriminator)是一个二元分类器,负责区分真实样本和由生成器(Generator)产生的伪造样本。 当提到loss增大时,在GAN上下文中: - 如果是指 **判别器的损失函数值增大**,这表明判别器难以分辨哪些图像是真实的,哪些是由生成器创建出来的虚假图像[^1]。然而,这种情况并不一定直接说明生成图像变得更逼真了。 - 若是 **生成器的损失函数值上升**,则可能暗示着生成器正在学习如何更好地欺骗判别器,即生成更加接近于真实数据分布的样本[^2]。但是需要注意的是,理想情况下,随着训练过程推进,两个网络都会逐渐改进自己,最终达到一种平衡状态——纳什均衡(Nash Equilibrium),此时两者的表现都趋于稳定,而不仅仅是单纯依靠某一方Loss的变化来衡量进步程度。 实际上,在理想的GAN收敛状态下,判别器应该无法有效地区分真假图片,从而使得其输出的概率趋近于0.5,这时无论是真实还是合成的数据都能获得相似评分。因此,单凭loss变化并不能完全反映生成质量的好坏;还需要借助其他指标如Inception Score (IS), Fréchet Inception Distance (FID)等来进行综合评价。 ```python import tensorflow as tf def discriminator_loss(real_output, fake_output): real_loss = cross_entropy(tf.ones_like(real_output), real_output) fake_loss = cross_entropy(tf.zeros_like(fake_output), fake_output) total_loss = real_loss + fake_loss return total_loss def generator_loss(fake_output): return cross_entropy(tf.ones_like(fake_output), fake_output) cross_entropy = tf.keras.losses.BinaryCrossentropy(from_logits=True) ```
评论
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

UnknownBody

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值