acceptodds
Under review as a conference paper at ICLR 2027

REPA: Residual-Space Pullback for Scale-Consistent Low-Rank Adaptation

Abstract

Parameter-efficient fine-tuning (PEFT) is widely used to adapt large pretrained models under constrained memory and computational budgets, with Low-Rank Adaptation (LoRA) serving as a common foundation. Existing LoRA variants primarily modify the parameterization, scaling, initialization, or optimization of low-rank updates, but generally do not account for how these updates are transformed after entering a residual stream followed by Root Mean Square Layer Normalization (RMSNorm), where the projected first-order response depends on residual scale. To address this residual-scale mismatch, this paper proposes REPA (Residual-Space Pullback for Scale-Consistent Low-Rank Adaptation). REPA projects the LoRA candidate update onto the subspace orthogonal to the current residual state, where the local RMSNorm Jacobian reduces to a scalar response, and rescales the projected update by the current residual RMS. This construction cancels the resulting inverse-scale modulation while preserving the standard LoRA rank and trainable parameter count. Theoretical analysis establishes exact local realization of the desired projected response, controlled second-order finite-step deviation, and uniqueness and minimum-norm properties of the resulting pullback. Across seven downstream benchmarks, REPA achieves state-of-the-art macro-average accuracy among the evaluated baselines at % and achieves optimal or near-optimal performance on every task. Compared with LoRA, REPA improves macro-average accuracy by percentage points with the same trainable parameter count, while requiring only times the full-training time. Controlled residual rescaling from times to times produces the predicted inverse-scale response for LoRA while keeping REPA near its reference response, confirming the scale-correction behavior.

Then back it, or bet against it.

Related papers

Open the market on this paper to see 7 more related papers.