arXiv:2602.12429cs.LG2026-02中稿 · ICML被引 6

无需全秩辅助,直接用低秩权重训练大模型,稳定且高效。

Stabilizing Native Low-Rank LLM Pretraining

  • 提出Spectron方法,通过谱归一化控制权重更新,解决低秩训练不稳问题。
  • 在LLaMA-7B规模上实现端到端低秩训练,损失无大幅波动,性能接近稠密模型。
  • 适用于追求高效推理的部署场景,尤其适合资源受限环境下的模型压缩。

基础模型取得了显著成功,但参数量持续增长带来了巨大的计算与内存挑战。低秩分解为降低训练与推理成本提供了有前景的路径,但目前尚缺乏仅使用低秩权重从零开始训练模型且性能媲美稠密模型的稳定方法。我们证明,大型语言模型(LLMs)可完全基于低秩分解权重从头训练,所有非嵌入矩阵均采用低秩形式,无需以往方法依赖的额外全秩引导。尽管原生低秩训练常出现不稳定性与损失突增,我们识别出权重更新的谱范数(最大奇异值)失控增长是主导因素。为此,我们提出Spectron:一种结合正交化与谱归一化的动态约束机制,根据当前因子的谱范数实时限制权重更新。该方法实现了稳定、无额外开销的端到端低秩训练。最后,我们建立了原生低秩Transformer的计算最优缩放规律,揭示了可预测的幂律行为,并在推理效率上优于稠密模型。

原文摘要 · Abstract (English)

Foundation models have achieved remarkable success, yet their growing parameter counts pose significant computational and memory challenges. Low-rank factorization offers a promising route to reduce training and inference costs, but the community lacks a stable recipe for training models from scratch using exclusively low-rank weights while matching the performance of the dense model. We demonstrate that Large Language Models (LLMs) can be trained from scratch using exclusively low-rank factorized weights for all non-embedding matrices without auxiliary "full-rank" guidance required by prior methods. While native low-rank training often suffers from instability and loss spikes, we identify uncontrolled growth in the spectral norm (largest singular value) of the weight matrix update as the dominant factor. To address this, we introduce Spectron: Spectral renormalization with orthogonalization, which dynamically bounds the resultant weight updates based on the current spectral norms of the factors. Our method enables stable, end-to-end factorized training with negligible overhead. Finally, we establish compute-optimal scaling laws for natively low-rank transformers, demonstrating predictable power-law behavior and improved inference efficiency relative to dense models.

低秩训练大模型压缩稳定训练谱归一化

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。