arXiv:2502.06155cs.CV2025-02被引 28

提出高效视频生成方法,提速7.4倍且画质损失小。

Efficient-vDiT: Efficient Video Diffusion Transformers With Attention Tile

  • 用分块稀疏注意力机制降低计算复杂度
  • 仅用0.1%数据训练,29帧视频生成快7.8倍
  • 适合需要快速生成高质量视频的场景

尽管扩散变换器(DiTs)在生成高保真视频方面前景广阔,但其3D全注意力机制因注意力计算复杂和采样步骤多导致推理成本高昂。例如,主流Open-Sora-Plan模型生成29帧视频需超过9分钟。本文从两方面解决效率问题:1)基于视频数据中普遍存在的分块重复模式,提出一种稀疏3D注意力机制,将计算复杂度降至线性;2)通过多步一致性蒸馏缩短采样过程,将整个采样轨迹分段蒸馏,激活少步生成能力。进一步设计三阶段训练流程融合低复杂度注意力与少步生成能力。结果表明,仅使用0.1%预训练数据,即可将Open-Sora-Plan-1.2模型转化为高效版本,在29帧和93帧720p视频生成上分别提速7.4倍至7.8倍,且在VBench评估中性能损失微小。此外,该方法支持分布式推理,在4张GPU上结合序列并行实现额外3.91倍加速。

原文摘要 · Abstract (English)

Despite the promise of synthesizing high-fidelity videos, Diffusion Transformers (DiTs) with 3D full attention suffer from expensive inference due to the complexity of attention computation and numerous sampling steps. For example, the popular Open-Sora-Plan model consumes more than 9 minutes for generating a single video of 29 frames. This paper addresses the inefficiency issue from two aspects: 1) Prune the 3D full attention based on the redundancy within video data; We identify a prevalent tile-style repetitive pattern in the 3D attention maps for video data, and advocate a new family of sparse 3D attention that holds a linear complexity w.r.t. the number of video frames. 2) Shorten the sampling process by adopting existing multi-step consistency distillation; We split the entire sampling trajectory into several segments and perform consistency distillation within each one to activate few-step generation capacities. We further devise a three-stage training pipeline to conjoin the low-complexity attention and few-step generation capacities. Notably, with 0.1% pretraining data, we turn the Open-Sora-Plan-1.2 model into an efficient one that is 7.4x -7.8x faster for 29 and 93 frames 720p video generation with a marginal performance trade-off in VBench. In addition, we demonstrate that our approach is amenable to distributed inference, achieving an additional 3.91x speedup when running on 4 GPUs with sequence parallelism.

视频生成扩散模型高效推理

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