arXiv:2605.09853cs.LG2026-05中稿 · TMLR 2026

让大模型推理更灵活:通过探索驱动优化提升答案多样性与准确率

Exploration-Driven Optimization for Test-Time Large Language Model Reasoning

论文配图:Exploration-Driven Optimization for Test-Time Large Language Model Reasoning
图 1 · 摘自论文原文
  • 在后训练中引入探索性目标,平衡采样多样性与强化学习的集中性
  • 在三个内分布基准上提升1.0-1.3%,五项跨分布任务平均增益1.5%
  • 适合依赖测试时计算(如自一致)的大模型推理场景

后训练结合推理时扩展显著提升大语言模型(LLM)的推理与对齐能力。然而存在根本矛盾:推理时方法依赖概率分布较平滑以实现多样采样,而基于强化学习(RL)的后训练会使其分布变尖锐。为此,我们提出探索驱动优化(EDO),将奖励偏置型探索目标扩展至迭代后训练,并融入标准RL目标,促进采样解的多样性,同时支持更高效的测试时计算。我们将EDO集成到迭代直接偏好优化(iDPO)和组相对策略优化(GRPO),形成ED-iDPO与ED-GRPO。大量实验表明,二者均显著提升解的多样性与推理能力,尤其在结合自一致性等测试时技术时表现更优。在三个内分布推理基准上,EDO相比最强基线提升1.0–1.3%;在五个跨分布任务上平均额外增益1.5%。此外,EDO维持模型熵并稳定训练动态,有效防止过优化崩溃。结果表明,EDO是平衡探索与利用的实用框架,尤其适用于依赖测试时扩展的推理场景。

原文摘要 · Abstract (English)

Post-training techniques combined with inference-time scaling significantly enhance the reasoning and alignment capabilities of large language models (LLMs). However, a fundamental tension arises: inference-time methods benefit from diverse sampling from a relatively flattened probability distribution, whereas reinforcement learning (RL)-based post-training inherently sharpens these distributions. To address this, we propose Exploration-Driven Optimization (EDO), which extends reward-biasing style exploration objectives to iterative post-training and integrates them into standard RL objectives, encouraging greater diversity in sampled solutions while facilitating more effective inference-time computation. We incorporate EDO into iterative Direct Preference Optimization (iDPO) and Group Relative Policy Optimization (GRPO), resulting in two variants: ED-iDPO and ED-GRPO. Extensive experiments demonstrate that both ED-iDPO and ED-GRPO exhibit greater solution diversity and improved reasoning abilities, particularly when combined with test-time computation techniques like self-consistency. Across three in-distribution reasoning benchmarks, EDO achieves a 1.0-1.3\% improvement over the strongest baselines, and delivers an additional 1.5\% average gain on five out-of-distribution tasks. Beyond accuracy, EDO preserves model entropy and stabilizes RL training dynamics, highlighting its effectiveness in preventing over-optimization collapse. Taken together, these results establish EDO as a practical framework for balancing exploration and exploitation in LLM reasoning, especially in settings that rely on test-time scaling.

大模型推理探索优化测试时计算

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