← 返回 2026-03-02

模式寻觅遇见均值寻觅:面向快速长视频生成的解耦训练范式 Mode Seeking meets Mean Seeking for Fast Long Video Generation

Shengqu Cai, Weili Nie, Chao Liu, Julius Berner, Lvmin Zhang, Nanye Ma, Hansheng Chen, Maneesh Agrawala, Leonidas Guibas, Gordon Wetzstein, Arash Vahdat 📅 2026-02-27 👍 41 2026-07-13 08:35
Flow Matching 扩散模型 知识蒸馏 视频生成 长视频

用解耦扩散Transformer将短视频保真度与长时序一致性分离,实现分钟级快速视频生成

前置知识

Rectified Flow / 整流流

整流流是一种生成模型参数化方式,通过构造从数据 $x_0$ 到噪声 $z$ 的线性插值路径 $x_t = (1-t)x_0 + tz$,并训练网络预测该路径上的速度场 $u_\theta$。采样时从 $t=1$(纯噪声)沿 ODE 轨迹积分回 $t=0$(干净数据)即可生成样本。相比 DDPM 等扩散模型,整流流的路径更直、采样步数更少,是当前主流视频生成模型(如 Wan、CogVideoX)的标准训练范式。

本文的 Flow Matching 头直接基于整流流目标训练,理解这一范式是理解方法细节的基础。

Reverse KL Divergence / 反向KL散度

KL散度衡量两个概率分布的差异。正向KL $D_{KL}(p\|q)$ 是 mode-covering(均值寻觅),倾向于让 $q$ 覆盖 $p$ 的所有模式;反向KL $D_{KL}(q\|p)$ 是 mode-seeking(模式寻觅),倾向于让 $q$ 集中在 $p$ 的某个高密度模式上。在蒸馏场景中,反向KL让学生模型专注于教师分布中最优质的样本模式,而不是平均化所有模式,这对于保持生成质量至关重要。

本文的核心创新就是利用反向KL的模式寻觅特性来对齐学生与教师的滑动窗口分布,理解这一性质才能理解为什么选择反向KL而非正向KL。

Distribution Matching Distillation (DMD) / 分布匹配蒸馏

DMD 是一种快速蒸馏技术,将预训练扩散模型(教师)的知识压缩到少步学生模型中。其核心思想是:计算学生生成分布与教师分布之间的反向KL梯度,该梯度可以用学生和教师在同一噪声状态下的速度/得分差异来近似表达。具体实现时,训练一个"假"得分估计器 $v_{fake}$ 来估计学生分布的得分,然后用教师速度与 $v_{fake}$ 的差作为梯度信号更新学生。

本文的 Distribution Matching 头采用 DMD 风格的梯度代理来实现滑动窗口级别的教师对齐,这是方法实现的关键技术组件。

Decoupled Diffusion Transformer (DDT) / 解耦扩散Transformer

DDT 是一种网络架构设计,将扩散模型分为共享编码器和两个独立解码头:一个负责标准扩散/流匹配训练,另一个负责蒸馏或特殊目标。两个头共享同一中间特征表示,但接收不同的梯度信号,避免不同训练目标之间的梯度干扰。这种设计在图像生成中已被证明有效。

本文采用DDT架构来分离均值寻觅的SFT目标和模式寻觅的教师对齐目标,是解决梯度冲突的核心架构选择。

Sliding Window Strategy / 滑动窗口策略

对于一个长为 $T_{long}$ 帧的长视频潜在表示,以窗口长度 $L$(对应约5秒)进行裁剪,得到一系列重叠的短视频片段 $\text{crop}_k(x_0^{long})$。每个窗口都与短视频教师的输出分布进行匹配,从而确保长视频的每一局部片段都具有短视频级别的质量。滑动窗口使得短视频教师无需理解长时序上下文即可指导长视频生成。

这是连接短视频教师与长视频学生的关键桥梁机制,使得方法无需额外短视频数据即可将短视频先验迁移到长视频生成中。

研究动机

当前视频生成模型在从秒级扩展到分钟级时面临一个根本性瓶颈。秒级短视频数据在互联网上大量存在(如广泛使用的视频-文本数据集),因此最强的短视频生成器已经通过海量数据训练和精细后处理达到了极高的保真度。然而,分钟级的高质量长视频数据极度稀缺,不仅数量有限、领域窄、异质性高,而且筛选和标注成本极高。这导致一个关键的失败模式:即使在混合长度数据上训练,能生成更长序列的模型往往以牺牲局部动态质量为代价——输出变得更模糊、更不生动、缺乏细节。这是因为模型被迫从数据和计算最匮乏的长视频数据中重新学习高保真的短视频先验,形成"保真度-时间跨度"(fidelity-horizon)的根本性矛盾。

本文的目标是本文的目标是设计一种训练范式,能够在不损失短视频级别局部保真度的前提下,将视频生成能力从秒级扩展到分钟级。具体而言,该方法应能在仅4步推理(NFE=4)的条件下,同时实现:(1) 长时序一致性——前景主体在数十秒内保持身份不变,背景平滑演化;(2) 局部高保真——纹理清晰、运动自然、细节丰富;(3) 快速推理——无需多步迭代即可生成分钟级视频。

与已有工作不同的是,本文抓住了一个被现有工作普遍忽视的核心观察:视频的时间维度与图像的空间分辨率本质上不同。从 $256\times256$ 到 $1024\times1024$ 的图像分辨率提升本质上是同一底层分布的插值——局部patch共享相同分布;但从5秒到1分钟的视频长度提升是时间外推——需要引入新事件、因果链和叙事结构,信息量级完全不同。基于这一洞察,作者认为将不同长度视频混合训练、期望模型"平滑插值"的隐含假设是根本性错误的,由此提出了将局部保真度与长时序一致性完全解耦的训练范式。

核心方法

本文的方法可以用一个简洁的类比来理解:想象一个画家要画一幅超长的全景画。他有两个师父——一个擅长画每个局部细节(短视频教师),另一个只能看到局部但不知道整体构图;另一个师父(长视频数据)告诉他整幅画应该是什么样的叙事结构。本文的做法是:让全景画学生同时向两个师父学习,但用不同的"笔"来接受不同的教导——一只笔专门画整体构图(Flow Matching头),另一只笔专门画局部细节(Distribution Matching头),两只笔共享同一块画布(共享编码器)。技术路线是:基于解耦扩散Transformer(DDT)架构,将长视频潜在表示输入共享编码器得到统一特征,然后通过Flow Matching头在真实长视频上进行监督学习(均值寻觅),同时通过Distribution Matching头将每个滑动窗口与冻结的短视频教师进行反向KL对齐(模式寻觅)。推理时仅使用Distribution Matching头,实现快速少步生成。

本文最本质的创新在于识别并解决了一个核心矛盾:Flow Matching的均值寻觅(mean-seeking)目标与Distribution Matching的模式寻觅(mode-seeking)目标之间存在梯度冲突。Flow Matching损失在噪声条件下训练网络预测条件均值,倾向于在歧义情况下平均化;而反向KL教师对齐则倾向于让学生集中在教师的高保真模式上。如果将这两个目标施加于同一个速度预测器,会产生梯度干扰——一个信号说"平均化",另一个说"专注模式"。本文的解决方案是将这两个目标完全解耦到两个独立的速度头中,但让它们共享同一个长上下文编码器。这样,编码器可以同时从两种信号中学习有用的长上下文表示,而每个头只接收与自己目标一致的梯度,避免了冲突。这是DDT架构在长视频生成场景的首次应用,也是首次将DMD风格的蒸馏扩展到滑动窗口级别的长视频对齐。

方法步骤详情

方法分为训练和推理两个阶段。训练阶段每步处理两个minibatch:(1) 真实长视频batch——从数据集采样长视频潜在表示 $x_0^{long}$,构造噪声版本 $x_t^{long} = (1-t)x_0^{long} + tz$,输入共享编码器 $E_\phi$ 得到特征 $h_t$,通过Flow Matching头 $D_{FM}^\theta$ 预测速度 $u_\theta$,计算SFT损失 $\mathcal{L}_{SFT}$ 更新编码器和FM头;(2) 学生rollout batch——用学生模型当前参数生成长视频,对每个滑动窗口 $\text{crop}_k$ 添加噪声,通过DM头 $D_{DM}^\psi$ 预测窗口速度 $v_\psi^{(k)}$,同时查询冻结的短视频教师获得教师速度 $u_{teacher}$,训练一个fake得分估计器 $v_{fake}$ 来估计学生分布得分,计算DMD梯度代理 $(v_{fake} - u_{teacher})^\top \nabla_{\hat{x}_0^{(k)}}$ 更新编码器和DM头。总损失为 $\mathcal{L}_{total} = \mathcal{L}_{SFT} + \lambda_{seg} \mathcal{L}_{seg}$。推理阶段丢弃FM头,仅用DM头 $v_\psi$ 端到端双向生成长视频,4步即可完成。

技术新颖性

本文的技术新颖性体现在三个层面:第一,首次在视频生成领域提出"时间维度不等价于空间分辨率"这一关键观察,从根本上质疑了混合长度训练的有效性;第二,将DDT(解耦扩散Transformer)架构从图像生成引入视频生成,并创新性地用它来分离均值寻觅和模式寻觅两种本质上冲突的目标,这在技术上比简单的多任务学习更优雅;第三,首次实现滑动窗口级别的DMD/VSD蒸馏应用于双向长视频生成——之前DMD主要应用于图像和短视频,本文通过巧妙的滑动窗口裁剪和VAE解码-重编码策略解决了窗口边界处图像/视频潜在语义不匹配的问题(从非零偏移位置裁剪的窗口开头是视频潜在而非图像潜在,需要重建),使得短视频教师可以仅通过局部窗口查询来指导长视频的每一帧。

方法总览:模式寻觅遇见均值寻觅
Figure 2: 方法总览:模式寻觅遇见均值寻觅

实验结果

本文在Wan 1.3B模型上进行了全面的定量和定性评估,测试集为200个描述长视频和事件的提示词,生成30秒视频,采用VBench-Long评估协议。核心发现如下:(1) 在仅4步推理(NFE=4)条件下,本文方法的综合一致性得分(Gemini-3-Pro评估)达到75.42,显著优于CausVid的39.30、Self-Forcing的37.60和InfinityRoPE的68.61,甚至接近需要50步推理的Long-context SFT(77.28)和Mixed-length SFT(74.63);(2) 在图像质量方面,本文取得0.5735(第二名),仅次于CausVid的0.6044,但CausVid的一致性得分仅为39.30(过饱和问题严重);(3) 在美学质量方面,本文以0.6982取得最佳,超越所有基线;(4) 在动态程度方面,本文以0.9453取得最佳,说明方法在保持长时序一致性的同时没有牺牲运动丰富性,而InfinityRoPE(0.7188)和CausVid(0.8594)的动态程度明显偏低;(5) 消融实验表明,去掉DDT双头设计导致一致性从0.9615降至0.9427、运动从0.9685降至0.9449、质量从0.6359降至0.5298,是最大的性能下降来源,验证了梯度冲突问题的严重性和解耦设计的必要性。

定量对比结果
Table 1: 定量对比结果
消融实验
Table 2: 消融实验
定性结果展示
Figure 3: 定性结果展示
与基线方法的定性对比
Figure 4: 与基线方法的定性对比
查看结构化数据
任务指标本文基线提升
长视频生成(30秒) 一致性 (Gemini-3-Pro) 75.42 CausVid: 39.30 / Self-Forcing: 37.60 / InfinityRoPE: 68.61 比InfinityRoPE提升6.81点,比CausVid提升36.12点
长视频生成(30秒) 美学质量 0.6982 CausVid: 0.6305 / InfinityRoPE: 0.6871 / MixSFT: 0.6683 超越所有基线,比MixSFT提升2.99%
长视频生成(30秒) 动态程度 0.9453 CausVid: 0.8594 / InfinityRoPE: 0.7188 / MixSFT: 0.8906 比MixSFT提升5.47%,比InfinityRoPE提升22.65%
长视频生成(30秒) 主体一致性 0.9682 CausVid: 0.9736 / InfinityRoPE: 0.9689 / MixSFT: 0.9667 与最优基线持平(差异在噪声范围内)
消融:去掉DDT双头 质量 0.6359(完整模型) 0.5298(去掉DDT) 双头设计提升质量10.61%

局限与改进

本文存在以下几个方面的局限性:首先,作者在附录中坦承,本文方法与因果自回归(causal AR)方法是正交的,当前实验仅在双向扩散模型上验证,尚未与最新的AR+扩散混合范式结合。其次,评估仅使用30秒视频,虽然论文标题强调"分钟级"生成,但定量结果未展示更长(如60秒以上)视频的质量衰减曲线。第三,所有实验基于Wan 1.3B模型,14B模型仅在补充网站展示定性结果,缺乏大规模模型的定量对比,难以评估方法在更大模型上的增益是否线性。第四,教师和学生使用同一架构(Wan 1.3B),当教师本身质量有限时,模式寻觅对齐可能会传递教师的缺陷。第五,滑动窗口DMD需要额外训练fake得分估计器(每5步学生更新一次),增加了训练复杂度和计算成本,论文未详细报告训练时间和资源消耗。

独立分析的弱点

从独立分析角度看,本文存在几个值得关注的弱点:(1) 评估不充分——200个提示词的测试集规模偏小,且仅评估30秒视频,对于声称"分钟级"生成能力的论文来说说服力不足。建议增加至少1000个提示词的测试,并报告60秒、120秒视频的质量衰减曲线。(2) 教师-学生同构限制——当前实验中教师和学生都是Wan 1.3B,实际应用中可能需要从更强的教师(如14B或专用短视频模型)向更轻的学生蒸馏。建议探索异构教师-学生组合。(3) 缺乏用户研究——VBench的自动指标和Gemini-3-Pro的评分未必与人类感知完全一致,特别是在运动自然性和叙事连贯性方面。建议增加人工评估。(4) 滑动窗口的粒度固定——当前窗口长度固定为约5秒,对于不同复杂度的视频,自适应窗口大小可能更优。(5) 训练效率未知——论文未报告完整训练的GPU小时数和收敛速度,难以评估实际可复现性。

未来方向

本文的未来研究方向可以从以下几个维度展开:第一,与因果自回归模型结合——作者在附录中明确提出这是"自然的后续工作",可以将本文训练好的双向长视频模型作为基础模型,进一步进行因果AR训练,或者将长上下文双向模型蒸馏为因果采样器(通过添加因果注意力掩码)。第二,与rollout鲁棒性技术结合——如Rolling Forcing、LongLive等方法可以进一步提升AR推理时的误差累积问题。第三,更长上下文外推——本文的编码器已具有长上下文能力,结合更长的位置编码方案(如InfinityRoPE的扩展版本)可能实现更长的生成。第四,交互式世界模型——长上下文编码器提供了一种持久的历史表示,可以在此基础上添加动作/交互条件,应用于具身智能和游戏世界建模。第五,探索更高效的DMD实现——当前每5步需要训练fake得分估计器,是否可以用更轻量的替代方案(如EMA教师)来简化训练流程。

复现评估

从复现角度看,本文具有一定可行性但存在几个挑战。开源方面:论文基于FastGen(NVIDIA开源仓库)实现,但未明确说明是否开源完整训练代码和预训练权重,这会显著影响复现难度。数据方面:训练数据来自Sekai数据集(公开)、MiraData子集(公开)和筛选的互联网视频,总计超过10万条10秒到分钟级视频,平均31秒,上限61秒——这些数据基本可获取,但互联网视频的筛选标准未详细说明。算力方面:训练在A100和GB200 GPU上进行,使用DeepSpeed Ulysses序列并行(A100组大小4,GB200组大小2),对于1.3B模型规模,预估需要数十到上百GPU天;14B模型的算力需求可能达到数百GPU天。关键依赖:Wan模型(开源)、DDT架构(需自行实现或从开源代码修改)、DMD/VSD梯度实现(技术细节较多)。总体复现难度为中等偏高,主要瓶颈在DMD滑动窗口实现的工程复杂性。