# MPQ-DM: Mixed Precision Quantization for Extremely Low Bit Diffusion Models

Source: [MPQ-DM: Mixed Precision Quantization for Extremely Low Bit Diffusion Models](<https://www.sota2.com/research/paper/68fa763d1e209131cea6c0c4>)

Collection index: [llms.txt](<https://www.sota2.com/research/llms.txt>)

Data updated: 2025-10-26T03:50:30.323Z

Authors: Weilun Feng, Haotong Qin, Chuanguang Yang, Zhulin An, Libo Huang, Boyu Diao, Fei Wang, Renshuai Tao, Yongjun Xu, Michele Magno

Published: 2024-12-16T08:31:55.000Z

## Abstract

Diffusion models have received wide attention in generation tasks. However, the expensive computation cost prevents the application of diffusion models in resource-constrained scenarios. Quantization emerges as a practical solution that significantly saves storage and computation by reducing the bit-width of parameters. However, the existing quantization methods for diffusion models still cause severe degradation in performance, especially under extremely low bit-widths (2-4 bit). The primary decrease in performance comes from the significant discretization of activation values at low bit quantization. Too few activation candidates are unfriendly for outlier significant weight channel quantization, and the discretized features prevent stable learning over different time steps of the diffusion model. This paper presents MPQ-DM, a Mixed-Precision Quantization method for Diffusion Models. The proposed MPQ-DM mainly relies on two techniques:(1) To mitigate the quantization error caused by outlier severe weight channels, we propose an Outlier-Driven Mixed Quantization (OMQ) technique that uses $Kurtosis$ to quantify outlier salient channels and apply optimized intra-layer mixed-precision bit-width allocation to recover accuracy performance within target efficiency.(2) To robustly learn representations crossing time steps, we construct a Time-Smoothed Relation Distillation (TRD) scheme between the quantized diffusion model and its full-precision counterpart, transferring discrete and continuous latent to a unified relation space to reduce the representation inconsistency. Comprehensive experiments demonstrate that MPQ-DM achieves significant accuracy gains under extremely low bit-widths compared with SOTA quantization methods. MPQ-DM achieves a 58\\% FID decrease under W2A4 setting compared with baseline, while all other methods even collapse.

## Recorded benchmark results

A research result applies to its stated dataset, split, metric and evaluation configuration. Compare like-for-like settings, preserve the metric's unit and direction, and distinguish paper-reported results from independent reproduction. The best result recorded here is not a claim of exhaustive, current global coverage.

- [Conditional Image Generation on ImageNet-1K 256x256 (val, CFG=1.5)](<https://www.sota2.com/research/sota/conditional-image-generation-on-imagenet-1k-256x256-val-cfg-1-5.md>): Conditional Image Generation; ImageNet 256x256 CFG=1.5 1K (val); IS: 76.35.

## Original sources and implementation

[Original paper](<https://arxiv.org/abs/2412.11549>)

[PDF](<https://arxiv.org/pdf/2412.11549.pdf>)
