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

缓解语言模型低精度微调中的异常激活

Mitigating Outlier Activations in Low-Precision Fine-Tuning of Language Models

McGill University(麦吉尔大学)

Alireza Ghaffari, Justin Yu, Mahsa Ghazvini Nejad, Masoud Asgharian, Boxing Chen, Vahid Partovi Nia

AI总结:

本文针对语言模型低精度微调中异常激活影响性能的问题,提出用8位整数表示异常值并结合算子分块避免16位整数矩阵乘法的方法,通过理论与实验证明其能提升模型鲁棒性和性能。

摘要:

语言模型的低精度微调作为一种经济高效且节能的方法,在各种应用中部署大规模模型方面日益受到重视。然而,这种方法容易受到激活中异常值的影响。激活中的异常值会对低精度机制下语言模型微调的性能产生负面影响,因为它们会影响缩放因子,从而使表示较小的值变得更加困难。本文研究了在语言模型的低精度整数微调中缓解异常激活的技术。我们提出的新方法使我们能够用8位整数而不是浮点(FP16)值来表示异常激活值。使用整数表示异常值的好处在于,它使我们能够利用算子分块来避免执行16位整数矩阵乘法,从而有效解决这一问题。我们提供了理论分析和支持性实验,以证明我们的方法在提高低精度微调语言模型的鲁棒性和性能方面的有效性。

英文摘要:

Low-precision fine-tuning of language models has gained prominence as a cost-effective and energy-efficient approach to deploying large-scale models in various applications. However, this approach is susceptible to the existence of outlier values in activation. The outlier values in the activation can negatively affect the performance of fine-tuning language models in the low-precision regime since they affect the scaling factor and thus make representing smaller values harder. This paper investigates techniques for mitigating outlier activation in low-precision integer fine-tuning of the language models. Our proposed novel approach enables us to represent the outlier activation values in 8-bit integers instead of floating-point (FP16) values. The benefit of using integers for outlier values is that it enables us to use operator tiling to avoid performing 16-bit integer matrix multiplication to address this problem effectively. We provide theoretical analysis and supporting experiments to demonstrate the effectiveness of our approach in improving the robustness and performance of low-precision fine-tuned language models.

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

↑