6Bit-Diffusion: Inference-Time Mixed-Precision Quantization for Video Diffusion Models
Organizations: Fudan University · Tsinghua University · Shanghai Jiao Tong University
Abstract
Diffusion transformers have demonstrated remarkable capabilities in generating videos. However, their practical deployment is severely constrained by high memory usage and computational cost. Post-Training Quantization provides a practical way to reduce memory usage and boost computation speed. Existing quantization methods typically apply a static bit-width allocation, overlooking the quantization difficulty of activations across diffusion timesteps, leading to a suboptimal trade-off between efficiency and quality. In this paper, we propose a inference time NVFP4/INT8 Mixed-Precision Quantization framework. We find a strong linear correlation between a block's input-output difference and the quantization sensitivity of its internal linear layers. Based on this insight, we design a lightweight predictor that dynamically allocates NVFP4 to temporally stable layers to maximize memory compression, while selectively preserving INT8 for volatile layers to ensure robustness. This adaptive precision strategy enables aggressive quantization without compromising generation quality. Beside this, we observe that the residual between the input and output of a Transformer block exhibits high temporal consistency across timesteps. Leveraging this temporal redundancy, we introduce Temporal Delta Cache (TDC) to skip computations for these invariant blocks, further reducing the computational cost. Extensive experiments demonstrate that our method achieves 1.92 end-to-end acceleration and 3.32 memory reduction, setting a new baseline for efficient inference in Video DiTs.
Figures & tables
| Method | Bit-width | Aesthetic | BG. | Overall | Dynamic | Subject | Imaging | Scene | Motion |
|---|---|---|---|---|---|---|---|---|---|
| (W/A) | Quality | Consist. | Consist. | Degree | Consist. | Quality | Consist. | Smooth. | |
| CogVideo-2B | |||||||||
| Original | W16/A16 | 0.5464 | 0.9500 | 0.2423 | 0.7639 | 0.9491 | 0.5713 | 0.3590 | 0.9714 |
| SmoothQuant | W4A8 | 0.5332 | 0.9441 | 0.2473 | 0.6806 | 0.9319 | 0.5460 | 0.3343 | 0.9680 |
| QuaRot | W4A8 | 0.5158 | 0.9427 | 0.2383 | 0.5139 | 0.9202 | 0.5258 | 0.2682 | 0.9715 |
| ViDiT-Q | W4A8 | 0.5332 | 0.9449 | 0.2464 | 0.6528 | 0.9272 | 0.5534 | 0.3372 | 0.9692 |
| Method | Bit-width | Speedup | Aesthetic | BG. | Overall | Dynamic | Subject | Imaging | Scene | Motion |
|---|---|---|---|---|---|---|---|---|---|---|
| (W/A) | Quality | Consist. | Consist. | Degree | Consist. | Quality | Consist. | Smooth. | ||
| Original | W16A16 | 0.5464 | 0.9500 | 0.2423 | 0.7639 | 0.9491 | 0.5713 | 0.3590 | 0.9714 | |
| Ours(DMPQ) | W4A6 | 0.5437 | 0.9455 | 0.2474 | 0.6806 | 0.9282 | 0.5511 | 0.3307 | 0.9691 | |
| Ours(Full) | W4A6 ∗ | 0.5366 | 0.9421 | 0.2492 | 0.7083 | 0.9255 | 0.5369 | 0.3307 | 0.9691 |
| Method | DMPQ | TDC | PDR | VQA-A | VQA-T | CLIP | CLIP-T | Flow Score |
|---|---|---|---|---|---|---|---|---|
| Original (FP16) | - | - | - | 74.6175 | 77.7548 | 0.1824 | 0.9973 | 5.8389 |
| Uniform W4A4 | - | - | - | 68.9105 | 73.6116 | 0.1818 | 0.9974 | 4.2518 |
| Uniform W4A8 | - | - | - | 71.2093 | 78.6070 | 0.1859 | 0.9972 | 5.3140 |
| Ablation 1 | ✓ | 75.5332 | 77.8404 | 0.1814 | 0.9976 | 5.3266 | ||
| Ablation 2 | ✓ | 72.2671 | 78.1517 | 0.1838 | 0.9975 | 5.7602 | ||
| Ablation 3 | ✓ | ✓ | 73.7243 | 77.2833 | 0.1835 | 0.9978 | 4.7271 |
Diffusion Model Quantization
Diffusion model quantization compresses model weights and activations to reduce inference memory and latency while preserving generated image, video, and speech quality.
See all 39 papers