arXiv:2605.21468cs.LGcs.CL2026-05被引 1

用极简方法预测大模型强化学习训练轨迹,性能媲美全量训练

You Only Need Minimal RLVR Training: Extrapolating LLMs via Rank-1 Trajectories

论文配图:You Only Need Minimal RLVR Training: Extrapolating LLMs via Rank-1 Trajectories
图 1 · 摘自论文原文
  • 通过线性回归从短周期数据推断参数变化主方向
  • 仅需15%训练步数即可达到甚至超越完整训练效果
  • 适合追求高效训练与低成本部署的研究者和开发者

强化学习结合可验证奖励(RLVR)已成为提升大语言模型推理能力的主流范式,但其参数轨迹的内在几何结构尚未被充分探索。本文发现,RLVR的权重轨迹具有极低秩且高度可预测:绝大多数下游性能提升可通过参数增量的秩-1近似捕捉,且该投影幅度随训练步数近似线性增长。基于此,我们提出简单高效的推断方法RELEX(REinforcement Learning EXtrapolation),仅需短时观察窗口即可估计秩-1子空间,并通过线性回归外推未来检查点,无需额外学习模型。在三个模型(Qwen2.5-Math-1.5B、Qwen3-4B-Base、Qwen3-8B-Base)上,RELEX生成的检查点在域内与域外基准上均匹配或超过原始RLVR性能,仅需全量训练15%的步骤。尤为显著的是,RELEX可在无训练成本下外推至观测窗口10–20倍的步数,且性能持续提升(例如仅观察前50步即可外推至1000步)。消融实验表明,增加子空间秩或引入非线性建模均无法进一步提升外推效果。最后,我们证实其成功源于‘去噪’效应:将更新投影至秩-1子空间可有效消除随机优化噪声,避免外推过程中的性能退化。

原文摘要 · Abstract (English)

Reinforcement learning with verifiable rewards (RLVR) has become a dominant paradigm for improving reasoning in large language models (LLMs), yet the underlying geometry of the resulting parameter trajectories remains underexplored. In this work, we demonstrate that RLVR weight trajectories are extremely low-rank and highly predictable. Specifically, we find that the majority of downstream performance gains are captured by a rank-1 approximation of the parameter deltas, where the magnitude of this projection evolves near-linearly with training steps. Motivated by this, we propose a simple and compute-efficient method RELEX (REinforcement Learning EXtrapolation), which estimates the rank-1 subspace from a short observation window and extrapolates future checkpoints via linear regression, with no learned model required. Across three models (i.e., Qwen2.5-Math-1.5B, Qwen3-4B-Base, and Qwen3-8B-Base), RELEX produces checkpoints that match or exceed RLVR performance on both in-domain and out-of-domain benchmarks, requiring as few as 15% steps of full RLVR training. Remarkably, RELEX is able to extrapolate far beyond the observation window at no training cost, predicting checkpoints up to 10-20$\times$ beyond the observed prefix with continued improvement (e.g., observe only the first 50 steps and extrapolate to 1000 steps). Our ablation analysis confirms the minimalist sufficiency of RELEX: neither increasing the subspace rank nor employing non-linear modeling yields further gains in extrapolation. Finally, we show that RELEX's success stems from a "denoising" effect: by projecting updates onto the rank-1 subspace, the model discards stochastic optimization noise that would otherwise degrade performance during extrapolation. Our code is available at https://github.com/weizhepei/RELEX.

强化学习模型外推高效训练低秩近似

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