← 返回 2026-10-16

面向高效自回归视频生成的投机强制 Speculative Forcing for Efficient Autoregressive Video Generation

Hyunwook Choi, Seungryong Kim, Dongyoon Han, Sangdoo Yun 📅 2026-10-15 👍 78 2026-11-11 18:30
世界模型 少步蒸馏 扩散加速 投机解码 自回归扩散 视频生成

把中间去噪估计当草稿逐token提前验收,质量不降前提下将自回归视频生成提速至1.6倍

前置知识

自回归视频扩散生成

将视频切成若干潜在块(chunk),按 $p(c_1,c_2,\ldots)=\prod_t p(c_t\mid c_{<t})$ 逐块生成;每块从高斯噪声出发经 $K$ 步去噪得到干净潜在表示,并以历史块为条件。历史 token 的各层键值(KV)被缓存,避免每个新块重复计算历史特征。

本文加速的正是这类模型的去噪循环,理解 chunk、token、KV 缓存的运作方式,才能明白提前验收 token 到底省掉了什么计算、又靠什么保持上下文连贯。

少步蒸馏(DMD 与 Self Forcing)

分布匹配蒸馏(DMD)把多步教师模型的输出分布蒸馏进学生网络,使其仅需 $K=4$ 步即可生成;Self Forcing 进一步在自回归 rollout 上训练,并随机抽取去噪出口步施加分布匹配目标,因此每一步的中间干净估计 $\hat{x}^{(k)}_0$ 都得到监督,本身就是合理的视频输出。

本文方法成立的前提正是中间估计可当草稿用:若中间估计未经监督,提前接受会导致质量崩塌。理解这一点才能明白为什么该加速只对 Self Forcing 系生成器有效。

投机解码(Speculative Decoding)

大语言模型推理加速范式:小模型快速产出草稿 token,大模型一次性并行验证,只接受与目标分布一致的部分,从而保证输出分布严格不变。名字里的投机即指先赌一把再验证。

本文标题与方法直接借用草稿—验证框架,但验证目标从分布不变换成了收敛预测,接受是有损的。理解原范式有助于把握这篇论文借了什么、改了什么。

KV 缓存与注意力汇(attention sink)

流式生成时缓存已生成 token 各层的 key/value,新 token 只需计算自身查询;attention sink 指把最早若干潜在帧固定保留为锚点,配合局部注意力窗口,在有限显存下维持长视频生成的稳定性。

本文被接受的 token 不再参与去噪,但其 KV 保留在缓存中供后续 token 注意,且缓存中混有不同去噪步的状态;这是方法正确性与工程实现的核心细节。

VBench 与 Drift 指标

VBench 是视频生成质量评测体系,含主体/背景一致性、运动平滑度、动态程度、成像质量等 16 个维度,并给出 Total/Quality/Semantic 聚合分;Drift 沿用 Reward Forcing 的定义,把成像质量(MUSIQ)分数按 2 秒非重叠分段求标准差再对视频取平均,衡量画质随时间的漂移。

论文所有不降质的结论都建立在这两套指标上;此外还用 RPE(相机位姿相对误差)评估世界模型的相机跟随精度,读懂指标定义才能判断结论的说服力。

研究动机

现有少步自回归视频生成器(如 LongLive 与 Lingbot-World 2.0,均为 1.3B 参数、4 步蒸馏模型)对所有 token 一视同仁地跑满全部去噪步:即使某个区域的估计在第 1 步就与最终输出几乎一致,它仍要完整走完 4 步。这带来直接的算力浪费:在单张 H100 上,5 秒视频吞吐只有约 26.4 fps,30 秒视频约 24.5 fps,镜头不断变化的世界模型 rollout 更低至 18.1 fps。作者的关键观察(Figure 2)是 token 间收敛速度差异巨大:与历史相似的背景和物体区域常在最初几步就逼近 4 步最终输出,而新出现的结构、精细细节、世界模型中新揭示的区域要持续变化到最后一步。已有加速方法各有不足:KV/特征缓存只复用计算、不减少步数;RAS、HSA 等空间自适应采样虽按区域分配计算,但每个 token 仍被带到噪声调度终点,从不宣告完成,误差须靠周期性全量更新兜底;DSA、Surprise Forcing 以整帧或整块为粒度分配去噪深度,无法区分同一块内已定型与仍在变化的 token。

本文的目标是本文要系统回答一个问题:是否每个 token 都需要去噪到最后一步?目标是构建一个不改动生成器权重、可即插即用的推理框架:在 4 步去噪过程中逐 token 预测当前中间估计与其满步输出的差距,一旦判定已收敛便将其冻结为最终输出、退出后续所有去噪前向,仅对仍在变化的 token 继续精修。量化目标是在短视频、长视频、可控世界模型三类场景下,把平均去噪步数从固定的 4.00 降到约 1.5 至 2.5,同时使 VBench 各质量维度、画质漂移指标 Drift、相机跟随精度 RPE 均不劣于 4 步基线,并在单张 H100 上兑现 1.35 至 1.65 倍的实测吞吐提升。

与已有工作不同的是,本文的独特切入点有三。其一,把 Self Forcing 式蒸馏的结构性质转化为加速机会:这类训练随机抽取去噪出口步施加分布匹配目标,使每一步的中间干净估计都受到监督、本身就是可用的生成结果,因此天然可以充当草稿(draft),这是普通蒸馏范式不具备的性质。其二,借来 LLM 投机解码的草稿—验证框架但替换其目标函数:投机解码要求接受后输出分布严格不变,而本文验证器预测的是剩余收敛间隙,接受意味着容忍与满步输出的微小偏差,用可控的有损换取大幅计算节省,实验证明这种有损在感知上中性。其三,验收粒度做到 token 级:同一块内已定型区域第 1 步就退出、畸形区域精修到第 4 步;被接受的 token 输出冻结但 KV 保留供后续注意,缓存中不同去噪步的状态允许共存,块末一次干净前向统一刷新。

核心方法

直觉是去噪算力应按需分配:已收敛的 token 继续算纯属浪费。技术路线是在标准 4 步去噪循环中嵌入逐验证机制,形成草稿—验证—提交的循环。具体地,每步 $k<K$ 时验证器对仍活跃(未被接受)的 token 打分,判断其草稿 $\hat{x}^{(k)}_0$ 与满步输出的距离是否已足够小;通过者被提交为最终输出并从后续去噪前向中剔除,未通过者重新加噪到下一噪声级继续精修;到第 $K$ 步仍活跃的 token 全部提交。被接受的 token 不再参与去噪,但其各层键值保留在缓存中供活跃 token 注意,保持上下文完整;每块生成完毕后,生成器对拼装结果在 $t=0$ 做一次干净前向,统一刷新 KV 缓存。由此每块的去噪 token 评估数从 $KN$ 降到 $\sum_i k^*(i)$($k^*(i)$ 为 token $i$ 的提交步),另加 $N$ 次干净前向开销,平均去噪深度 $\bar{k}=\frac{1}{N}\sum_i k^*(i)$ 成为效率的核心度量。

核心创新是把何时可以停止去噪形式化为可预测的判别问题。作者定义收敛间隙 oracle:$\text{rel}_k(i)=\frac{\|\hat{x}^{(k)}_0(i)-\hat{x}^{(K)}_0(i)\|_2}{\sigma_t}$,其中 $\hat{x}^{(K)}_0$ 是同噪声同条件下满步输出的参考,$\sigma_t$ 为该块满步输出的 token 范数中位数;$\text{rel}_k(i)\le\epsilon$ 即视为收敛。推理时满步参考不可得,需验证器提前预测该间隙,论文给出两条互补路线。免训练验证器用两个代理信号:第 1 步用冗余度 $s_{\text{red}}(i)=\min_{h\in H}\|\hat{x}^{(1)}_0(i)-h\|_2$,即与注意力窗口内干净历史及参考帧(含 attention sink)中最相似 token 的距离;第 2 步起用稳定性 $s^{(k)}_{\text{stab}}(i)=\|\hat{x}^{(k)}_0(i)-\hat{x}^{(k-1)}_0(i)\|_2$。学习式验证器则从冻结生成器最后一块 Transformer 的隐状态直接回归间隙 $\hat{r}_k(i)$,证明中间特征本身就编码了这个预测还会变多少的信息。所有分数先经时空平滑算子 $\mathcal{S}$ 再阈值化,保证验收决策在空间与时间上连贯。

方法步骤详情

(1)初始化:块内 $N$ 个 token 全部活跃,第 1 步去噪产出每个 token 的草稿 $\hat{x}^{(1)}_0$。(2)验证:免训练路线在 $k=1$ 算 $s_{\text{red}}$、$k\ge2$ 算 $s_{\text{stab}}$,阈值 $(\tau_{\text{red}},\tau_{\text{stab}})$ 在留出数据上按 oracle 准则校准;学习路线用 4.11M 参数卷积头(LayerNorm 加三个 3D 卷积,通道 $1536\to96\to48\to1$,前两层 $3\times3\times3$ 带 GELU、末层 $1\times1\times1$,跨步共享,输出经 $2\times2\times2$ 平均池化再最近邻上采样回原网格)从最后 Transformer 块隐状态预测 $\hat{r}_k(i)$。(3)提交:分数经平滑后与 $\epsilon$ 比较,达标 token 的输出固定为 $\hat{x}^{(k)}_0(i)$、当步各层 KV 写入缓存,后续去噪不再计算它;未达标 token 重加噪到下一噪声级续跑。(4)收尾:$k=K$ 时剩余 token 全部提交,块输出 $c_t(i)=\hat{x}^{(k^*(i))}_0(i)$ 混有多步估计,随后在 $t=0$ 做干净前向统一刷新 KV。(5)训练验证头:冻结生成器跑满 $K$ 步 rollout,收集各步隐状态与 oracle 间隙,用 Huber 损失 $\mathcal{L}=\frac{1}{N(K-1)}\sum_{i,k}\text{smooth}_{\ell_1}(\hat{r}_k(i),\text{rel}_k(i))$ 训练,Adam、学习率 $10^{-3}$、批量 4;LongLive 用 1K 条 VidProM 提示词,Lingbot 用 1K 张不相交场景图像。

技术新颖性

与已有方法的本质区别:相比 DeepCache、ToCa 等 KV/特征缓存方法,本文不是复用计算而是提前终止计算,且被终止的 token 状态被转正为最终输出与后续上下文;相比 RAS、HSA 的空间自适应采样,那些方法仍把每个 token 推到调度终点、靠周期性全量更新约束误差,而本文显式宣告 token 完成,因而在这类 4 步蒸馏的极短调度下依然有效——论文指出替换式缓存正因短调度下无步可替而失效;相比 DSA、Surprise Forcing 的帧/块级步数分配,token 级验收能在同一块内同时保留已定型区域并继续精修畸形区域,消融显示相同平均步数下块级验收 Quality 82.54 对 82.65、Drift 3.214 对 2.839;相比 LLM 投机解码,本文不追求分布无损,草稿来自生成器自身的中间估计而非外部小模型,验证信号是从隐状态回归的收敛间隙(AUROC 0.87 至 0.94)而非输出分布比对。工程上还有一个新颖点:缓存允许不同去噪步的 KV 共存,用块末干净前向统一刷新,兼顾早提交与上下文质量。

Speculative Forcing. Baseline runs every token through the full denoising schedule. Speculative Forcing instead treats each token's intermediate clean estimate as a draft and asks a verifier to predict its remaining gap relk(i) to the full-step output. Tokens predicted to have converged are accepted and committed, after which they enter the next step only as keys and values, so the following denoising pass is computed for the rejected tokens alone.
Figure 3: Speculative Forcing. Baseline runs every token through the full denoising schedule. Speculative Forcing instead treats each token's intermediate clean estimate as a draft and asks a verifier to predict its remaining gap relk(i) to the full-step output. Tokens predicted to have converged are accepted and committed, after which they enter the next step only as keys and values, so the following denoising pass is computed for the rejected tokens alone.

实验结果

实验覆盖三类场景,生成器冻结,单张 H100 测吞吐。短视频(LongLive 5 秒、VBench 16 维):4 步基线 Total 83.08、26.40 fps;OursH 平均 1.68 步达 Total 83.21、41.74 fps(1.58×),OursTF 1.65 步、43.61 fps(1.65×),均超 2/3 步基线。长视频(30 秒、VBench-Long):OursTF/OursH 用 1.71/1.72 步跑到 38.8/37.8 fps(约 1.5×),Total 82.75/82.81 不低于基线 82.69,Drift 1.685/1.707 低于 1.722,动态程度保持约 41,说明画质变稳并非靠压制运动。世界模型(Lingbot-World 2.0,200 场景、每段 253 帧):镜头运动使任务更难,只降到 2.53/2.55 步,OursH 以 24.5 fps(1.35×)保住 Quality 82.65(基线 82.62),Drift 2.839 显著低于 3.094,RPE 平移/旋转 0.328/0.260 与基线 0.328/0.253 相当。消融:容差扫描显示 $\epsilon\in[0.20,0.24]$ 是甜点区(Quality 82.72/82.65 且 Drift 最低),$\epsilon=0.32$ 跌至 82.07;路由策略对比中通道方差信号崩至 78.23、时间差分 80.97、块级验收 82.54,均逊于 OursH 的 82.65,后者几乎追平 oracle 验收的 82.61;学习头验证精度 AUROC 0.87 至 0.94,全面领先冗余度、稳定性与运动信号;对比 Surprise Forcing(5 秒需 3.59 步、27.38 fps、Total 83.11),OursH 用 1.68 步、41.74 fps 拿到 83.21。整体来看,免训练与学习式两条验证路线表现接近,说明收敛信号本身易于获取,学习头的主要价值在世界生成这类高畸变场景下更稳。

Comparison on 5s videos with LongLive. Mean Steps is the measured average number of denoising passes per token. Standard VBench 16-dimension protocol, throughput on a single H100.
Table 1: Comparison on 5s videos with LongLive. Mean Steps is the measured average number of denoising passes per token. Standard VBench 16-dimension protocol, throughput on a single H100.
Comparison on 30s long videos with LongLive. Evaluated on VBench-Long.
Table 2: Comparison on 30s long videos with LongLive. Evaluated on VBench-Long.
World-model comparison on dynamic videos. Results with the autoregressive Lingbot-World 2.0, evaluated on VBench. RPE is the relative pose error of the camera trajectory. Quality Score aggregates the six dimensions reported here.
Table 3: World-model comparison on dynamic videos. Results with the autoregressive Lingbot-World 2.0, evaluated on VBench. RPE is the relative pose error of the camera trajectory. Quality Score aggregates the six dimensions reported here.
One test-time threshold sweeps speed against quality. At ϵ ∈ [0.20, 0.24] our sampler exceeds the uniform 4-step baseline on quality metrics. Drift is lowest in this range, not at the full budget.
Table 4: One test-time threshold sweeps speed against quality. At ϵ ∈ [0.20, 0.24] our sampler exceeds the uniform 4-step baseline on quality metrics. Drift is lowest in this range, not at the full budget.
Qualitative comparisons on video generation and world modeling. Both our head and training-free verifiers preserve visual quality comparable to the four-step baseline without visibly increasing quality drift over the illustrated rollouts on our benchmark.
Figure 4: Qualitative comparisons on video generation and world modeling. Both our head and training-free verifiers preserve visual quality comparable to the four-step baseline without visibly increasing quality drift over the illustrated rollouts on our benchmark.
Qualitative ablations of verification and commitment granularity. Our learned verifier handles distortions that challenge training-free verification. Premature chunk-wise acceptance propagates errors to later chunks, while motion-based verification struggles to distinguish camera-induced changes from distortions.
Figure 5: Qualitative ablations of verification and commitment granularity. Our learned verifier handles distortions that challenge training-free verification. Premature chunk-wise acceptance propagates errors to later chunks, while motion-based verification struggles to distinguish camera-induced changes from distortions.
查看结构化数据
任务指标本文基线提升
5 秒短视频生成(LongLive,VBench 16 维) VBench Total 分数 / 平均去噪步数 / H100 吞吐 OursH:1.68 步,41.74 fps,Total 83.21;OursTF:1.65 步,43.61 fps,Total 83.12 4 步均匀采样:4.00 步,26.40 fps,Total 83.08 平均步数 -58%,吞吐 1.58×(OursTF 达 1.65×),Total 反而 +0.13
30 秒长视频生成(LongLive,VBench-Long) Total 分数 / Drift 画质漂移 / 吞吐 OursTF:1.71 步,38.8 fps,Total 82.75,Drift 1.685;OursH:1.72 步,37.8 fps,82.81,Drift 1.707 4 步:4.00 步,24.5 fps,Total 82.69,Drift 1.722 吞吐约 1.54×,Total +0.06 至 +0.12,Drift 更低(长时程不累积误差)
可控世界模型 rollout(Lingbot-World 2.0,200 场景 × 253 帧) VBench Quality(6 维聚合)/ RPE 相机位姿误差 / Drift / 吞吐 OursH:2.55 步,24.5 fps,Quality 82.65,RPE 平移/旋转 0.328/0.260,Drift 2.839 4 步:4.00 步,18.1 fps,Quality 82.62,RPE 0.328/0.253,Drift 3.094 吞吐 1.35×,Quality 持平,Drift -0.255,相机跟随精度不受损
与 Surprise Forcing(帧/块级步数分配)对比(5 秒 LongLive) 平均步数 / 吞吐 / Total 分数 OursH(ε=0.28):1.68 步,41.74 fps,Total 83.21 Surprise Forcing:3.59 步,27.38 fps,Total 83.11 步数少 53%、吞吐高 52%,质量还略高
验证信号准确率(LongLive,识别需继续精修的 token) AUROC(按去噪步 k 与容差 ε 分解) OursH 在 ε=0.24 下 k=1/2/3 分别 0.934/0.890/0.906 冗余度 0.828/0.692/0.611,稳定性(k≥2)0.774/0.854,运动信号 0.701/0.592/0.561 第 1 步领先冗余度 +0.106,各步全面领先,说明隐状态确实编码收敛信息
学习验证器 vs oracle 验收(Lingbot 世界生成) Quality 分数 / Drift OursH:Quality 82.65,Drift 2.839(2.55 步) Oracle(推理时真跑满 4 步算真实残差):Quality 82.61,Drift 2.903(2.46 步) 不用任何满步参考即几乎追平 oracle 上界,Quality 还高 0.04

局限与改进

作者承认两点局限。其一,加速幅度依赖底层生成器对减步的敏感度和领域难度:世界生成中镜头运动不断引入新内容与较强畸变,平均步数只能降到约 2.5 而非 1.7;若换一个对步数更敏感的骨干网,可提前接受的 token 更少,无法保证同等加速。其二,块末干净前向是固定开销,理想加速比为 $5/(\bar{k}+1)$(如 $\bar{k}=1.65$ 时约 1.89×),且每块要执行到最后一个 token 被接受为止,实际吞吐受块内 $\max_i k^*(i)$ 而非均值 $\bar{k}$ 制约,因此实测 1.35 至 1.65 倍低于理想值。我自己的观察:质量保持是以生成器自身 4 步输出为参照定义的,4 步基线自身的缺陷会被完整继承,方法并不逼近多步教师;评测全部依赖 VBench 类自动指标,没有人类感知评测,30 秒/253 帧之外更长 rollout 中早期接受误差是否累积缺乏验证;免训练验证器的两个阈值需按场景校准,冗余度假设在全新内容占主导时会失效;实验仅限 1.3B 模型,向更大规模模型的迁移未验证。

独立分析的弱点

具体弱点与改进方向:(1)掉队者问题——每块必须执行到 $\max_i k^*(i)$,个别顽固不收敛的 token 会拖累整块吞吐;可引入跨块异步调度,或对迟迟不收敛的 token 动态降级(用外推估计提前提交并接受其误差)。(2)干净前向的固定开销本可消除:作者自己指出若像 CausalRCM 那样复用带噪 KV,理想加速比可逼近 $4/\bar{k}$,该技术与本文正交、可以叠加。(3)学习验证器按生成器/领域分别训练(LongLive 用提示词、Lingbot 用图像,各 1K),跨模型泛化能力未知;可探索多模型联合训练通用验证头,或把免训练信号蒸馏进单一头。(4)验收决策完全在潜空间做出,无像素域监督,理论上可能出现局部纹理伪影(作者用时空平滑缓解,但无硬保证);可加入轻量解码域或频域判别信号。(5)容差 $\epsilon$ 全局统一,但同一块内不同区域理应有不同容忍度(天空与人脸不该同权);区域条件化或按内容自适应的容差是自然延伸。

未来方向

作者提出的方向:把 token 级验收用于条件强约束的场景——说话人头像生成(背景基本静止、运动集中于面部与上身)、以真实场景为底座的环境接地世界模型(如 Seoul World Model、World Observer,其中扭曲的观测与参考图已为大部分视口提供视觉证据、仅新覆盖区域需真正生成)、提供静态场景渲染的灵巧世界模型,以及结合已知环境与生成内容的 AR/VR 应用。基于本文成果可延伸的:与带噪 KV 复用结合消掉干净前向、把理想加速推向 $4/\bar{k}$;把预测剩余间隙推广为联合预测最优退出步与噪声级,统一步数分配与 token 验收两套机制;在更大规模(远超 1.3B)模型和分钟级 rollout 上检验误差累积;为交互式世界模型设计动作条件化的验证信号——动作改变的区域天然应获得更多步数,这与本文已见区域早收的发现正好互补。

复现评估

复现条件较好。基座 LongLive 与 Lingbot-World 2.0(Causal-Fast,均 1.3B)是公开权重;本方法只训练 4.11M 参数的验证头,生成器全程冻结,1K 条 VidProM 提示词(视频)或 1K 张不相交场景图像(世界模型)即可生成训练 rollout,Adam(学习率 $10^{-3}$、批量 4)训练,成本很低。附录 A 完整给出超参:时间步 $[1000.0, 937.5, 833.3, 625.0]$、$\epsilon$(短视频 0.28、长视频/世界 0.24)、免训练阈值 $(\tau_{\text{red}},\tau_{\text{stab}})$ 三组取值(5 秒视频为 0.497/0.327)、平滑核(9×9、$\sigma=1.0$)等;附录 B 给出评测细节:VBench/VBench-Long 协议、Drift 用 MUSIQ 按 2 秒分段、RPE 用 Depth Anything 3 恢复位姿并做 Sim3 Umeyama 对齐、世界基准由 Qwen2.5-7B-Instruct 写描述、Z-Image-Turbo 生成 1376×768 初始图。推理在单张 H100 上完成,训练一张 H100 应可胜任;项目页 speculative-forcing.github.io 提供补充材料,正文未明确承诺开源代码。综合看复现难度中等偏低,主要工作量在共享索引、原地 KV 更新与干净前向合并等工程优化。

相似论文推荐