ReLoRA在小模型上反而加剧了秩缺陷,不适用于低算力预训练。
Investigating ReLoRA: Effects on the Learning Dynamics of Small Language Models
- 通过反复合并与重置低秩适配器,尝试提升小模型的秩扩展能力。
- 在11M-66M参数模型上,ReLoRA性能低于全秩训练,且规模越大差距越明显。
- 揭示了小模型对高秩更新敏感,适合研究秩缺陷与高效训练方法。
参数高效方法如LoRA已革新大语言模型微调。ReLoRA将这一思想扩展至预训练阶段,通过重复合并和重初始化低秩适配器,在保持更新低成本的同时增加累积秩,这与高容量模型通过局部低秩轨迹逐步扩展学习的观察一致。然而,近期研究表明小语言模型(SLMs)存在秩缺陷,未能充分利用其可用维度。这引发一个自然问题:ReLoRA的秩扩展更新规则能否引导SLMs走向更健康的训练动态,缓解在容量受限下的秩瓶颈?我们提出,SLMs是理想的测试平台:训练速度快,便于控制变量分析,且秩现象更易测量。本文首次系统研究了ReLoRA在小型模型(11M-66M参数)上的表现,评估了性能与学习动态。在损失、Paloma困惑度和BLiMP指标上,ReLoRA均逊于全秩训练,且性能差距随模型规模增大而扩大。通过对比例有效秩和条件数的分析发现,ReLoRA放大了现有秩缺陷,并在训练早期引入病态更新。结果表明,尽管ReLoRA的合并-重启策略可在大模型中扩展秩,但无法直接适用于容量受限的小模型,提示应采用自适应秩或混合秩方法以实现低算力预训练。
原文摘要 · Abstract (English)
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.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。