arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2604.20140cs.AIcs.LG
收藏

HiPO:用于大语言模型自适应推理的分层偏好优化

HiPO: Hierarchical Preference Optimization for Adaptive Reasoning in LLMs

Vellore Institute of Technology(维洛雷理工学院);University of Massachusetts Amherst(马萨诸塞大学阿姆赫斯特分校);Northwestern University(西北大学);Yale University(耶鲁大学);Algoverse AI Research(Algoverse AI研究)

Darsh Kachroo, Arjun Prasaath Anbazhagan, Adriana Caraeni, Brennan Lagasse, Kevin Zhu

AI总结:

HiPO通过分层优化提升大语言模型在复杂推理任务中的表现,结合偏好优化与结构化推理的优势,实现更高效的训练和更一致的输出。

摘要:

直接偏好优化(DPO)是对齐大语言模型与人类偏好有效的框架,但难以处理复杂推理任务。DPO优化生成首选响应的整体可能性,缺乏对多步骤解决方案子部分提供反馈的粒度。现有方法在稳定偏好学习(如DPO变种KTO和RSO)或结构化推理(如ReMA的多智能体RL框架、思维树)方面表现优异,但未能融合这些互补优势。我们提出HiPO(分层偏好优化),作为DPO的扩展,将响应分为推理段(查询澄清和上下文、推理步骤、答案),并计算损失作为各段DPO损失的加权和。我们的方法实现了段特定训练,同时保持DPO的计算效率和训练稳定性。我们证明,对于使用HiPO和DPO在Math Stack Exchange偏好数据集上微调的多个7B LLMs,训练使用HiPO的模型在多种常见数学基准测试中表现更优,并在GPT-4.1评估中显示出更组织、逻辑流畅和一致的输出。

英文摘要:

Direct Preference Optimization (DPO) is an effective framework for aligning large language models with human preferences, but it struggles with complex reasoning tasks. DPO optimizes for the likelihood of generating preferred over dispreferred responses in their entirety and lacks the granularity to provide feedback on subsections of many-step solutions typical of reasoning tasks. Existing methods excel at either stable preference learning (e.g., DPO variants like KTO and RSO) or structured reasoning (e.g., ReMA's multi-agent RL framework, Tree of Thoughts), but fail to merge these complementary strengths. We propose HiPO (Hierarchical Preference Optimization), an extension of DPO that separates responses into reasoning segments (query clarification and context, reasoning steps, and answer) and computes loss as a weighted sum of the DPO loss for each segment. Our approach enables segment-specific training while maintaining DPO's computational efficiency and training stability. We demonstrate that for multiple 7B LLMs fine-tuned using HiPO and DPO on the Math Stack Exchange preference dataset, the models trained with HiPO outperform the others on a variety of common math benchmarks and achieve greater organization, logical flow, and consistency as measured by GPT-4.1.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑