arXiv:2606.18844cs.LG2026-06被引 1

让大模型通过自身错误生成修正轨迹,实现精准纠错。

Learning from Your Own Mistakes: Constructing Learnable Micro-Reflective Trajectories for Self-Distillation

论文配图:Learning from Your Own Mistakes: Constructing Learnable Micro-Reflective Trajectories for Self-Distillation
图 1 · 摘自论文原文
  • 基于模型自身错误与正确推理的对比,构建可学习的纠错轨迹。
  • 在AIME/HMMT测试中,相比GRPO方法提升推理准确率,且纠错能力更强。
  • 适合需要提升自主纠错能力的大模型研究者和开发者。

自蒸馏通过使用模型自身生成的推理路径作为训练信号来提升大语言模型的推理能力,通常采用隐式对齐方式最小化与理想目标分布之间的KL散度。然而,由于这些监督信号来自不受控采样,无法提供模型具体错误的诊断信息或针对性修正指导,导致模型仅模仿理想分布,而非获得细粒度的失败原因纠正。为此,本文提出轨迹增强策略优化(TAPO),将自蒸馏从隐式的分布对齐推进为显式的轨迹构建。在强化学习训练中,模型对同一问题生成正确与错误的推理路径,TAPO利用这种对比结构构造微反思性修正轨迹:保留模型错误推理至失败点,随后插入自然语言诊断及由同组正确参考引导的修正推理。每条轨迹锚定于模型自身的前缀与解法,比基于位置的KL对齐更贴近其策略分布。为整合这些轨迹,TAPO引入能力边界处的难度感知候选选择,以及解耦优势估计,防止梯度污染。在AIME 2024、AIME 2025和HMMT 2025上的实验表明,相同训练步数下,TAPO优于GRPO。进一步分析显示,其显著增强了首次推理与错误修正能力。

原文摘要 · Abstract (English)

Self-distillation improves reasoning in large language models by using the model's own rollouts as training signal, typically through implicit logit-level alignment that minimizes KL divergence toward a privileged target distribution. However, because this supervision is generated via uncontrolled sampling, it provides no diagnostic insight into the model's specific errors or corrective guidance for its individual failure patterns. Consequently, the model learns to imitate a privileged distribution rather than receiving fine-grained corrections that pinpoint where and why its reasoning fails. In this paper, we propose Trajectory-Augmented Policy Optimization (TAPO), which advances self-distillation from implicit distributional alignment to explicit trajectory construction. During RL training, the model produces both correct and incorrect rollouts to the same query, and TAPO leverages this contrastive structure to construct micro-reflective corrections, new training trajectories that retain the model's erroneous reasoning up to the point of failure, then insert a natural-language diagnosis and corrected reasoning guided by a correct reference from the same sampling group. Since each trajectory is anchored in the learner's own prefix and solutions, the corrective signal preserves the model's on-policy distribution to a greater extent than the position-wise alignment imposed by KL-based methods. To integrate these trajectories, TAPO introduces difficulty-aware candidate selection at the model's capability boundary and decoupled advantage estimation to prevent gradient contamination. Experiments on AIME 2024, AIME 2025, and HMMT 2025 show that TAPO achieves consistent improvements over GRPO under the same number of training steps. Further analysis demonstrates that TAPO strengthens both first-pass reasoning and error-correction effectiveness.

自蒸馏大模型推理增强强化学习

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