AI 生成视频,边生成边实时播放,再不用等了!Adobe 与 MIT 联手推出自回归实时视频生成技术 ——CausVid。
思路就像从下载整部电影到直接观看流媒体的转变,在模型生成首帧画面后,视频便可以即时播放,后续内容则动态生成并无缝衔接。
如果你用过视频生成模型,一定对漫长的等待时间记忆深刻,生成一段 10 秒的视频,往往需要等待好几分钟才可以开始观看。
研究团队表示,这一延迟的根本原因在于:传统视频生成模型普遍采用的双向注意力机制,每一帧都需要参考前后帧的信息。
这就像写故事时必须先构思好整个剧情的所有细节才能动笔,在完整视频生成完毕前,你看不到任何画面。
为此,他们提出了一种全新的解决方案,通过蒸馏预训练的双向扩散模型(DiT),构建自回归生成模型。
实验中,CausVid 基于自回归生成的特性,无需额外训练就能支持多种应用,生成速度和质量均显著超越现有方法。
研究团队还表示将很快开源基于开源模型的实现代码。
如前所述,研究团队通过蒸馏预训练的双向扩散模型(DiT),构建自回归生成模型。
为了进一步提速实现实时视频生成,作者通过分布匹配蒸馏(DMD)将生成步骤从 50 步缩减到仅需 4 步。
DMD 是一种扩散模型蒸馏技术,将多步扩散模型转换为快速的单步生成器。DMD 此前已在图像生成中取得成功,Adobe Firefly 文生图的快速模式就是基于此技术。
本次研究团队将其创新性地应用到视频扩散模型中,实现了显著加速。
然而,自回归模型有一个核心难题 —— 误差累积。
每一帧视频都基于之前的帧生成,早期生成的任何细微缺陷都会被放大,导致生成的视频逐渐偏离预期轨迹。
为了解决这一问题,团队提出了非对称蒸馏策略。具体来说:
引入一个拥有未来信息的双向教师模型,在蒸馏训练阶段指导自回归的单向学生模型。这种教师-学生结构允许模型在生成未来帧时具备更强的精确度。
使用双向教师模型生成的的噪声-数据配对来预训练单向学生模型,提升其后蒸馏训练过程的稳定性。
在训练过程中,针对不同时间点的视频帧施加不同强度的噪声,这一策略使模型能够在测试时基于干净的已生成帧对当前帧进行去噪。
通过这种创新性的非对称蒸馏方法,CausVid 显著减少了自回归模型的误差累积问题,并生成了更高质量的视频内容。
这种非对称蒸馏形式中,学生模型和教师模型使用了不同的架构,而这只有在 DMD 风格的蒸馏中才可行。其他方法,例如渐进式蒸馏(Progressive Distillation)或一致性模型(Consistency Distillation),都要求学生模型和教师模型使用相同的架构。
下面是自回归扩散视频模型的误差累积示例(左图)和 CausVid 结果(右图)对比:
实验中,CausVid 表现惊艳:
得益于单向注意力机制,CausVid 完全支持在大语言模型中广泛应用的 KV 缓存推理技术,从而显著提升了生成效率。结合滑动窗口机制,CausVid 突破了传统模型的长度限制。
尽管训练阶段仅接触过 10 秒的视频,CausVid 依然能够生成长达 30 秒甚至更长的视频,其生成速度和质量均显著超越现有方法。
基于自回归生成的特性,CausVid 无需额外训练就能支持多种应用: