arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2401.16367cs.LGcs.AIcs.CL
收藏

TQCompressor:通过置换改进神经网络中的张量分解方法

TQCompressor: improving tensor decomposition methods in neural networks via permutations

Terra Quantum AG(特拉量子公司)

V. Abronin, A. Naumov, D. Mazur, D. Bystrov, K. Tsarova, Ar. Melnikov, I. Oseledets, S. Dolgov, R. Brasher, M. Perelshtein

AI总结:

针对预训练语言模型的计算存储压力,该研究提出带置换增强的TQCompressor张量分解压缩方法,结合多步知识蒸馏,用少量数据将GPT-2_small压缩至8100万参数,性能优于同类模型,助力资源受限场景部署。

摘要:

我们提出TQCompressor,这是一种采用改进型张量分解的神经网络模型压缩新方法。我们探讨了NLP任务中预训练语言模型的计算与存储需求带来的挑战,并提出基于置换的Kronecker分解增强方案。该增强方案能够降低通常与因子分解相关的模型表达能力损失。我们在GPT-2_small上验证了该方法,压缩后得到TQCompressedGPT-2模型,参数量为8100万,而GPT-2_small的参数量为1.24亿,我们已公开TQCompressedGPT-2。我们进一步通过多步知识蒸馏训练策略提升TQCompressedGPT-2的性能,仅使用3.1%的OpenWebText数据。在对比评估中TQCompressedGPT-2优于DistilGPT-2和KngPT-2,为资源受限环境下高效部署模型提供了新进展。

英文摘要:

We introduce TQCompressor, a novel method for neural network model compression with improved tensor decompositions. We explore the challenges posed by the computational and storage demands of pre-trained language models in NLP tasks and propose a permutation-based enhancement to Kronecker decomposition. This enhancement makes it possible to reduce loss in model expressivity which is usually associated with factorization. We demonstrate this method applied to the GPT-2$_{small}$. The result of the compression is TQCompressedGPT-2 model, featuring 81 mln. parameters compared to 124 mln. in the GPT-2$_{small}$. We make TQCompressedGPT-2 publicly available. We further enhance the performance of the TQCompressedGPT-2 through a training strategy involving multi-step knowledge distillation, using only a 3.1% of the OpenWebText. TQCompressedGPT-2 surpasses DistilGPT-2 and KnGPT-2 in comparative evaluations, marking an advancement in the efficient and effective deployment of models in resource-constrained environments.

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

↑