arXiv:2502.15077cs.CVcs.AI2025-02中稿 · MIPR 2025被引 1

提出无需动态量化即可部署视频生成模型的静态量化方法。

Hardware-Friendly Static Quantization Method for Video Diffusion Transformers

  • 用每步校准数据+通道/张量量化实现静态量化
  • 视频质量接近FP16和动态量化,CLIP/VQA指标达标
  • 适合在资源受限设备上部署视频扩散模型

视频生成领域的扩散变压器自SORA展现卓越性能后备受关注。尽管动态量化已在GPU上实现高效部署,但资源受限设备无法支持动态量化,需依赖静态量化以在AI处理器上高效运行。本文提出一种无需动态量化技术的OpenSora视频扩散变压器后训练静态量化新方法。该方法利用每步校准数据,对每个时间步生成静态量化模型,结合权重量化采用通道级、激活量化采用张量级策略。通过进一步引入smooth-quantization技术,可获得高质量视频输出。大量实验表明,静态量化可作为视频扩散变压器的可行替代方案,在不损失性能的前提下提供更高效的部署路径。

原文摘要 · Abstract (English)

Diffusion Transformers for video generation have gained significant research interest since the impressive performance of SORA. Efficient deployment of such generative-AI models on GPUs has been demonstrated with dynamic quantization. However, resource-constrained devices cannot support dynamic quantization, and need static quantization of the models for their efficient deployment on AI processors. In this paper, we propose a novel method for the post-training quantization of OpenSora\cite{opensora}, a Video Diffusion Transformer, without relying on dynamic quantization techniques. Our approach employs static quantization, achieving video quality comparable to FP16 and dynamically quantized ViDiT-Q methods, as measured by CLIP, and VQA metrics. In particular, we utilize per-step calibration data to adequately provide a post-training statically quantized model for each time step, incorporating channel-wise quantization for weights and tensor-wise quantization for activations. By further applying the smooth-quantization technique, we can obtain high-quality video outputs with the statically quantized models. Extensive experimental results demonstrate that static quantization can be a viable alternative to dynamic quantization for video diffusion transformers, offering a more efficient approach without sacrificing performance.

视频生成扩散模型量化静态量化

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。