arXiv Computer Vision
Sep 4

DSAQuant: Denoising-Stage-Aligned Quantization-Aware Training for Video Generation

The paper introduces DSAQuant, a quantization‑aware training framework tailored for video diffusion models (VDMs). It aligns quantization with the denoising stages of VDMs, using denoising‑stage‑oriented supervision during training and denoising‑stage gated guidance during inference to preserve structure while improving detail reconstruction. Experiments on Wan and CogVideoX models under aggressive W3A3 and W4A4 quantization settings show that DSAQuant outperforms state‑of‑the‑art QAT baselines, boosting VBench scores by up to 6.60 while maintaining strong text‑video alignment.

By Shuaiting Li, Zelin Gao, Haibin Shen, Yujun Shen, Haotong Qin, Yinghao Xu
arXiv Computer Vision
Sep 21

Quantization-Aware Kalman Estimation for Diffusion Sampling

The paper introduces QuAKE, a Quantization-Aware Kalman Estimator designed to correct errors in diffusion model sampling when using quantized denoisers. By treating sampling as an online estimation problem, QuAKE leverages the history of quantized outputs to recover full-precision estimates, updating a posterior in closed form at each step. The method is lightweight, plug‑and‑play, and works with any high‑order multistep ODE sampler, outperforming existing correction techniques on W4A4‑quantized text‑to‑image diffusion models.

By Qitan Shi, Cheng Jin, Jiawei Zhang, Yuantao Gu