← 返回 2026-06-15

RhymeFlow:通过异步去噪流调度实现免训练视频生成加速 RhymeFlow: Training-Free Acceleration for Video Generation with Asynchronous Denoising Flow Scheduling

Chensheng Dai, Shengjun Zhang, Yifan Li, Zhang Zhang, Zheng Zhu, Yueqi Duan 📅 2026-06-04 👍 11 2026-07-13 08:37
DiT优化 免训练加速 关键帧选择 异步去噪调度 扩散模型 视频生成加速

异步帧调度让关键帧走完去噪、非关键帧跳步,配合潜轨迹投影保时序一致性

前置知识

视频扩散 Transformer (Video DiT)

Video DiT 是把 2D Diffusion Transformer 扩展到 3D 时空维度的主干架构,对视频的潜空间帧序列执行时空联合注意力。Wan 2.1 在潜空间有 21 帧,每帧 3600 token,3D 注意力随 token 数呈 $\mathcal{O}(N^2)$ 二次方增长。

理解 Video DiT 的二次方复杂度是看懂 RhymeFlow 动机的关键:它解释了为什么 81 帧、720p 视频的一次完整去噪就要 993.5 秒。

Rectified Flow / Flow Matching

Rectified Flow 把扩散建模为常微分方程路径 $dz/dt = v_\theta(z,t)$,路径近似直线。本文用此特性对跳过的中间潜状态做线性插值 $\hat{z}_\tau = (1-\alpha) z_{start} + \alpha z_{end}$。

线性插值假设是 Latent Trajectory Projection 的理论基石——只有当跳步轨迹近似直线时,插值才能逼近真实中间状态而不引入严重误差。

3D 时空注意力

3D 时空注意力让每帧 token 同时关注同帧空间 token 和其他帧时间 token,注意力图呈 $\text{softmax}(QK^\top/\sqrt{d})V$ 全连接。Wan 2.1 中每帧 3600 token 全帧互相 attention。

理解时空注意力要求所有帧在同一时间步都处于已计算状态,是看懂为什么必须用投影填补跳步帧的根本原因。

KV-Cache 机制

KV-Cache 在推理时缓存 Key 和 Value 张量避免重复计算。本文缓存最近两步 post-attention hidden state $h_{t_1}, h_{t_2}$,投影生成的瞬时状态用完即丢弃,使显存峰值低于原稠密模型。

理解 KV-Cache 的设计取舍对评估 RhymeFlow 显存收益(峰值从 49.9 GB 降到 42.6 GB)至关重要。

余弦相似度选帧

通过 $\cos(\hat{z}_0^{(t)}, \hat{z}_0^{(k)})$ 衡量两帧潜码相似度,低于阈值 $\tau$ 即视为新关键帧。先单步去噪估计 $\hat{z}_0$ 再比较,可识别语义转折时刻。

这种基于潜空间语义相似度的选帧策略是 RhymeFlow 比均匀选帧 / 首帧选帧分别提升 SSIM 0.14 和 LPIPS 0.015 的核心算法。

免训练稀疏注意力

免训练稀疏注意力不修改权重,通过减少单步 token 数(SVG、SAP)或缓存中间结果(DeepCache、FasterCache)压缩算力。RhymeFlow 优化帧级时间维度调度,与这些方法正交可叠加。

理解 sparse attention 才能看懂 RhymeFlow 的 inter-step 正交性——这是论文 Table 2 中 Ours+SAP 在 Wan 2.1 上达到 1.66× 加速的关键。

研究动机

当前基于 Diffusion Transformer 的视频生成模型(Wan 2.1、CogVideoX-v1.5、HunyuanVideo 等)虽然效果惊艳,但推理成本极高:在单张 A800 上,Wan 2.1 生成 81 帧 720p 视频需要 993.5 秒,CogVideoX-v1.5 需 625.0 秒,瓶颈来自 3D 时空注意力的二次方复杂度以及数十次顺序去噪步。已有免训练加速主要走三条路线:KV-Cache(如 FasterCache 把相邻步的深层特征复用)、模型压缩(Q-Diffusion 量化到 W8A8 甚至 4-bit)以及稀疏注意力(SpargeAttn、SVG、SAP 通过空间 / 时间 / 语义掩码减少单步 token 数)。然而这些方法都默认遵守一个隐含约束——视频中每一帧都必须在全部去噪时间步上完成稠密、逐步的推理。本文作者通过观察发现这个约束本身是冗余的:相邻帧内容重叠、连续运动,相邻帧在潜空间的轨迹高度可预测,让非关键帧在某些时间步跳过完整推理并不会带来明显质量损失。

本文的目标是本文的核心目标是设计一个免训练加速框架,把视频序列中每一帧都做完整稠密去噪的同步调度范式打破。具体而言,要实现三件事:第一,提出基于潜空间语义相似度的序列式关键帧选择算法,自动识别视频中哪些帧承载语义转折、需要高保真生成;第二,构造异步去噪调度,让关键帧执行完整逐步去噪,而非关键帧在指定步数上跳过完整 DiT 前向;第三,引入一个轻量级的潜轨迹投影模块,在不调用 DiT 的前提下解析地估计跳过的中间潜状态,让关键帧在 3D 注意力中仍然能与完整时序对话,从而保持全局一致性。

与已有工作不同的是,本文的独特切入角度是开辟一个与现有方法正交的加速维度——inter-step 帧间异步而非 intra-step token 内稀疏。已有稀疏注意力方法(SpargeAttn、SVG、SAP、MInference、PAB)都还在单步内部优化 token-to-token 交互,DeepCache、FasterCache、EasyCache、DiCache 等缓存方法也只是复用单步内的特征。RhymeFlow 直接跳过某些帧在某些时间步的整个 DiT 前向,把每帧的去噪轨迹解耦并赋予异构步长,再通过线性插值投影填补缺失状态。这种按帧分配异构调度的设计在概念上完全不同于稀疏注意力(优化注意力的形状)和 KV-Cache(复用已有计算结果),且天然正交——论文中把 RhymeFlow 与 SAP 结合,在 Wan 2.1 上拿到 1.66× 加速(单 SAP 仅 1.63×),在 CogVideoX-v1.5 上拿到 1.93× 加速(单 SVG 仅 1.62×),验证了正交叠加性。

核心方法

RhymeFlow 一句话思路:让关键帧走完整去噪、非关键帧跳步走,用线性插值填补跳过的潜状态以维持时序一致性。技术分四步。第一步序列式关键帧选择:先一步去噪估计每帧伪干净态 $\hat{z}_0^{(i)}$,从首帧出发顺序遍历,若与最近关键帧的余弦相似度低于阈值 $\tau$ 则加入 $\mathcal{K}$。第二步异构异步调度:先 $T_w=8$ 步同步 warm-up;之后关键帧每步跑 DiT、非关键帧按渐进步长 $n_{skip}$ 一次前进多步($n_{small}=2$ 高噪声阶段、$n_{large}=3$ 低噪声阶段);轨迹交汇的 Rhythmic Points 做全局同步重对齐。第三步 Latent Trajectory Projection:用 Rectified Flow 线性假设 $\hat{z}_\tau^{(j)} = (1-\alpha) z_{t_{start}}^{(j)} + \alpha z_{t_{end}}^{(j)}$ 插值填补跳过的中间时刻,即时算 $K,V$ 给关键帧 attention。第四步 KV-Cache:每层为每个非关键帧维护仅含最近两步 hidden state 的滚动缓存,显存峰值从 49.9 GB 压到 42.6 GB。

RhymeFlow 的核心创新是把 inter-step 帧间异步作为一种新的加速维度引入视频扩散,与已有的 intra-step token 内稀疏和 intra-step 特征缓存正交。本质区别在于:已有方法都假设每一帧在每一个去噪步都必须被处理,优化的只是每步内部做什么或每步结果如何复用;RhymeFlow 直接打破这个假设,让某些帧在某些步完全不进 DiT。为了弥补时序上下文缺失,作者利用 Rectified Flow 的轨迹近似线性特性做了解析投影——这是一种用结构假设换算力的设计哲学,相比让非关键帧参与真实计算或干脆忽略它们的中间状态,投影在几乎零成本下保住了关键帧 attention 的完整时序视野。

方法步骤详情

RhymeFlow 推理流水线含七步。输入:长度 $N$ 的噪声潜帧 $\{z_T^{(i)}\}_{i=1}^N$、文本 prompt、Video DiT $\theta$。第一步同步 Warm-up:$t=T$ 到 $t=T-T_w$ 间对每帧执行 $z_{t-1}^{(i)} = \text{DiT}(z_t^{(i)}, \text{prompt})$,全帧共用 3D 注意力。第二步关键帧选择:用 $T_w$ 步内最后输出做单步去噪得 $\hat{z}_0^{(i)}$,从 $i=1$ 顺序遍历,若 $\cos(\hat{z}_0^{(i)}, \hat{z}_0^{(k)}) < \tau$ 则加入 $\mathcal{K}$。第三步渐进异步调度:$t > T_{mid}$ 时非关键帧每 $n_{small}=2$ 步跳一次;$t \le T_{mid}$ 时每 $n_{large}=3$ 步更新一次。第四步关键帧更新:非关键帧用 $\hat{z}_\tau^{(j)} = (1-\alpha) z_{t_{start}}^{(j)} + \alpha z_{t_{end}}^{(j)}$ 投影后再算 $K,V$。第五步 Rhythmic Points:轨迹交汇时全帧重做同步 3D 注意力重对齐。第六步 KV-Cache 维护每层滚动缓存。第七步循环至 $t=0$ 后 VAE 解码输出像素视频。

技术新颖性

本文的技术新颖性可以分四个层面理解。第一层是问题形式化层面:首次显式地把每帧每步同步去噪作为可破除的约束提出来,并给出关键帧走稠密、非关键帧跳步的替代范式,这是稀疏注意力社区长期默认但从未直接挑战的假设。第二层是算法层面:基于余弦相似度的序列式关键帧选择(Table 4 显示比均匀选帧 PSNR 高 2.0 dB、比随机选帧高 5.7 dB)+ 渐进式异步步长(高噪声阶段步长小、低噪声阶段步长大)结合 Rectified Flow 的线性假设做解析投影,三者耦合紧密。第三层是工程层面:post-attention rolling cache 设计把非关键帧中间状态从持久存储压缩为瞬时计算,使显存峰值低于原始稠密模型。第四层是范式层面:通过与 SAP 组合实现 1.66×(Wan 2.1)/1.93×(CogVideoX-v1.5)/2.60×(HunyuanVideo)的加速比,证明该方法可与所有主流稀疏注意力正交叠加,是首个系统化证明 inter-step 帧间调度可作为独立加速轴的工作。

Pipeline
Figure 2: Pipeline

实验结果

实验分四条主线。主线一(Wan 2.1 超参消融,Table 1):$T_w=8, M=4$ 达最优 PSNR 26.291、SSIM 0.783、SubCons. 0.8831、加速 1.53×;$M$ 从 3 增到 5 时 PSNR 升至 27.707 但加速降到 1.39×。主线二(SOTA 对比,Tables 2-3):Wan 2.1 上 PSNR 26.291 超 SVG 22.419、SAP 24.454,Ours+SAP 达 1.66×;CogVideoX-v1.5 上 1.78× 加速超 PAB 1.41×、SVG 1.62×、MInference 1.48×,SubCons. 0.986 几乎与稠密 0.987 持平;HunyuanVideo 上 PSNR 26.34、SSIM 0.918、LPIPS 0.060 全领先 EasyCache、DiCache、VGDFR,Ours+SAP 达 2.60×。主线三消融:Random/First/Uniform 选帧 PSNR 仅 20.630/19.220/24.293;去投影模块使 PSNR 从 26.291 跌至 20.630(-5.7 dB);KV-Cache 把峰值显存从 49.9 GB 压到 42.6 GB。主线四用户研究:82 人双盲中 Ours vs SVG/SAP 三维度均 p<.001 显著优于对手,vs Dense 三维度 p>.1 无显著差异。

Quantitative evaluation of the efficiency-fidelity trade-off on Wan 2.1
Table 1: Quantitative evaluation of the efficiency-fidelity trade-off on Wan 2.1
Quality and efficiency benchmarking on Wan 2.1 and CogVideoX-v1.5
Table 2: Quality and efficiency benchmarking on Wan 2.1 and CogVideoX-v1.5
Quality and efficiency benchmarking on HunyuanVideo
Table 3: Quality and efficiency benchmarking on HunyuanVideo
Ablation on key frame selection strategy
Table 4: Ablation on key frame selection strategy
Ablation study on Architectural Components
Table 5: Ablation study on Architectural Components
Ablation study on KV-Cache management
Table 6: Ablation study on KV-Cache management
Latent-state analysis against dense sampler on Wan 2.1
Table 7: Latent-state analysis against dense sampler on Wan 2.1
Double-blind user study results (%)
Table 8: Double-blind user study results (%)
Qualitative visual comparisons
Figure 3: Qualitative visual comparisons
查看结构化数据
任务指标本文基线提升
Wan 2.1 文本到视频 PSNR / SSIM / 加速比 PSNR 26.291 / SSIM 0.783 / 1.53× SAP: PSNR 24.454 / 1.63×;SVG: PSNR 22.419 / 1.40× PSNR 比 SAP +1.84 dB、比 SVG +3.87 dB;加速比 SVG +9.3%
CogVideoX-v1.5 文本到视频 PSNR / Subject Consistency / 加速比 PSNR 26.890 / SubCons. 0.986 / 1.78× SVG: PSNR 24.130 / 1.62×;PAB: 1.41× PSNR 比 SVG +2.76 dB;加速比 SVG +9.9%、PAB +26.2%
HunyuanVideo 高动态片段 PSNR / SSIM / 加速比 PSNR 26.34 / SSIM 0.918 / 2.26×(Ours+SAP: 2.60×) SAP: 2.52×;EasyCache: 2.33×;DiCache: 2.36× PSNR 比 SAP +1.70 dB、比 EasyCache +2.83 dB;最高加速 2.60×
Wan 2.1 关键帧策略消融 (Table 4) PSNR / SSIM PSNR 26.291 / SSIM 0.783(基于语义相似度) Random: 20.630/0.525;First: 19.220/0.515;Uniform: 24.293/0.643 比 Random +5.66 dB SSIM,比 Uniform +2.00 dB / SSIM +0.140
Wan 2.1 模块消融 (Table 5) PSNR / 加速比 完整 RhymeFlow: PSNR 26.291 / 1.53× w/o Progressive: 25.399/1.40×;w/o Projection: 20.630/1.60× 投影模块贡献 PSNR +5.66 dB;渐进调度贡献加速 +9.3%
Wan 2.1 显存管理 (Table 6) Peak Memory / Average Memory 42.6 GB / 21.5 GB w/o KV-Cache: 49.9 GB / 27.1 GB;Original: 44.3 GB / 24.4 GB 峰值显存比无缓存 -7.3 GB,比原始稠密模型还低 1.7 GB
用户研究 Ours vs Dense (Table 8) p-value 双侧检验 所有三个维度 p>0.1(统计无显著差异) 稠密基线 RhymeFlow 与稠密基线视觉无显著差异,但加速 1.53×

局限与改进

本文局限性可分五点。第一,关键帧选择依赖余弦相似度阈值 $\tau$,对快速变化的高动态场景并不鲁棒——Figure 4 显示非关键帧潜状态误差在剧烈运动处明显高于关键帧,Latent Trajectory Projection 的线性假设在剧烈运动处可能失效。第二,论文承认线性插值假设在复杂场景下会产生误差(Table 7 中 w/o 投影时 $E_{projected-nonkeyframe}$ 升至 0.0240),并未给出非线性投影方案。第三,$n_{small}=2, n_{large}=3$ 是手动设计的分段常数函数,对不同 DiT、不同分辨率未必最优,作者也在 Conclusion 提到用学习到的连续调度函数是未来方向。第四,所有实验仅在单张 A800、720p 81 帧上进行,对更长视频、更高分辨率或消费级 GPU 尚未验证可扩展性。第五,Ours+SAP 组合需把 SAP 稀疏率从 0.3 放宽到 0.5,叠加并非无条件——若 SAP 已被稀疏到极限,RhymeFlow 的叠加空间会受限。

独立分析的弱点

独立分析,RhymeFlow 有四个可改进的弱点。第一,$T_w=8$ 步 warm-up 是硬开销,可探索 warm-up 异步化或轻量早停。第二,关键帧选择仅看相邻关键帧余弦相似度,没利用 motion magnitude 或光流信息,对强运动场景可换成 motion-aware score。第三,Projection 假设 ODE 轨迹线性,但 Rectified Flow 只是近似线性,$n_{skip}=3$、低噪声阶段非关键帧可能经历大幅结构变化;可训练 hypernetwork 预测非线性残差 $\delta_\tau^{(j)}$ 让 $\hat{z}_\tau^{(j)} = (1-\alpha) z_{start} + \alpha z_{end} + \delta_\tau^{(j)}$,或用 KL 监控发散超阈值则动态调小 $n_{skip}$。第四,KV-Cache 只缓存最近两步 hidden state,$n_{skip} > 2$ 时部分中间 $\tau$ 离缓存步太远,可改成缓存三步。第五,用户研究只对比 SVG、SAP、Dense,未与 EasyCache、DiCache、FasterCache、TeaCache 等缓存类方法直接对比。

未来方向

作者明确提出用学习到的连续调度函数替代手动设计的渐进步长,可延伸出多个方向。第一,可把 $(T_w, n_{small}, n_{large}, T_{mid}, \tau)$ 五个超参联合建模为 RL 策略,让 agent 在每段视频上自动决定配置,用下游 VBench 分数作 reward。第二,把关键帧选择从静态余弦相似度升级为在线 learned predictor,输入 $\hat{z}_0^{(i)}$ 拼接时间位置 embedding,输出该帧作为关键帧的概率。第三,把 Latent Trajectory Projection 推广到非线性形式,用 Rectified Flow 模型自身作 predictor 做 few-step rollout 得到更精确的中间状态。第四,把 inter-step 帧间异步扩展到 inter-chunk 时序分块——把整个视频分成长度为 4-8 帧的块,块内异步、块间同步,进一步压低长视频延迟。第五,把异步调度与 token-level sparse attention 做端到端联合 NAS 搜索。第六,把方法推广到图像扩散和多视角视频生成,验证 inter-step 范式的通用性。

复现评估

本文可复现性整体良好。作者开源了项目页 https://simon-dcs.github.io/Website-of-RhymeFlow/ 和 GitHub 仓库 https://github.com/Simon-Dcs/RhymeFlow,且所有核心超参都给出默认值($T_w=8, M=4, n_{small}=2, n_{large}=3$)。实验基于两个主流开源模型 Wan2.1-T2V-v1.3B-Diffusers 与 CogVideoX-v1.5-T2V 以及 HunyuanVideo,无私有权重依赖。评估使用 Penguin Benchmark + VBench 评测协议,PSNR/SSIM/LPIPS/SubCons./ImgQual. 全是公开标准。算力方面单次评估在 A800 跑 Wan 2.1 约需 16-17 分钟,整体复现估计需 4-8 张 A800 × 3-5 天。复现难度中等偏高(涉及 DiT 内部 attention hook 修改和 KV-Cache 替换)。潜在不确定点:余弦相似度阈值 $\tau$ 和 $T_{mid}$ 具体小数位论文未完全给到,需查代码;Rectified Flow 轨迹线性度对不同 DiT 不一定一致,复现者可能需微调 $\tau$。