让大模型学会从多步反馈中提炼推理能力,不再依赖外部反馈循环。
MoL-RL: Distilling Multi-Step Environmental Feedback into LLMs for Feedback-Independent Reasoning
- 用双目标优化框架融合多步反馈信号,保留上下文信息。
- 在MATH-500和CodeAgent-Test上超越现有方法,Qwen3-8B表现最优。
- 适合需要强推理能力的数学与代码生成场景。
大语言模型在利用序列环境反馈(如自然语言评价)进行无反馈链式思考(CoT)推理时面临挑战。现有方法或将反馈转为标量奖励,丢失上下文信息;或使用精炼数据集,无法利用反馈的多步与离散特性。为此,我们提出MoL-RL,一种通过双目标优化框架将多步环境反馈融入大模型的新训练范式。该方法结合基于损失混合(MoL)的持续训练(解耦领域特定反馈信号,以交叉熵优化;通用语言能力,以KL散度保持),以及基于GRPO的后训练,将序列反馈交互压缩为单步推理。实验表明,在数学推理(MATH-500、AIME24/AIME25)与代码生成(CodeAgent-Test)基准上,采用Qwen3-8B模型的MoL-RL达到最先进性能,并在不同模型规模(Qwen3-4B)下保持良好泛化能力。本工作为利用多步文本反馈提升大模型推理能力提供了有效路径。
原文摘要 · Abstract (English)
Large language models (LLMs) face significant challenges in effectively leveraging sequential environmental feedback (EF) signals, such as natural language evaluations, for feedback-independent chain-of-thought (CoT) reasoning. Existing approaches either convert EF into scalar rewards, losing rich contextual information, or employ refinement datasets, failing to exploit the multi-step and discrete nature of EF interactions. To address these limitations, we propose MoL-RL, a novel training paradigm that integrates multi-step EF signals into LLMs through a dual-objective optimization framework. Our method combines MoL (Mixture-of-Losses) continual training, which decouples domain-specific EF signals (optimized via cross-entropy loss) and general language capabilities (preserved via Kullback-Leibler divergence), with GRPO-based post-training to distill sequential EF interactions into single-step inferences. This synergy enables robust feedback-independent reasoning without relying on external feedback loops. Experimental results on mathematical reasoning (MATH-500, AIME24/AIME25) and code generation (CodeAgent-Test) benchmarks demonstrate that MoL-RL achieves state-of-the-art performance with the Qwen3-8B model, while maintaining strong generalization across model scales (Qwen3-4B). This work provides a promising approach for leveraging multi-step textual feedback to enhance LLMs' reasoning capabilities in diverse domains.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。