arXiv:2606.03503cs.AI2026-06被引 3

通过细粒度偏好学习,让大模型自动剪掉冗余推理步骤,更高效地得出答案。

ThoughtFold: Folding Reasoning Chains via Introspective Preference Learning

论文配图:ThoughtFold: Folding Reasoning Chains via Introspective Preference Learning
图 1 · 摘自论文原文
  • 在正确推理路径中识别冗余步骤,生成多种简化候选路径
  • 用掩码偏好优化目标惩罚冗余探索,使推理链缩短56%
  • 适合追求高效推理的部署场景,尤其对资源受限设备友好

大型推理模型(LRMs)在基于可验证奖励的强化学习(RLVR)上取得显著进展,但长链思维(CoT)天然包含试错过程,主流方法仅记忆结果正确的路径,导致冗余探索被强化,引发过度思考问题。现有方法虽倾向短路径,但学习信号仍依赖结果,无法减少冗余记忆。为此,我们提出ThoughtFold,通过内省策略识别正确路径中的冗余,并生成候选子路径谱。基于此,设计掩码偏好优化目标,显式惩罚冗余探索,鼓励模型直接连接关键推理片段,有效折叠推理链。大量实验表明,ThoughtFold将DeepSeek-R1-Distill-Qwen-7B的令牌使用量降低约56%,同时保持顶尖准确率。

原文摘要 · Abstract (English)

Large Reasoning Models (LRMs) have achieved remarkable progress thanks to Reinforcement Learning with Verifiable Rewards (RLVR) on Chain-of-Thoughts (CoTs). However, since long CoTs naturally contain trial and errors and mainstream RLVR approaches choose outcome-correct CoT trajectories for memorization, the redundant explorations in long CoTs are inevitably reinforced, which results in the over-thinking issues of LRMs. Previous attempts to resolve this issue mainly give more advantage to shorter trajectories, yet their learning signals are still outcome-based and cannot reduce the memorization of redundant explorations in long CoTs. Therefore, we propose ThoughtFold, a framework that leverages fine-grained preference learning to mitigate redundant explorations for efficient reasoning. ThoughtFold employs an introspective strategy to identify redundancy within each correct trajectory, which yields a spectrum of candidate sub-trajectories. Leveraging this spectrum, we introduce a masked preference optimization objective that explicitly penalizes redundant explorations and encourages the model to directly bridge essential reasoning segments, effectively folding its reasoning chains into a more concise path. Extensive experiments show that ThoughtFold significantly enhances efficiency. It reduces the token usage of DeepSeek-R1-Distill-Qwen-7B by approximately 56% while maintaining state-of-the-art accuracy.

推理优化强化学习模型压缩高效推理

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