English

Flash-VAED: Plug-and-Play VAE Decoders for Efficient Video Generation

Computer Vision and Pattern Recognition 2026-02-24 v1

Abstract

Latent diffusion models have enabled high-quality video synthesis, yet their inference remains costly and time-consuming. As diffusion transformers become increasingly efficient, the latency bottleneck inevitably shifts to VAE decoders. To reduce their latency while maintaining quality, we propose a universal acceleration framework for VAE decoders that preserves full alignment with the original latent distribution. Specifically, we propose (1) an independence-aware channel pruning method to effectively mitigate severe channel redundancy, and (2) a stage-wise dominant operator optimization strategy to address the high inference cost of the widely used causal 3D convolutions in VAE decoders. Based on these innovations, we construct a Flash-VAED family. Moreover, we design a three-phase dynamic distillation framework that efficiently transfers the capabilities of the original VAE decoder to Flash-VAED. Extensive experiments on Wan and LTX-Video VAE decoders demonstrate that our method outperforms baselines in both quality and speed, achieving approximately a 6×\times speedup while maintaining the reconstruction performance up to 96.9%. Notably, Flash-VAED accelerates the end-to-end generation pipeline by up to 36% with negligible quality drops on VBench-2.0.

Keywords

Cite

@article{arxiv.2602.19161,
  title  = {Flash-VAED: Plug-and-Play VAE Decoders for Efficient Video Generation},
  author = {Lunjie Zhu and Yushi Huang and Xingtong Ge and Yufei Xue and Zhening Liu and Yumeng Zhang and Zehong Lin and Jun Zhang},
  journal= {arXiv preprint arXiv:2602.19161},
  year   = {2026}
}

Comments

Code will be released at https://github.com/Aoko955/Flash-VAED

R2 v1 2026-07-01T10:46:14.649Z