让大模型在强化学习中更专注推理而非死记硬背。
Reasoning or Memorization? Direction-Aware Diversity Exploration in LLM Reinforcement Learning

- 通过识别模型内部的推理-记忆方向,引导探索
- 在数学和通用推理任务上显著优于现有方法
- 适合希望提升模型真实推理能力的研究者
强化学习已成为激发大语言模型推理能力的关键范式,其中探索对发现有效解题路径至关重要。现有探索方法通常在语义或梯度空间中鼓励多样性,却未区分多样性的驱动因素。一条轨迹看似新颖,可能源于新的推理过程,也可能只是记忆模式或捷径的变异。若同等奖励两类变化,可能引导探索走向记忆而非真正的推理改进。本文提出DiRL——一种方向感知的强化学习框架,将探索锚定于策略内部的推理-记忆方向。具体而言,DiRL从模型表征中提取该方向,构建方向加权梯度特征以刻画回溯更新,并设计奖励机制放大与推理一致的探索,抑制与记忆一致的变异。DiRL可无缝集成至标准的组相对策略优化(GRPO)中。在数学与通用推理基准上的大量实验表明,DiRL效果显著优于多种现有探索方法。
原文摘要 · Abstract (English)
Reinforcement learning has become a key paradigm for eliciting reasoning abilities in large language models, where exploration is crucial for discovering effective solution trajectories. Existing exploration methods typically encourage diversity in semantic or gradient spaces, without distinguishing what drives this diversity. A trajectory may appear novel because it follows a new reasoning process, or because it varies memorized patterns and shortcuts. Rewarding both cases equally may steer exploration toward memorization rather than genuine reasoning improvement. In this paper, we propose DiRL, a Direction-Aware Reinforcement Learning framework that anchors exploration to an internal reasoning-memorization direction of the policy. Specifically, DiRL extracts this direction from model representations, constructs direction-weighted gradient features to characterize rollout updates, and shapes rewards to amplify reasoning-aligned exploration while suppressing memorization-aligned variations. DiRL integrates seamlessly into standard Group Relative Policy Optimization (GRPO). Extensive experiments on mathematical and general reasoning benchmarks demonstrate the effectiveness of DiRL, showing significant improvements over various existing exploration methods.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。