arXiv:2510.26491cs.LG2025-10被引 8

用影响函数选关键数据,让大模型强化学习更省数据、更快训练。

Data-Efficient RLVR via Off-Policy Influence Guidance

  • 基于影响函数评估数据重要性,理论指导数据选择。
  • 1.5B模型仅用10%数据,训练速度提升2.66倍。
  • 适合追求高效训练的大模型强化学习研究者。

在具有可验证奖励的强化学习(RLVR)中,数据选择对提升大语言模型(LLM)推理能力至关重要。现有方法多依赖启发式策略,缺乏理论保障和泛化能力。本文提出一种基于影响函数的理论化数据选择方法,通过离线轨迹估算数据贡献,克服在线影响估计的高计算成本。为应对大模型梯度的高维性,采用稀疏随机投影降维,提升存储与计算效率。基于此,构建了多阶段强化学习框架CROPI,迭代筛选当前策略下最具影响力的数据。实验表明,在最大70亿参数的模型上,该方法显著加速训练;在15亿参数模型上,每阶段仅使用10%数据即可实现2.66倍的步级加速,凸显基于影响函数的数据选择在高效RLVR中的巨大潜力。

原文摘要 · Abstract (English)

Data selection is a critical aspect of Reinforcement Learning with Verifiable Rewards (RLVR) for enhancing the reasoning capabilities of large language models (LLMs). Current data selection methods are largely heuristic-based, lacking theoretical guarantees and generalizability. This work proposes a theoretically-grounded approach using influence functions to estimate the contribution of each data point to the learning objective. To overcome the prohibitive computational cost of policy rollouts required for online influence estimation, we introduce an off-policy influence estimation method that efficiently approximates data influence using pre-collected offline trajectories. Furthermore, to manage the high-dimensional gradients of LLMs, we employ sparse random projection to reduce dimensionality and improve storage and computation efficiency. Leveraging these techniques, we develop \textbf{C}urriculum \textbf{R}L with \textbf{O}ff-\textbf{P}olicy \text{I}nfluence guidance (\textbf{CROPI}), a multi-stage RL framework that iteratively selects the most influential data for the current policy. Experiments on models up to 7B parameters demonstrate that CROPI significantly accelerates training. On a 1.5B model, it achieves a 2.66x step-level acceleration while using only 10\% of the data per stage compared to full-dataset training. Our results highlight the substantial potential of influence-based data selection for efficient RLVR.

强化学习数据高效大模型影响函数

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