arXiv:2605.19433cs.CLcs.AI2026-05被引 2

提出新方法缓解大模型推理蒸馏中的双重偏差问题。

Backtracking When It Strays: Mitigating Dual Exposure Biases in LLM Reasoning Distillation

  • 动态监测学生生成路径,偏离安全边界时回溯并由教师纠正
  • 在两个数据集上实现约3%的推理性能提升
  • 适合关注高效推理蒸馏与错误控制的研究者

大语言模型(LLM)通过长链思维(CoT)在复杂推理任务中表现卓越,但计算开销巨大,制约实际部署。推理蒸馏通过将教师模型的能力迁移至轻量学生模型来缓解此问题。然而,现有蒸馏范式面临根本困境:离策略蒸馏严格使用教师生成的最优轨迹,因训练分布与学生推理上下文不匹配,导致长链条推理中错误传播;而同策略蒸馏虽允许学生自主探索,却引入反向暴露偏差——当教师模型基于学生生成的次优上下文进行指导时,也难以提供有效反馈。为解决这一双重暴露偏差问题,我们提出监控轨迹并偏离时回溯(MOTAB)的新蒸馏框架。MOTAB动态监测学生同策略生成路径与自适应安全边界的关系,一旦超出阈值则回溯至最近安全状态,并由教师介入纠正方向。该方法既容忍轻微误差以缓解暴露偏差,又防止次优上下文以规避反向暴露偏差。在LIMO-v2和AceReason数据集上的大量实验表明,MOTAB有效缓解双重暴露偏差,使推理任务平均性能提升约3%。

原文摘要 · Abstract (English)

Large language models (LLMs) have achieved remarkable success in complex reasoning tasks via long chain-of-thought (CoT), yet their immense computational overhead hinders real-world deployment. LLM reasoning distillation addresses this by transferring reasoning capabilities from formidable teacher models to compact student models. However, existing distillation paradigms face a fundamental dilemma. Typical off-policy distillation strictly utilizes teacher-generated golden trajectories, suffering from an exposure bias due to the mismatch between training distributions and student-generated inference contexts, which leads to error cascades in long CoT reasoning. To address this, on-policy distillation allows students to explore their own trajectories, but we demonstrate that it inherently introduces a reciprocal reversed exposure bias: the teacher model also struggles to provide positive guidance when conditioned on student-generated sub-optimal contexts. To resolve this dual exposure biases problem, we propose Monitoring Trajectories and Backtracking when it strays (MOTAB), a new LLM reasoning distillation pipeline. Specifically, MOTAB dynamically monitors the student's on-policy generation against an adaptive safety boundary. When the generation strays and exceeds this threshold, MOTAB backtracks to the last safe state and leverages teacher intervention to correct the course. This approach inherently tolerates minor student errors to mitigate exposure bias, while preventing sub-optimal contexts to circumvent reversed exposure bias. Extensive experiments on the LIMO-v2 and AceReason datasets demonstrate that MOTAB effectively alleviates the dual exposure biases, yielding a roughly 3% average performance improvement in reasoning tasks.

推理蒸馏链式思维偏差缓解

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