别再把KL散度加进loss了!Bengio团队实证:回归Reward才是无偏正解

数据派THU 2026-01-07 17:00
文章摘要
本文背景是强化学习与人类反馈(RLHF/RLVR)已成为大模型后训练的主流方法,其核心目标是在最大化奖励的同时,利用KL散度约束策略模型不偏离参考模型。研究目的是揭示当前主流实现(将KL散度惩罚项K3估算器直接置于损失函数中)在数学上存在梯度偏差,并探讨其对模型训练稳定性和泛化能力的影响。结论表明,这种有偏估计会损害模型在域外任务上的性能;而将KL散度作为惩罚项移回奖励函数并使用朴素的K1估算器,能实现无偏梯度估计,在实验中显著提升了模型泛化能力,在域外任务上带来近20%的性能提升,且在不同架构下更具鲁棒性。
别再把KL散度加进loss了!Bengio团队实证:回归Reward才是无偏正解
本站注明稿件来源为其他媒体的文/图等稿件均为转载稿,本站转载出于非商业性的教育和科研之目的,并不意味着赞同其观点或证实其内容的真实性。如转载稿涉及版权等问题,请作者速来电或来函联系。
数据派THU
最新文章
热门类别
相关文章
联系我们:info@booksci.cn Book学术提供免费学术资源搜索服务,方便国内外学者检索中英文文献。致力于提供最便捷和优质的服务体验。 Copyright © 2023 布克学术 All rights reserved.
京ICP备2023020795号-1
ghs 京公网安备 11010802042870号
Book学术文献互助
Book学术文献互助群
群 号:604180095
Book学术官方微信