MUXQ: Mixed-to-Uniform Precision MatriX Quantization via Low-Rank Outlier Decomposition
Organizations: School of Electrical Engineering, Korea University, Seoul, Korea
Abstract
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.
Figures & tables
| model | granularity | target layers | bits | perplexity | ||||
| IA(bits) | W(bits) | naive | MUXQ | llm.int8() | fp16 | |||
| GPT2-small(0.1B) | per-vector | 8 | 8 | 25.3825 | 25.28 | 25.2493 | 25.188 | |
| 7 | 8 | 25.8291 | 25.487 | 25.3867 | 25.188 | |||
| 6 | 8 | 30.2001 | 26.9466 | 26.1048 | 25.188 | |||
| 5 | 8 | 152.1251 | 43.129 | 30.5337 | 25.188 | |||
| per-tensor | attention | 8 | 8 | 50.618 | 29.5534 | 28.4595 | 25.188 | |
| perplexity | |||||
|---|---|---|---|---|---|
| IA(bits) | W(bits) | naive | MUXQ | llm.int8() | fp16 |
| 8 | 5 | 27.5149 | 27.3616 | 26.5359 | 25.188 |
| 8 | 4 | 41.6420 | 41.1245 | 35.4289 | 25.188 |
LLM Quantization
Large language model quantization reduces numerical precision to lower memory use and accelerate training or inference.
See all 386 papers