DualQuant:大模型双尺度无校准量化方法
DualQuant: Dual-scale Calibration-free Weight-and-Activation Quantization for LLMs
清华团队推出DualQuant,无需校准数据就能量化大模型,精度媲美SmoothQuant,性能提升超2倍。
DualQuant是一种双尺度无校准后训练量化框架,直接最小化权重重建均方误差。该方法支持MX系列和NVFP4等多种格式,在WikiText-2、C4基准测试和零样本下游任务中,多数模型和格式的精度匹配或超过基于校准的SmoothQuant。在Llama-3.1-8B上实现真实W4A4推理路径,内存减少最高达2.61倍,吞吐量提升最高2.46倍。
DualQuant: Dual-scale Calibration-free Weight-and-Activation Quantization for LLMs
Low-bit inference on accelerators with native 4-bit support requires quantizing both weights and activations to realize the full throughput gains these formats provide. While post-training weight quantization has been extensively studied, activation quantization remains considerably more challenging and typically requires calibration data. In this work, we introduce DUALQUANT, a dual-scale, calibration-free post-training quantization framework that directly minimizes the weight-reconstruction Mean Squared Error (MSE) and, as an immediate by-product, yields per-channel activation rescalings prior to activation quantization. Our key observation is that the scaling factors obtained by minimizing weight-reconstruction error provide effective activation rescalings, enabling competitive activation quantization without calibration data. We formulate DUALQUANT as a non-differentiable weight-reconstruction problem and develop an efficient solver based on variable splitting and block coordinate descent, where each subproblem admits a closed-form update. DUALQUANT provides a unified algorithm supporting a wide range of formats, including the full MX family and NVFP4. To our knowledge, DUALQUANT is the first dual-scaling method to formally support MX and NV block formats. Across Llama and Qwen models evaluated on WikiText-2, C4, and zero-shot downstream tasks, DUALQUANT matches or exceeds the accuracy of calibration-based SmoothQuant across most models and formats, and substantially outperforms calibration-free baselines, with large gains for MX and NV block formats. We validate these gains with a real packed W4A4 inference path for Llama-3.1-8B on an RTX 4090. The resulting implementation closely matches the accuracy of fake quantization while delivering up to 2.61x memory reduction and up to 2.46x throughput speedup over BF16.