arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2505.14871cs.CLcs.LG

Saten:用于大型语言模型训练后压缩的稀疏增强张量网络

Saten: Sparse Augmented Tensor Networks for Post-Training Compression of Large Language Models

University of California-Santa Barbara(加州大学圣巴巴拉分校);Amazon(亚马逊)

Ryan Solgi, Kai Zhen, Rupak Vignesh Swaminathan, Nathan Susanj, Athanasios Mouchtaris, Siegfried Kunzmann, Zheng Zhang

AI总结:

针对预训练大型语言模型因高秩特性和缺乏预训练数据导致训练后压缩困难的问题,提出稀疏增强张量网络(Saten)框架,通过微调期间的低秩张量化实现全模型压缩,提升了准确性与压缩效率并达到最先进性能。

摘要:

大型语言模型(LLMs)的高效实现对在资源受限设备上的部署至关重要。低秩张量压缩技术,例如张量训练(TT)网络,已被广泛应用于研究过度参数化的神经网络。然而,由于预训练LLMs的高秩特性以及无法获取预训练数据,将其应用于压缩预训练大型语言模型(LLMs)以处理下游任务(训练后)仍然具有挑战性。在本研究中,我们探讨了微调过程中的低秩张量化LLMs,并提出了稀疏增强张量网络(Saten)以提升其性能。所提出的Saten框架能够实现全模型压缩。实验结果表明,Saten提升了张量化语言模型的准确性和压缩效率,取得了最先进的性能。

英文摘要:

The efficient implementation of large language models (LLMs) is crucial for deployment on resource-constrained devices. Low-rank tensor compression techniques, such as tensor-train (TT) networks, have been widely studied for over-parameterized neural networks. However, their applications to compress pre-trained large language models (LLMs) for downstream tasks (post-training) remains challenging due to the high-rank nature of pre-trained LLMs and the lack of access to pretraining data. In this study, we investigate low-rank tensorized LLMs during fine-tuning and propose sparse augmented tensor networks (Saten) to enhance their performance. The proposed Saten framework enables full model compression. Experimental results demonstrate that Saten enhances both accuracy and compression efficiency in tensorized language models, achieving state-of-the-art performance.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑