通过选择性稀疏化低秩更新,实现小存储下大模型高效推理。
Optimal Singular Damage: Efficient LLM Inference in Low Storage Regimes
- 利用参数更新的低秩与稀疏特性,设计选择性稀疏化方法。
- 相同内存预算下,精度显著优于单独使用低秩或稀疏化。
- 适合资源受限场景下的大模型部署与微调模型存储优化。
大型语言模型(LLMs)在众多应用中日益普及,但其巨大规模限制了存储与计算能力,仅少数资源丰富机构可承担。多数应用依赖预训练模型的微调版本,然而微调后模型的存储仍是挑战。研究发现微调主要影响少量参数,提示需更高效的存储方案。本文聚焦微调后参数更新的存储效率问题,基于微调更新兼具低秩与稀疏性的特点,提出最优奇异损伤(Optimal Singular Damage)方法:通过分析奇异向量的交错重要性,选择性稀疏化低秩近似更新,保留最具表达力的成分。实验表明,在相同内存预算下,该方法在存储效率和模型精度上均优于独立使用低秩近似或稀疏化的方案。
原文摘要 · Abstract (English)
Large language models (LLMs) are increasingly prevalent across diverse applications. However, their enormous size limits storage and processing capabilities to a few well-resourced stakeholders. As a result, most applications rely on pre-trained LLMs, fine-tuned for specific tasks. However, even storing the fine-tuned versions of these models remains a significant challenge due to the wide range of tasks they address. Recently, studies show that fine-tuning these models primarily affects a small fraction of parameters, highlighting the need for more efficient storage of fine-tuned models. This paper focuses on efficient storage of parameter updates in pre-trained models after fine-tuning. To address this challenge, we leverage the observation that fine-tuning updates are both low-rank and sparse, which can be utilized for storage efficiency. However, using only low-rank approximation or sparsification may discard critical singular components that enhance model expressivity. We first observe that given the same memory budget, sparsified low-rank approximations with larger ranks outperform standard low-rank approximations with smaller ranks. Building on this, we propose our method, optimal singular damage, that selectively sparsifies low-rank approximated updates by leveraging the interleaved importance of singular vectors, ensuring that the most impactful components are retained. We demonstrate through extensive experiments that our proposed methods lead to significant storage efficiency and superior accuracy within the same memory budget compared to employing the low-rank approximation or sparsification individually.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。