mlsanityData to 7 Oct 2026

When Gradient Importance Lies: Adaptive LoRA Rank Allocation Fails Under GRPO

arXiv 2605.07366 ↗First posted 8 May 2026cs.CLcited by 0

Not on the map. Why?

Abstract (arXiv)

Adaptive rank allocation for LoRA - allocating more parameters to important layers and fewer to unimportant ones - consistently improves efficiency under supervised fine-tuning (SFT). We test whether this success transfers to reinforcement learning, specifically Group Relative Policy Optimization (GRPO). Using gradient-magnitude profiling on Qwen 2.5 1.5B with GSM8K, we find that, in our setting, it does not: proportional rank allocation degrades accuracy by 4.5 points compared to uniform allocation (70.0% vs. 74.5%), despite using identical parameter budgets. We identify two mechanisms behind this failure. First, the gradient landscape under GRPO is fundamentally flatter than under SFT: the max-to-min layer importance ratio is only 2.17x, whereas the layer concentration reported by Shi et al. (2024) for SFT (top 30% of layers carrying >80% of the gradient signal) implies a max/min ratio well above 10x. All layers carry meaningful gradient signal; none are truly idle. Second, we observe a gradient amplification effect: non-uniform allocation widens the importance spread from 2.17x to 3.00x, creating a positive feedback loop where high-rank layers absorb more gradient while low-rank layers are progressively silenced. A random-allocation control yields the same amplification (r=0.972 correlation between assigned rank and resulting gradient share), indicating that rank causally determines gradient importance rather than the reverse. The negative result is single-seed and single-task; we present it as preliminary evidence that gradient importance does not predict capacity requirements under RL, and that naive transfer of SFT-era rank allocation strategies to alignment training should be evaluated cautiously.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 6 of its 11 references that are in the map. First posted on arXiv 8 May 2026.

Builds on

6 of 11 references.

References · 6

  1. 2021Training Verifiers to Solve Math Word Problemscited by 9,743
  2. 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 8,851
  3. 2024Understanding Layer Significance in LLM Alignmentcited by 19
  4. 2026IGU-LoRA: Adaptive Rank Allocation via Integrated Gradients and Uncertainty-Aware Scoringcited by 5
  5. 2026Why Is RLHF Alignment Shallow? A Gradient Analysiscited by 2
  6. 2026Aletheia: Gradient-Guided Layer Selection for Efficient LoRA Fine-Tuning Across Architecturescited by 2

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2026Not How Many, But Which: Parameter Placement in Low-Rank Adaptation
  2. 2025TIC-GRPO: Provable and Efficient Optimization for Reinforcement Learning from Human Feedback
  3. 2026FIM-LoRA: Task-Informative Rank Allocation for LoRA via Calibration-Time Gradient-Variance Estimation
  4. 2026ReCo: Reweighting GRPO Against Distributional Concentration
  5. 2026Why GRPO Needs Normalization: A Local-Curvature Perspective on Adaptive Gradients
  6. 2025Learning Without Critics? Revisiting GRPO in Classical Reinforcement Learning Environments
  7. 2026On the Impossibility of Unbiased and Length-Invariant Policy Optimization with Outcome Rewards
  8. 2026Matching Accuracy, Different Geometry: Evolution Strategies vs GRPO in LLM Post-Training