arXiv:2412.15282cs.CLcs.AI2024-12被引 8

通过合成数据系统研究指令遵循中偏好学习的优化方法

A Systematic Examination of Preference Learning through the Lens of Instruction-Following

  • 构建48,000条带23个约束的合成指令数据,实现精细化评估
  • 高对比度偏好对提升模型性能更有效,适度难度提示促进泛化
  • MCTS生成的共享前缀对齐效果更稳定,适合复杂训练场景

偏好学习是广泛采用的大语言模型后训练技术,用于对齐人类偏好并提升下游任务能力。本文系统研究了偏好数据集属性对指令遵循任务中模型对齐与性能的影响。我们设计了一种新型合成数据生成管道,生成48,000条独特的指令遵循提示,涵盖23个可验证约束,支持对模型响应的细粒度、自动化质量评估。利用该合成提示,采用拒绝采样(RS)和蒙特卡洛树搜索(MCTS)两种数据筛选方法获取(优选,次选)响应对。实验考察了三方面影响:(1)优选与次选响应间的共享前缀存在性;(2)优选与次选响应的对比度与质量;(3)训练提示的复杂度。结果表明,由MCTS生成的共享前缀虽仅带来微弱但持续的性能提升,并在复杂训练配置下更具稳定性。高对比度偏好对通常优于低对比度对,但二者结合往往能兼顾多样性与学习效率,取得最佳表现。此外,使用中等难度提示训练,相比过于复杂的提示,在更复杂评估场景下仍能实现更好泛化能力。研究为指令遵循任务中的偏好数据优化提供了可操作的洞察,构建了一个可扩展且高效的模型训练与对齐框架。

原文摘要 · Abstract (English)

Preference learning is a widely adopted post-training technique that aligns large language models (LLMs) to human preferences and improves specific downstream task capabilities. In this work we systematically investigate how specific attributes of preference datasets affect the alignment and downstream performance of LLMs in instruction-following tasks. We use a novel synthetic data generation pipeline to generate 48,000 unique instruction-following prompts with combinations of 23 verifiable constraints that enable fine-grained and automated quality assessments of model responses. With our synthetic prompts, we use two preference dataset curation methods - rejection sampling (RS) and Monte Carlo Tree Search (MCTS) - to obtain pairs of (chosen, rejected) responses. Then, we perform experiments investigating the effects of (1) the presence of shared prefixes between the chosen and rejected responses, (2) the contrast and quality of the chosen, rejected responses and (3) the complexity of the training prompts. Our experiments reveal that shared prefixes in preference pairs, as generated by MCTS, provide marginal but consistent improvements and greater stability across challenging training configurations. High-contrast preference pairs generally outperform low-contrast pairs; however, combining both often yields the best performance by balancing diversity and learning efficiency. Additionally, training on prompts of moderate difficulty leads to better generalization across tasks, even for more complex evaluation scenarios, compared to overly challenging prompts. Our findings provide actionable insights into optimizing preference data curation for instruction-following tasks, offering a scalable and effective framework for enhancing LLM training and alignment.

偏好学习指令遵循数据生成大模型对齐

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