arXiv:2505.17122cs.CL2025-05被引 4

只用前半段回复就能更好对齐大模型,关键在早期文本信号。

Shallow Preference Signals: Large Language Model Aligns Even Better with Truncated Data?

  • 发现人类偏好信号集中在回复开头,称为浅层偏好信号。
  • 仅用前40%内容训练,模型性能反而优于全量数据。
  • 提出两种新解码策略,提升对齐效率与效果。

大语言模型对齐人类偏好仍是关键挑战。现有基于偏好的优化方法(如RLHF、DPO)依赖人工标注数据集。本文发现:偏好信号常集中于响应的早期词元。我们系统性地在不同位置截断偏好数据集,并用截断数据训练奖励模型和DPO模型。令人意外的是,仅保留前一半或更少词元的模型性能与全量数据相当甚至更优。例如,在Skywork-Reward-Preference-80K-v0.2数据集上,40%截断数据训练的奖励模型表现超越全量训练。该现象在多个数据集上一致,表明浅层偏好信号普遍存在。我们进一步通过解码策略分析奖励信号分布,提出两种受此观察启发的简单策略:长度控制解码与KL阈值控制解码,能有效平衡对齐效果与计算效率。结果再次验证假设。浅层偏好信号揭示了当前对齐方法的潜在问题:过度关注响应初始部分,忽视整体内容,可能导致与真实人类偏好脱节,影响对齐质量。

原文摘要 · Abstract (English)

Aligning large language models (LLMs) with human preferences remains a key challenge in AI. Preference-based optimization methods, such as Reinforcement Learning with Human Feedback (RLHF) and Direct Preference Optimization (DPO), rely on human-annotated datasets to improve alignment. In this work, we identify a crucial property of the existing learning method: the distinguishing signal obtained in preferred responses is often concentrated in the early tokens. We refer to this as shallow preference signals. To explore this property, we systematically truncate preference datasets at various points and train both reward models and DPO models on the truncated data. Surprisingly, models trained on truncated datasets, retaining only the first half or fewer tokens, achieve comparable or even superior performance to those trained on full datasets. For example, a reward model trained on the Skywork-Reward-Preference-80K-v0.2 dataset outperforms the full dataset when trained on a 40\% truncated dataset. This pattern is consistent across multiple datasets, suggesting the widespread presence of shallow preference signals. We further investigate the distribution of the reward signal through decoding strategies. We consider two simple decoding strategies motivated by the shallow reward signal observation, namely Length Control Decoding and KL Threshold Control Decoding, which leverage shallow preference signals to optimize the trade-off between alignment and computational efficiency. The performance is even better, which again validates our hypothesis. The phenomenon of shallow preference signals highlights potential issues in LLM alignment: existing alignment methods often focus on aligning only the initial tokens of responses, rather than considering the full response. This could lead to discrepancies with real-world human preferences, resulting in suboptimal alignment performance.

大模型对齐偏好学习解码策略

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