arXiv:2606.31813cs.LGcs.AI2026-06中稿 · ICML被引 1

提出几何保持的正交初始化,让LoRA在强化学习中更稳定高效。

Geometry-Preserving Orthonormal Initialization for Low-Rank Adaptation in RLVR

论文配图:Geometry-Preserving Orthonormal Initialization for Low-Rank Adaptation in RLVR
图 1 · 摘自论文原文
  • 基于正交初始化设计新方法,确保低秩适配与全量微调对齐
  • 在数学推理任务上,新方法训练更稳定且性能超越标准LoRA
  • 解释了已有方法在强化学习中失效的原因,适合关注微调稳定性的研究者

低秩适配(LoRA)及其变体在监督微调(SFT)下实现参数高效微调,但在可验证奖励的强化学习(RLVR)中表现尚不明确。特别是两种结构化初始化的变体PiSSA和MiLoRA,在SFT中优于标准LoRA,但在RLVR中可能表现更差甚至训练不稳定。本文通过理论分析揭示:正交初始化能使LoRA结果与全量微调之间的差距最小化。据此提出几何保持的正交初始化,发展出两个新变体RLPO和RLMO。在数学推理基准上的实验表明,该初始化显著提升训练稳定性并超越标准LoRA,而此前表现优异的方法在RLVR中失效的原因也得以解释。代码与模型检查点公开于https://github.com/Richard-ZZZ/geometry-preserving-orthonormal-init-rlvr。

原文摘要 · Abstract (English)

Low-rank adaptation (LoRA) and its variants enable parameter-efficient fine-tuning of large language models under the supervised fine-tuning (SFT) paradigm. However, their efficacy and behavior under Reinforcement learning with verifiable rewards (RLVR) are less well understood. In particular, two structurally initialized LoRA variants, PiSSA and MiLoRA, which outperform standard LoRA under SFT, can underperform standard LoRA under RLVR and may even exhibit training instability. These observations suggest that how to initialize the low-rank matrices in RLVR remains unclear. In this work, we develop a theoretical analysis of LoRA in RLVR, showing that orthonormal initialization achieves the minimal gap between LoRA outcome and that of full fine-tuning. Guided by this insight, we propose geometry-preserving orthonormal initialization for low-rank adaptation in RLVR, leading to two new variants, RLPO and RLMO. Experiments on mathematical reasoning benchmarks show that the proposed orthonormal initialization stabilizes RLVR training and outperforms standard LoRA, contrasting with PiSSA and MiLoRA. Finally, our unified analysis for LoRA initialization also explains why PiSSA and MiLoRA can underperform in RLVR, which may be of independent interest. Code and checkpoints are publicly available at https://github.com/Richard-ZZZ/geometry-preserving-orthonormal-init-rlvr.

低秩适配强化学习微调优化正交初始化

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