让大模型推理更省资源,还能灵活调节大小。
SALAAD: Sparse And Low-Rank Adaptation via ADMM for Large Language Model Inference
- 用优化算法自动学习稀疏低秩结构,不改模型架构。
- 同一训练可生成多种容量版本,部署时无需重训。
- 在不同内存预算下保持性能,内存占用大幅降低。
现代大语言模型在计算和内存受限环境下部署日益普遍,灵活控制模型容量成为核心挑战。尽管稀疏与低秩结构天然可在容量与性能间权衡,但现有方法常依赖启发式设计,忽略层与矩阵异质性,或需特定模型结构修改。本文提出SALAAD,一种即插即用框架,可在训练中诱导稀疏低秩结构。通过在增广拉格朗日框架下建模结构化权重学习,并引入自适应控制器动态平衡训练损失与结构约束,SALAAD在保持标准训练稳定性的同时,实现对有效模型容量演化过程的显式控制。跨模型规模实验表明,SALAAD显著降低部署时的内存消耗,且性能接近专用方法。更重要的是,单次训练即可生成连续容量谱,支持在不同内存预算下平滑弹性部署,无需重新训练。
原文摘要 · Abstract (English)
Modern large language models are increasingly deployed under compute and memory constraints, making flexible control of model capacity a central challenge. While sparse and low-rank structures naturally trade off capacity and performance, existing approaches often rely on heuristic designs that ignore layer and matrix heterogeneity or require model-specific architectural modifications. We propose SALAAD, a plug-and-play framework applicable to different model architectures that induces sparse and low-rank structures during training. By formulating structured weight learning under an augmented Lagrangian framework and introducing an adaptive controller that dynamically balances the training loss and structural constraints, SALAAD preserves the stability of standard training dynamics while enabling explicit control over the evolution of effective model capacity during training. Experiments across model scales show that SALAAD substantially reduces memory consumption during deployment while achieving performance comparable to ad-hoc methods. Moreover, a single training run yields a continuous spectrum of model capacities, enabling smooth and elastic deployment across diverse memory budgets without the need for retraining.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。