CentriQ: Calibration-Free Quantization of Diffusion Transformers via Exact Mean Centering
Organizations: Tenstorrent · EPFL
Abstract
Diffusion transformers (DiTs) achieve state-of-the-art image generation, but their sampling cost limits deployment. Quantizing both weights and activations to 4 bits reduces this cost, yet existing methods fall short in one of two ways. Calibration-based methods are tied to a specific checkpoint and prompt distribution, whereas data-free Hadamard rotation, effective for LLMs, loses quality on DiTs. We show that this loss has a structural cause. Adaptive layer-norm conditioning adds a per-token mean to the activations, and at the widths of the evaluated DiTs, the Hadamard rotations used by data-free methods cannot spread this mean uniformly across coordinates. A single dominant direction therefore survives the rotation and sets the quantization range. We introduce CentriQ, a calibration-free quantizer that centers each token before rotation and restores the mean exactly through a rank-1 full-precision branch, so that per-token scales follow in closed form without data. Weights are fitted under a robust objective that tracks the dense mode of each group and discounts heavy tails. Across three DiTs, CentriQ matches the quality of calibrated SVDQuant at 4 bits, whereas calibration-free weight quantizers with plain per-token activation quantization collapse or degrade substantially. CentriQ outperforms the strongest calibration-free method reported to date at 2-bit weights. It is also the first calibration-free method to retain usable image quality at 2-bit activations.
Figures & tables
| MJHQ | sDCI | ||||||||
| Method | Calib. | FID | IR | PSNR | LPIPS | FID | IR | PSNR | LPIPS |
| PixArt- | |||||||||
| FP16 | 16.7 | 0.952 | – | – | 24.9 | 0.973 | – | – | |
| RTN | 481.5 | -2.273 | 6.14 | 0.995 | 470.1 | -2.277 | 6.60 | 0.957 | |
| NF4 | 479.6 | -2.273 | 6.16 | 0.997 | 468.6 | -2.277 | 6.46 | 0.966 | |
| HQQ | 462.6 | -2.273 | 6.37 | 0.991 | 456.0 | -2.277 | 6.58 | 0.963 | |
| W4A4 | W4A2 | W2A4 | ||||
|---|---|---|---|---|---|---|
| Configuration | FID | IR | FID | IR | FID | IR |
| No transform | 40.62 | -0.067 | 437.41 | -2.270 | 363.14 | -2.244 |
| rotation (QuaRot-style) | 25.65 | 0.768 | 455.30 | -2.271 | 35.84 | 0.371 |
| centering (exact rank-1 re-injection) | 18.46 | 0.961 | 444.21 | -2.274 | 21.81 | 0.744 |
| percentile clip ( CentriQ ) | 18.80 | 0.939 | 35.59 | 0.394 | 21.56 | 0.728 |
| BF16 | 19.45 | 0.966 | 19.45 | 0.966 | 19.45 | 0.966 |
Appendix figures & tables15 assets
Supplementary material from the paper’s appendix.
Appendix
| Setting | Value |
|---|---|
| Weight grid, group | , |
| Scale and zero point | FP16 (PixArt- ), BF16 (FLUX, SANA) |
| Objective exponent | 0.7 |
| Penalty | 10, fixed |
| Outer rounds / IRLS / proximal iterations | / / 20 |
| Stopping tolerance | , relative |
| MJHQ | sDCI | |||||||
| Precision | FID | IR | PSNR | LPIPS | FID | IR | PSNR | LPIPS |
| BF16 | 19.45 | 0.966 | – | – | 20.92 | 0.984 | – | – |
| W4A4 | 18.80 | 0.939 | 18.40 | 0.173 | 20.6 | 0.970 | 17.29 | 0.173 |
| W4A2 | 35.59 | 0.394 | 13.62 | 0.457 | 28.61 | 0.683 | 12.85 | 0.429 |
| W2A4 | 21.56 | 0.728 | 14.35 | 0.367 | 22.21 | 0.873 | 13.48 | 0.350 |
| Method | Calib. | Single | Two | Counting | Colors | Position | Color attr. | Overall |
|---|---|---|---|---|---|---|---|---|
| BF16 | 0.997 | 0.879 | 0.572 | 0.742 | 0.273 | 0.540 | 0.667 | |
| W4A4 | ||||||||
| Q-DiT † | ✓ | 0.741 | 0.424 | 0.378 | 0.418 | 0.073 | 0.208 | 0.373 |
| SmoothQuant † | ✓ | 0.619 | 0.293 | 0.272 | 0.317 | 0.043 | 0.143 | 0.281 |
| QuaRot † | 0.819 | 0.543 | 0.472 | 0.519 | 0.118 | 0.275 | 0.458 | |
| ViDiT-Q † | ✓ | 0.888 | 0.586 | 0.516 | 0.585 | 0.130 | 0.268 | 0.495 |
| PixArt- | FLUX.1-schnell | SANA-1.6B | ||||
| Weight quantizer | FID | IR | FID | IR | FID | IR |
| FP16 | 16.66 | 0.952 | 19.45 | 0.966 | 15.96 | 1.097 |
| RTN | 32.56 | 0.721 | 18.97 | 0.936 | 18.06 | 0.993 |
| NF4 | 24.40 | 0.834 | 19.19 | 0.926 | 17.97 | 0.942 |
| HQQ | 21.04 | 0.866 | 18.92 | 0.940 | 16.30 | 1.073 |
| RoDS | 19.13 | 0.891 | 18.80 | 0.939 | 16.19 | 1.072 |
| PixArt- | FLUX.1-schnell | SANA-1.6B | ||||
| Method | FID | IR | FID | IR | FID | IR |
| FP16 | 16.66 | 0.952 | 19.45 | 0.966 | 15.96 | 1.097 |
| W4A16 | ||||||
| RTN | 23.04 | 0.823 | 19.02 | 0.968 | 16.32 | 1.107 |
| NF4 | 18.74 | 0.889 | 19.19 | 0.953 | 16.07 | 1.103 |
| HQQ | 16.40 | 0.925 | 18.87 | 0.958 | 16.15 | 1.096 |