arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2602.02405cs.LGcs.AI

通过自蒸馏使专家推理可学习

Making Expert Reasoning Learnable with Self-Distillation

Georgia Institute of Technology, Atlanta, Georgia(佐治亚理工学院,亚特兰大,佐治亚州)

Ethan Mendes, Jungsoo Park, Alan Ritter

AI总结:

提出分布对齐模仿学习(DAIL),通过两步自蒸馏方法弥合专家解决方案与模型分布之间的差距,利用少量高质量专家数据显著提升大语言模型的推理能力。

摘要:

提升大语言模型(LLM)的推理能力通常依赖于模型采样正确解以进行强化,或存在更强模型来解决问题。然而,许多难题即使对当前前沿模型也难以处理,阻碍了有效训练信号的提取。一个有前景的替代方案是利用高质量的人类专家解决方案,但直接模仿这些数据从根本上存在分布外问题:专家解决方案通常具有教学性质,包含为人类读者而非计算模型设计的隐含推理间隙。此外,高质量专家解决方案成本高昂,需要可泛化且样本高效的训练方法。我们提出分布对齐模仿学习(DAIL),一种两步自蒸馏方法,通过首先将专家解决方案转化为详细的、分布内的推理轨迹,然后应用对比目标使学习聚焦于专家见解和方法,从而弥合分布差距。我们发现,DAIL可以利用少于1000个高质量专家解决方案,在Qwen2.5-Instruct和Qwen3上实现高达31%的pass@128增益,推理效率翻倍,并实现域外泛化。

英文摘要:

Improving the reasoning capabilities of large language models (LLMs) typically relies either on the model's ability to sample a correct solution to be reinforced or the existence of a stronger model able to solve the problem. However, many difficult problems remain intractable for even current frontier models, preventing the extraction of valid training signals. A promising alternative is to leverage high-quality expert human solutions, yet naive imitation of this data fails because it is fundamentally out-of-distribution: expert solutions are typically didactic, containing implicit reasoning gaps intended for human readers rather than computational models. Furthermore, high-quality expert solutions are expensive, necessitating generalizable, sample-efficient training methods. We propose Distribution Aligned Imitation Learning (DAIL), a two-step self-distillation method that bridges the distributional gap by first transforming expert solutions into detailed, in-distribution reasoning traces and then applying a contrastive objective to focus learning on expert insights and methodologies. We find that DAIL can leverage fewer than 1000 high-quality expert solutions to achieve up to 31% pass@128 gains on Qwen2.5-Instruct and Qwen3, double reasoning efficiency, and enable out-of-domain generalization.

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

↑