低秩预训练不等于全秩训练,方法间差异显著。
Beyond Perplexity: A Geometric and Spectral Study of Low-Rank Pre-Training

- 用几何与谱分析对比五种低秩方法和全秩训练
- 相同困惑度下,各方法收敛到不同解空间区域
- 激活差异随层数加深,适合关注模型本质的读者
大语言模型预训练受限于全秩参数、梯度和优化器状态的存储开销。低秩预训练应运而生,方法迅速增多。核心问题仍悬而未决:低秩方法能否达到与全秩训练相当的泛化能力?现有比较多依赖单一种子运行的验证困惑度,但困惑度并非解质量的良好代理指标——两种方法可具有相同困惑度,却收敛至损失景观中不同区域及内部表示。本文通过16项指标,在三个模型规模(60M、130M、350M)上系统评估五种低秩方法(GaLore、Fira、CoLA、SLTrain、ReLoRA)与全秩训练的差异,涵盖:1维损失景观沿随机/前K主成分方向、检查点间插值、权重与学习更新的谱结构、以及激活相似性。结果表明,低秩方法彼此不同,也不同于全秩训练,即使困惑度接近。全秩训练在随机方向收敛至更尖锐的谷底,而在前1个主成分方向则相反。各方法收敛至几何上不同的谷底。低秩激活随训练进程在深层逐渐偏离全秩,其中GaLore最接近。此外,验证困惑度不能可靠预测下游性能,加入几何与谱指标后可提升预测能力。
原文摘要 · Abstract (English)
Pre-training large language models is dominated by the memory cost of storing full-rank weights, gradients, and optimizer states. Low-rank pre-training has emerged to address this, and the space of methods has grown rapidly. A central question remains open: do low-rank methods produce models that generalize comparably to full-rank training, or does the rank constraint fundamentally alter the solutions reached? Existing comparisons rely almost entirely on validation perplexity from single-seed runs, often carried forward from prior literature. Yet perplexity is a poor proxy for solution quality; two methods can match on perplexity while converging to different loss landscape regions and internal representations. We close this gap by characterizing the solutions found by five low-rank pre-training methods, GaLore and Fira (memory-efficient optimizers), CoLA and SLTrain (architecture reparameterizations), and ReLoRA (adapter-style updates with periodic resets), against full-rank training at three model scales (60M, 130M, 350M). We evaluate each along 16 metrics across four dimensions: 1-D loss landscape along random/top-K PCA directions, 1-D interpolation between checkpoints, spectral structure of the weights and learned updates, and activation similarity to full-rank training. We show that low-rank methods are not equivalent to full-rank training, nor to one another, even when validation perplexity is close. Full-rank training settles into a sharper basin than low-rank methods along random directions, while the reverse holds for the top-1 PCA direction. Each method converges to a geometrically distinct basin. Low-rank activations diverge from full-rank in later layers as training progresses, with GaLore tracking full-rank most closely. Further, validation perplexity does not translate to downstream performance at every scale. Adding geometric and spectral metrics improves the prediction.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。