arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2604.04701cs.LGcs.AI
收藏

MUXQ: 通过低秩异常分解实现混合到均匀精度矩阵量化

MUXQ: Mixed-to-Uniform Precision MatriX Quantization via Low-Rank Outlier Decomposition

School of Electrical Engineering, Korea University(高丽大学电气工程学院)

Seoungsub Lee, In Seo Kim, Seon Wook Kim

AI总结:

MUXQ通过低秩异常分解检测输入激活中的异常通道,引入辅助矩阵重新分布异常幅度,实现低精度整数量化,降低内存占用并提升边缘设备上的LLM推理效率。

摘要:

大型语言模型(LLMs)在自然语言处理任务中表现出色,但其庞大的参数量导致内存和计算开销巨大。在NPU设备环境中,FP16/FP32计算效率低,整数量化至关重要。然而,现有方法如ZeroQuant、LLM.int8()和SmoothQuant未能充分解决输入激活异常和硬件效率问题。为克服这些限制,我们提出了MUXQ(混合到均匀精度量化)。MUXQ检测输入激活中的异常通道,并引入一个小的辅助矩阵,将异常幅度重新分布到通道中,从而缓解异常问题。这使得即使激活异常也能在低精度INT级别上量化,同时保持硬件友好的计算结构。在WikiText-2数据集上,对GPT-2模型(0.1B、0.3B和0.7B参数)的实验表明,MUXQ在低精度整数量化下实现了比朴素量化更低的困惑度。特别是,在每张张量量化下,MUXQ将激活和权重量化为INT8,同时保持与FP16相近的精度。在仅有适度计算开销的情况下,MUXQ实现了稳定的低精度推理,并能与其他量化技术相结合。这些结果表明,MUXQ为在边缘设备上高效准确的LLM推理提供了一个有前景的方向。

英文摘要:

Large language models (LLMs) have achieved outstanding performance across a wide range of natural language processing tasks, but their enormous parameter counts impose ubstantial memory and computational overheads. This challenge is particularly critical in NPU-based on-device environments, where FP16/FP32 computation is inefficient and integer (INT) quantization is therefore essential. However, existing methods, including ZeroQuant, LLM.int8(), and SmoothQuant, do not fully address input-activation outliers and the associated hardware inefficiencies. To overcome these limitations, we propose MUXQ (Mixed-to-Uniform Quantization). MUXQ detects outlier channels in input activations and introduces a small auxiliary matrix that redistributes outlier magnitudes across channels, thereby alleviating the outlier problem. This enables even activation outliers to be quantized at low-precision INT levels while preserving a hardware-friendly computation structure. Experiments on GPT-2 models at three scales (0.1B, 0.3B, and 0.7B parameters) using the WikiText-2 dataset show that MUXQ consistently achieves lower perplexity than naive quantization. In particular, under per-tensor quantization, MUXQ quantizes both activations and weights to INT8 while maintaining accuracy close to that of FP16. With only modest computational overhead, MUXQ enables stable low-precision inference and can be readily combined with other quantization techniques. These results suggest that MUXQ provides a promising direction for efficient and accurate LLM inference on edge devices.

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

↑