用不确定性估计提升大模型自进化中的反馈质量,减少噪声干扰。
Self-Evolutionary Large Language Models through Uncertainty-Enhanced Preference Optimization
- 引入贝叶斯神经网络的蒙特卡洛丢弃法,对偏好对进行不确定性评估。
- 在多个基准上显著降低噪声数据影响,提升迭代优化性能。
- 适合关注大模型自我进化与奖励模型偏差问题的研究者。
迭代偏好优化已成为大语言模型(LLM)训练的主流范式,但因循环中产生大量噪声偏好数据,性能仍不理想。为此,我们提出一种不确定性增强的偏好优化(UPO)框架,使LLM能基于可靠反馈实现自进化。核心思路是通过成对不确定性估计,缓解当前策略和奖励模型产生的噪声偏好数据。为此,我们引入一个估计器模型,采用贝叶斯神经网络(BNN)中的蒙特卡洛(MC)丢弃法,对来自LLM策略的偏好对进行不确定性评估。相比直接根据奖励分数过滤生成结果的方法,该估计器从成对角度关注模型不确定性,有效规避了奖励模型的确认偏差问题。此外,我们还设计了一种不确定性增强的自进化算法,提升偏好优化的鲁棒性,并鼓励模型生成高奖励且高确定性的响应。在多个基准上的大量实验表明,该框架显著缓解了噪声问题,提升了迭代偏好优化的效果。
原文摘要 · Abstract (English)
Iterative preference optimization has recently become one of the de-facto training paradigms for large language models (LLMs), but the performance is still underwhelming due to too much noisy preference data yielded in the loop. To combat this issue, we present an \textbf{U}ncertainty-enhanced \textbf{P}reference \textbf{O}ptimization (UPO) framework to make the LLM self-evolve with reliable feedback. The key idea is mitigating the noisy preference data derived from the current policy and reward models by performing pair-wise uncertainty estimation and judiciously reliable feedback sampling. To reach this goal, we thus introduce an estimator model, which incorporates Monte Carlo (MC) dropout in Bayesian neural network (BNN) to perform uncertainty estimation for the preference data derived from the LLM policy. Compared to the existing methods that directly filter generated responses based on the reward score, the estimator focuses on the model uncertainty in a pair-wise manner and effectively bypasses the confirmation bias problem of the reward model. Additionally, we also propose an uncertainty-enhanced self-evolution algorithm to improve the robustness of preference optimization and encourage the LLM to generate responses with both high reward and certainty. Extensive experiments over multiple benchmarks demonstrate that our framework substantially alleviates the noisy problem and improves the performance of iterative preference optimization.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。