筛选难样本可显著提升大模型对齐效果,避免过难数据拖累训练。
Principled Data Selection for Alignment: The Hidden Risks of Difficult Examples
- 按样本难度筛选数据,避免模型能力不足时处理过难例子。
- 在四个模型、两个数据集上,剔除难例使胜率提升9-16%。
- 适合关注模型对齐效率与数据质量的开发者与研究者。
大语言模型对齐常假设更多干净数据带来更好结果,却忽略了模型能力与样本难度的匹配问题。本文提出新原则:偏好数据存在难度差异,过于困难的样本会超出模型能力,阻碍对齐。通过系统实验验证三个关键发现:(1) 偏好样本难度不一,学习顺序在不同对齐运行中保持一致;(2) 过于困难的样本显著降低四个 LLM 和两个数据集上的性能;(3) 模型能力决定其可处理难度阈值,凸显数据选择与模型容量间的关键关联。基于此,我们提出 Selective DPO,过滤掉过难样本。该简单改进在 AlpacaEval 2 基准上相较 DPO 基线提升 9-16% 胜率,优于多种 DPO 变体。结果揭示了对齐策略中数据难度与模型能力匹配的重要性,为提升大模型对齐提供全新视角。代码已开源。
原文摘要 · Abstract (English)
The alignment of large language models (LLMs) often assumes that using more clean data yields better outcomes, overlooking the match between model capacity and example difficulty. Challenging this, we propose a new principle: Preference data vary in difficulty, and overly difficult examples hinder alignment, by exceeding the model's capacity. Through systematic experimentation, we validate this principle with three key findings: (1) preference examples vary in difficulty, as evidenced by consistent learning orders across alignment runs; (2) overly difficult examples significantly degrade performance across four LLMs and two datasets; and (3) the capacity of a model dictates its threshold for handling difficult examples, underscoring a critical relationship between data selection and model capacity. Building on this principle, we introduce Selective DPO, which filters out overly difficult examples. This simple adjustment improves alignment performance by 9-16% in win rates on the AlpacaEval 2 benchmark compared to the DPO baseline, suppressing a series of DPO variants with different algorithmic adjustments. Together, these results illuminate the importance of aligning data difficulty with model capacity, offering a transformative perspective for improving alignment strategies in LLMs. Code is available at https://github.com/glorgao/SelectiveDPO.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。