跳到正文
arXiv cs.AI· Yashuo Luo, Qianren Mao, Ziqi Qin, Junnan Liu, Haitao Li, Qili Zhang, Yuan Gao, Zhijun Chen, Xu Wang·· 11 小时前AI 评分31

CARD:将求解轨迹转化为教学轨迹的认知对齐推理蒸馏框架

From Solver Trajectories to Teaching Trajectories: Cognition-Aligned Reasoning Distillation

AI 导读

研究者提出基于 RL 的认知对齐推理蒸馏框架 CARD,将大推理模型的求解轨迹直接转化为与学生认知对齐的教学轨迹,无需昂贵重采样。CARD 用分布对齐奖励和逻辑连贯性奖励两个信号优化,在多个学生模型和设置上优于原始教师轨迹蒸馏,域外性能提升显著。代码已开源。

正文 · AI 翻译

查看 PDF HTML(实验性)

摘要:从大型推理模型(LRM)中进行知识蒸馏,通常使用原始的教师轨迹作为监督。然而,这种做法会造成蒸馏目标与教师轨迹性质之间的不匹配。先进的 LRM 通常通过奖励驱动的后训练(如 RLHF)来提升其推理能力。这种训练将它们优化为强大的解题者,而不是能够产出小模型易于学习的推理过程的教师。由于小模型与更大的教师模型相比具有不同的“认知容量”,直接模仿原始教师轨迹有时效果不佳,且往往需要大量高质量样本。为弥合这一不匹配,我们提出认知对齐推理蒸馏(CARD),这是一个基于 RL 的框架,可将解题者轨迹转化为与学生认知对齐的教学轨迹。为避免昂贵的重采样,我们的框架直接转换现有轨迹。我们用两种奖励信号优化 CARD:(1)分布对齐奖励,用于匹配学生的预测分布;(2)逻辑连贯性奖励,用于减少误导性的失败推理步骤。在多个学生模型和设置上的全面评估表明,CARD 在域内设置中优于从原始教师轨迹进行蒸馏,同时在分布外性能上取得显著提升。代码可在 此 https URL 获取。
主题: 人工智能(cs.AI)
引用为: arXiv:2610.36445 [cs.AI]
  (或 arXiv:2610.36445v1 [cs.AI] 用于此版本)
  https://doi.org/10.48550/arXiv.2610.36445

由 DataCite 颁发的 arXiv DOI(待注册)

提交历史

来自:Yashuo Luo [查看电子邮件]
[v1] 2026年10月27日星期二 00:46:23 UTC(1,613 KB)

来源:arXiv cs.AI · arxiv.org