VDOT:通过最优传输蒸馏实现高效统一视频创作
VDOT: Efficient Unified Video Creation via Optimal Transport Distillation
将最优传输距离引入分布匹配蒸馏,替代KL散度以缓解少步生成中的梯度崩溃问题,为扩散模型加速提供可复用的训练目标设计。
原文依据 we additionally employ a novel computational optimal transport (OT) technique to optimize the discrepancy between the real and fake score distributions
摘要译文
由模型翻译 · 术语统一 · 请对照原文核对关键结论生成模型的快速发展显著推动了图像与视频应用。其中,旨在各种条件下生成视频的视频创作受到了广泛关注。然而,现有的视频创作模型要么仅关注少数特定条件,要么因复杂的模型推理导致生成时间过长,难以在实际应用中落地。为缓解这些问题,我们提出了一种高效统一的视频创作模型VDOT。具体而言,我们采用分布匹配蒸馏(DMD)范式对训练过程建模。我们未使用Kullback-Leibler(KL)最小化,而是额外采用一种新颖的计算最优传输(OT)技术来优化真实与虚假分数分布之间的差异。OT距离本身施加几何约束,缓解了在少步生成场景中基于KL的蒸馏可能出现的零强制或梯度崩溃问题,从而提升了蒸馏过程的效率与稳定性。此外,我们引入判别器使模型能够感知真实视频数据,进而提升生成视频的质量。为支持统一视频创作模型的训练,我们提出了一条全自动的视频数据标注与过滤流水线,可适配多种视频创作任务。同时,我们构建了一个统一的测试基准UVCBench以推动该领域发展。实验表明,我们的4步VDOT优于或匹配其他50步去噪步骤的基线方法。代码见https://vdot-page.github.io。
英文原文摘要 1700 字符
The rapid development of generative models has significantly advanced image and video applications. Among these, video creation, aimed at generating videos under various conditions, has gained substantial attention. However, existing video creation models either focus solely on a few specific conditions or suffer from excessively long generation times due to complex model inference, making them impractical for real-world applications. To mitigate these issues, we propose an efficient unified video creation model, named VDOT. Concretely, we model the training process with the distribution matching distillation (DMD) paradigm. Instead of using the Kullback-Leibler (KL) minimization, we additionally employ a novel computational optimal transport (OT) technique to optimize the discrepancy between the real and fake score distributions. The OT distance inherently imposes geometric constraints, mitigating potential zero-forcing or gradient collapse issues that may arise during KL-based distillation within the few-step generation scenario, and thus, enhances the efficiency and stability of the distillation process. Further, we integrate a discriminator to enable the model to perceive real video data, thereby enhancing the quality of generated videos. To support training unified video creation models, we propose a fully automated pipeline for video data annotation and filtering that accommodates multiple video creation tasks. Meanwhile, we curate a unified testing benchmark, UVCBench, to advance the field. Experiments demonstrate that our 4-step VDOT outperforms or matches the performance of other baselines with 50 denoising steps. Code is available at https://vdot-page.github.io.