arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2502.08922cs.AI
收藏

内部奖励模型的自一致性提升自奖励语言模型

Self-Consistency of the Internal Reward Models Improves Self-Rewarding Language Models

Fudan University(复旦大学)

Xin Zhou, Yiwen Guo, Ruotian Ma, Tao Gui, Qi Zhang, Xuanjing Huang

AI总结:

针对LLM内部多个奖励模型偏好预测不一致的问题,提出SCIR框架,通过不一致性惩罚和一致数据筛选提升自奖励训练的可靠性与对齐性能。

摘要:

使大型语言模型(LLM)与人类偏好保持一致,对于其在现实应用中的部署至关重要。自奖励语言模型的最新进展表明,LLM可以利用其内部奖励模型(例如LLM-as-a-Judge)生成偏好数据,从而在无需昂贵人工标注的情况下提升对齐性能。然而,我们发现同一LLM内部的不同奖励模型往往会产生不一致的偏好。这种不一致引发了对自生成偏好数据可靠性的担忧,阻碍了整体对齐性能,并凸显出需要进一步研究,以确保与人类偏好进行可靠且连贯的对齐。为解决这一局限,我们提出Self-Consistent Internal Rewards(SCIR),这是一种旨在训练过程中增强内部奖励模型之间一致性的新型框架。在每个训练步骤中,我们收集来自多个预定义内部奖励模型的偏好预测,并通过不一致性惩罚机制强制实现一致性和置信度,从而提高这些内部奖励模型的可靠性。我们选择性地使用预测一致的数据进行偏好优化,以确保偏好数据的质量。通过采用自一致内部奖励,我们的方法显著提升了LLM的对齐性能和奖励建模能力,并以明显优势优于基线方法。

英文摘要:

Aligning Large Language Models (LLMs) with human preferences is crucial for their deployment in real-world applications. Recent advancements in Self-Rewarding Language Models suggest that an LLM can use its internal reward models (such as LLM-as-a-Judge) \cite{yuanself} to generate preference data, improving alignment performance without costly human annotation. However, we find that different internal reward models within the same LLM often generate inconsistent preferences. This inconsistency raises concerns about the reliability of self-generated preference data, hinders overall alignment performance, and highlights the need for further research to ensure reliable and coherent alignment with human preferences. To address this limitation, we propose Self-Consistent Internal Rewards (SCIR), a novel framework designed to enhance consistency among internal reward models during training. In each training step, we collect preference predictions from multiple pre-defined internal reward models and enforce consistency and confidence through an inconsistency penalty mechanism, thereby improving the reliability of these internal reward models. We selectively use data with consistent predictions for preference optimization, ensuring the quality of the preference data. By employing self-consistent internal rewards, our method significantly improves the alignment performance and reward modeling capability of LLMs, outperforming baseline methods by a notable margin.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑