通过鼓励多样推理路径提升大模型探索效率。
Diversity-Incentivized Exploration for Versatile Reasoning
- 用全局序列多样性作为内在奖励,驱动深度探索。
- 在多个任务上优于基线方法,Pass@1和Pass@k均显著提升。
- 适合需要多样化推理能力的复杂任务研究者。
基于可验证奖励的强化学习(RLVR)已成为激励大语言模型推理能力的关键范式。由于推理任务中状态-动作空间庞大且奖励稀疏,现有方法常面临探索不足与样本效率低下的问题。本文提出DIVER(多样性激励探索框架),强调全局序列级多样性对多样化推理的促进作用。我们首先通过实证研究发现全局多样性与推理能力呈强正相关。基于此,引入全局多样性激励作为内在奖励,以在语义结构化空间中促进深度探索。结合该奖励,设计了基于势能的奖励塑造机制,保持最优策略不变性,并采用简单启发式方法缓解潜在的奖励欺骗问题。实验表明,DIVER在多种探索策略下均优于主流的RLVR基线,在域内与域外任务上均表现优异,尤其在Pass@1和Pass@k评估中领先。代码已开源。
原文摘要 · Abstract (English)
Reinforcement Learning with Verifiable Rewards (RLVR) has emerged as a crucial paradigm for incentivizing reasoning capabilities in Large Language Models (LLMs). Due to vast state-action spaces and reward sparsity in reasoning tasks, existing methods often struggle with deficient exploration and poor sample efficiency. In the paper, we propose \textbf{DIVER} (\textbf{D}iversity-\textbf{I}ncentivized Exploration for \textbf{V}ersatil\textbf{E} \textbf{R}easoning), an innovative framework that highlights the pivotal role of global sequence-level diversity to incentivize deep exploration for versatile reasoning. We first conduct a primary empirical study to reveal a strong positive correlation between global diversity and reasoning capacity. Building on this insight, we introduce global diversity incentives as an intrinsic reward to promote deep exploration in a semantically structured space. Incorporating the intrinsic reward, we develop a potential-based reward shaping mechanism to preserve optimal policy invariance and design simple heuristics to mitigate possible reward hacking. Experimental results show that DIVER outperforms competitive RLVR baselines with various exploration strategies on both in-domain and out-of-domain tasks, excelling in both Pass@1 and Pass@k evaluations. Our code is available at https://github.com/NJU-RL/DIVER.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。