arXiv:2508.02668cs.LG2025-08被引 9

用低秩与稀疏结合的方法,让大模型高效预训练而不降性能。

LOST: Low-rank and Sparse Pre-training for Large Language Models

  • 通过奇异值分解保留主要低秩结构,剩余部分构造通道稀疏
  • 60M到7B参数模型均达全量级性能,显存和算力降低显著
  • 适合资源受限下从零训练大模型的研究者与开发者

尽管大语言模型在众多任务上表现优异,但其巨大规模带来了高昂的计算与内存开销。近期研究尝试用低秩参数化降低模型大小与训练成本。在此背景下,稀疏性常被用来弥补低秩压缩丢失的关键信息,通过捕捉残差空间中的显著特征来恢复表达能力。然而,现有方法通常以简单或随意方式结合低秩与稀疏成分,导致性能明显下降。本文提出一种新方法——低秩与稀疏预训练(LOST),巧妙融合低秩与稀疏结构,在严格效率约束下实现大模型从头训练。LOST对权重矩阵进行奇异值分解,保留主导低秩成分,将剩余奇异值用于构建通道稀疏组件,以增强低秩训练的表达能力。我们在60M至7B参数范围的大模型预训练中评估该方法,实验表明,LOST在保持竞争力甚至超越全秩模型表现的同时,显著降低内存与计算开销。代码已公开于 LOST Repo。

原文摘要 · Abstract (English)

While large language models (LLMs) have achieved remarkable performance across a wide range of tasks, their massive scale incurs prohibitive computational and memory costs for pre-training from scratch. Recent studies have investigated the use of low-rank parameterization as a means of reducing model size and training cost. In this context, sparsity is often employed as a complementary technique to recover important information lost in low-rank compression by capturing salient features in the residual space. However, existing approaches typically combine low-rank and sparse components in a simplistic or ad hoc manner, often resulting in undesirable performance degradation compared to full-rank training. In this paper, we propose \textbf{LO}w-rank and \textbf{S}parse pre-\textbf{T}raining (\textbf{LOST}) for LLMs, a novel method that ingeniously integrates low-rank and sparse structures to enable effective training of LLMs from scratch under strict efficiency constraints. LOST applies singular value decomposition to weight matrices, preserving the dominant low-rank components, while allocating the remaining singular values to construct channel-wise sparse components to complement the expressiveness of low-rank training. We evaluate LOST on LLM pretraining ranging from 60M to 7B parameters. Our experiments show that LOST achieves competitive or superior performance compared to full-rank models, while significantly reducing both memory and compute overhead. Moreover, Code is available at \href{https://github.com/JiaxiLi1/LOST-Low-rank-and-Sparse-Training-for-Large-Language-Models}{LOST Repo}

大模型低秩稀疏训练

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