通过自反馈机制提升大模型推理准确性和简洁性。
In-Token Rationality Optimization: Towards Accurate and Concise LLM Reasoning via Self-Feedback
- 在单次前向传播中实现逐词探索与自生成反馈。
- 数学推理任务准确率最高提升20%,且推理过程更简洁。
- 适用于跨领域推理,具备强泛化能力,适合复杂推理场景。
训练大语言模型进行思维链推理面临重大挑战:对单一‘最优’推理路径的监督微调会损害泛化能力,因其惩罚了同样有效的替代路径;而基于可验证奖励的强化学习则存在信用分配难题和高昂计算成本。为此,我们提出InTRO(In-Token Rationality Optimization)框架,支持细粒度的逐词探索与自反馈机制,实现精准且简洁的推理。InTRO不直接优化难以处理的全部有效推理路径目标,而是利用生成策略与其答案条件版本之间的信息差异,估算每个词的修正因子(即重要性权重),以指导更具信息量的下一词选择。该方法使模型能在单次前向传播中完成探索与自我反馈,从而引导出更准确、更简洁的推理链条。在六个数学推理基准上,InTRO持续优于其他基线,解决方案准确率相对基础模型最高提升20%。其思维链也明显更紧凑,减少冗余。此外,InTRO具备跨领域迁移能力,成功适应超出数学范畴的推理任务,展现出强大泛化性能。
原文摘要 · Abstract (English)
Training Large Language Models (LLMs) for chain-of-thought reasoning presents a significant challenge: supervised fine-tuning on a single "golden" rationale hurts generalization as it penalizes equally valid alternatives, whereas reinforcement learning with verifiable rewards struggles with credit assignment and prohibitive computational cost. To tackle these limitations, we introduce InTRO (In-Token Rationality Optimization), a new framework that enables both token-level exploration and self-feedback for accurate and concise reasoning. Instead of directly optimizing an intractable objective over all valid reasoning paths, InTRO leverages correction factors-token-wise importance weights estimated by the information discrepancy between the generative policy and its answer-conditioned counterpart, for informative next token selection. This approach allows the model to perform token-level exploration and receive self-generated feedback within a single forward pass, ultimately encouraging accurate and concise rationales. Across six math-reasoning benchmarks, InTRO consistently outperforms other baselines, raising solution accuracy by up to 20% relative to the base model. Its chains of thought are also notably more concise, exhibiting reduced verbosity. Beyond this, InTRO enables cross-domain transfer, successfully adapting to out-of-domain reasoning tasks that extend beyond the realm of mathematics, demonstrating robust generalization.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。