arXiv:2503.22137cs.AIcs.LG2025-03被引 4

用夏普比率选最优标注数据,降低大模型对齐成本

Sharpe Ratio-Guided Active Learning for Preference Optimization in RLHF

  • 基于夏普比率评估每条数据的风险收益,动态筛选高价值标注样本
  • 在有限标注数据下,相比基线提升5%胜率,效果稳定跨模型和数据集
  • 适合资源受限场景,尤其适用于需要高质量偏好数据的对齐任务

从人类反馈中进行强化学习(RLHF)已成为大语言模型训练与对齐的核心流程。近期的直接偏好优化(DPO)简化了偏好学习步骤,但偏好数据的收集仍面临挑战且成本高昂,通常依赖专家标注。可通过精心选择待标注数据点来缓解此成本问题。本文提出一种基于夏普比率的风险评估策略,用于主动学习中高效筛选提示与偏好对。为应对标注前偏好未知的问题,方法通过评估所有潜在标注的梯度影响来衡量其对模型更新的作用,从而实现不依赖标注结果的数据点风险评估。结合DPO损失的导数推导出每条数据对的闭式夏普比率表达式,确保方法可计算且高效。引入两种变体,分别基于不同先验信息假设。实验表明,在多个语言模型和真实数据集上,使用有限人类偏好数据时,该方法相较基线最高提升5%的胜率。

原文摘要 · Abstract (English)

Reinforcement learning from human feedback (RLHF) has become a cornerstone of the training and alignment pipeline for large language models (LLMs). Recent advances, such as direct preference optimization (DPO), have simplified the preference learning step. However, collecting preference data remains a challenging and costly process, often requiring expert annotation. This cost can be mitigated by carefully selecting the data points presented for annotation. In this work, we propose an active learning approach to efficiently select prompt and preference pairs using a risk assessment strategy based on the Sharpe Ratio. To address the challenge of unknown preferences prior to annotation, our method evaluates the gradients of all potential preference annotations to assess their impact on model updates. These gradient-based evaluations enable risk assessment of data points regardless of the annotation outcome. By leveraging the DPO loss derivations, we derive a closed-form expression for computing these Sharpe ratios on a per-tuple basis, ensuring our approach remains both tractable and computationally efficient. We also introduce two variants of our method, each making different assumptions about prior information. Experimental results demonstrate that our method outperforms the baseline by up to 5% in win rates against the chosen completion with limited human preference data across several language models and real-world datasets.

RLHF主动学习偏好优化夏普比率

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。