English

Investigating ReLoRA: Effects on the Learning Dynamics of Small Language Models

Computation and Language 2025-10-03 v2 Artificial Intelligence

Abstract

Parameter-efficient methods like LoRA have revolutionised large language model (LLM) fine-tuning. ReLoRA extends this idea to pretraining by repeatedly merging and reinitialising low-rank adapters, increasing cumulative rank while keeping updates cheap. This aligns well with observations that high-capacity models learn through locally low-rank trajectories that expand over time. By contrast, recent work suggests that small language models (SLMs) exhibit rank deficiencies and under-utilise their available dimensionality. This raises a natural question: can ReLoRA's rank-expanding update rule \textit{steer} SLMs toward healthier learning dynamics, mitigating rank bottlenecks in a capacity-constrained regime? We argue SLMs are an ideal testbed: they train quickly, enable controlled ablations, and make rank phenomena more measurable. We present the first systematic study of ReLoRA in SLMs (11M-66M parameters), evaluating both performance and learning dynamics. Across loss, Paloma perplexity, and BLiMP, we find that ReLoRA underperforms full-rank training, with gaps widening at larger scales. Analysis of proportional effective rank and condition numbers shows that ReLoRA amplifies existing rank deficiencies and induces ill-conditioned updates early in training. Our results suggest that while ReLoRA's merge-and-restart strategy can expand ranks in larger models, it does not straightforwardly translate to capacity-limited SLMs, motivating adaptive-rank or hybrid-rank approaches for low-compute pretraining.

Keywords

Cite

@article{arxiv.2509.12960,
  title  = {Investigating ReLoRA: Effects on the Learning Dynamics of Small Language Models},
  author = {Yuval Weiss and David Demitri Africa and Paula Buttery and Richard Diehl Martinez},
  journal= {arXiv preprint arXiv:2509.12960},
  year   = {2025}
}

Comments

12 Pages, 6 Tables, 8 Figures

R2 v1 2026-07-01T05:39:00.791Z