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

Mamba-PTQ:循环大语言模型中的异常通道

Mamba-PTQ: Outlier Channels in Recurrent Large Language Models

Intel Labs(英特尔实验室);LMU Munich(慕尼黑大学);University of Groningen(格罗宁根大学)

Alessandro Pierro, Steven Abreu

AI总结:

本研究针对循环大语言模型Mamba的后训练量化问题,发现其存在与注意力模型相似的激活异常通道,并提供了基线结果及异常感知量化的初步方案。

摘要:

现代循环层正成为基础模型边缘部署的一条有前景的路径,尤其是在大语言模型(LLM)的背景下。将整个输入序列压缩为有限维表示,使得循环层能够建模长距离依赖,同时保持每个令牌的恒定推理成本和固定的内存需求。然而,在资源受限环境中实际部署LLM通常需要进一步的模型压缩,例如量化和剪枝。虽然这些技术对于基于注意力的模型已经成熟,但它们对循环层的影响仍未得到充分探索。在这项初步工作中,我们专注于循环LLM的后训练量化,并表明Mamba模型表现出与基于注意力的LLM中观察到的相同的异常通道模式。我们表明,量化SSM困难的原因是由激活异常引起的,类似于在基于Transformer的LLM中观察到的现象。我们报告了不考虑激活异常的Mamba后训练量化的基线结果,并提出了面向异常感知量化的初步步骤。

英文摘要:

Modern recurrent layers are emerging as a promising path toward edge deployment of foundation models, especially in the context of large language models (LLMs). Compressing the whole input sequence in a finite-dimensional representation enables recurrent layers to model long-range dependencies while maintaining a constant inference cost for each token and a fixed memory requirement. However, the practical deployment of LLMs in resource-limited environments often requires further model compression, such as quantization and pruning. While these techniques are well-established for attention-based models, their effects on recurrent layers remain underexplored. In this preliminary work, we focus on post-training quantization for recurrent LLMs and show that Mamba models exhibit the same pattern of outlier channels observed in attention-based LLMs. We show that the reason for the difficulty of quantizing SSMs is caused by activation outliers, similar to those observed in transformer-based LLMs. We report baseline results for post-training quantization of Mamba that do not take into account the activation outliers and suggest first steps for outlier-aware quantization.

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

↑