← 返回 2026-03-05

Helios:真正的实时长视频生成模型 Helios: Real Real-Time Long Video Generation Model

Shenghai Yuan, Yuanyang Yin, Zongjian Li, Xinwei Huang, Xiao Yang, Li Yuan 📅 2026-03-04 👍 190 2026-07-13 08:35
实时生成 扩散模型 模型蒸馏 视频生成 长视频

首个在单H100上以19.5 FPS运行的14B视频生成模型,支持分钟级长视频

前置知识

Diffusion Transformer (DiT)

扩散变换器是将Transformer架构与扩散模型结合的生成模型。在视频生成中,DiT通过迭代去噪过程从随机噪声生成视频帧。具体来说,模型学习预测噪声场的速度场(velocity field),通过ODE求解器逐步将噪声转化为清晰视频。DiT使用自注意力机制处理时空token,能够捕获长距离依赖关系,但计算复杂度随序列长度呈二次增长,这是实时生成的主要瓶颈。

Helios基于Wan-2.1-T2V-14B这个DiT模型构建,理解DiT的注意力机制和去噪过程是理解本文效率优化的基础

Flow Matching(流匹配)

流匹配是一种生成建模方法,定义了从噪声到数据的线性插值路径。给定噪声 $\epsilon$ 和干净数据 $x_0$,在时间步 $t$ 的插值为 $x_t = (1-\lambda_t)x_0 + \lambda_t\epsilon$,其中 $\lambda_t$ 控制噪声水平。模型学习预测这条路径上的速度场 $v = x_0 - \epsilon$。与DDPM不同,流匹配直接学习ODE而非随机微分方程,训练更稳定且采样更高效。

Helios的Deep Compression Flow和Pyramid Unified Predictor Corrector都基于流匹配框架,理解其数学形式对理解多尺度采样至关重要

Classifier-Free Guidance (CFG)

无分类器引导是扩散模型中控制生成质量的技术。在采样时,同时计算条件预测和无条件预测,然后通过公式 $\hat{\epsilon} = \epsilon_{uncond} + s \cdot (\epsilon_{cond} - \epsilon_{uncond})$ 进行引导,其中 $s$ 是引导强度。较大的 $s$ 值增强文本-视频对齐但可能降低多样性。标准方法需要两次前向传播(条件和无条件),计算开销翻倍。

Helios通过蒸馏消除了CFG需求,将采样步数从50步减少到3步,这是实现实时生成的关键

Distribution Matching Distillation (DMD)

分布匹配蒸馏是一种将多步扩散模型压缩为少步生成器的技术。核心思想是训练一个fake-score estimator $p_{fake}$ 来匹配teacher模型定义的分布,通过最小化real score和fake score之间的差异来更新生成器。与一致性蒸馏不同,DMD不需要逐步匹配,而是直接匹配最终分布,训练更稳定但需要额外的score estimator网络。

Helios的Adversarial Hierarchical Distillation基于DMD框架改进,理解DMD的训练流程是理解蒸馏创新的前提

RoPE(旋转位置编码)

旋转位置编码通过复数旋转将位置信息注入注意力机制。对于位置 $m$ 的token,其query和key向量通过旋转矩阵 $R_m$ 进行变换:$q_m R_m = q_m e^{im\theta}$。RoPE的周期性使得模型对相对位置敏感,但在长序列生成中,绝对位置索引可能导致模型在超出训练长度时出现漂移,因为它会回溯到早期位置。

Helios提出Relative RoPE来解决长视频生成中的位置漂移问题,这是理解抗漂移机制的关键

研究动机

当前视频生成模型面临三个核心困境。首先,主流模型只能生成5-10秒的短视频,即使这些短片也需要几十分钟才能合成。例如,使用Wan2.1 14B在单个A100 GPU上生成5秒视频大约需要50分钟。其次,现有的实时长视频方法主要基于1.3B小模型(如CausVid、Self-Forcing、Rolling Forcing等),这些模型容量有限,难以表示复杂运动,往往导致高频细节模糊。Krea-RealTime-14B虽然扩大了模型规模,但在单个H100上只能达到6.7 FPS。最关键的是,这些方法严重依赖Self-Forcing的train-as-infer rollout策略来缓解漂移,但这导致漂移鲁棒性与训练时的rollout长度紧密耦合:当训练限制在5秒片段时,推理超过5秒就会出现严重漂移。此外,基于因果掩码的方法从根本上改变了预训练双向模型的推理机制,可能限制可达到的质量上限。

本文的目标是本文的目标是构建第一个能在单个H100 GPU上以实时速度(大于等于15 FPS)运行的14B视频生成模型,同时支持分钟级别的长视频生成,并且在质量上匹配甚至超越现有的基础模型。具体而言,作者希望实现:(1) 在不使用self-forcing、error-banks、关键帧采样等传统抗漂移策略的情况下,生成分钟级高质连贯视频;(2) 在不使用KV-cache、稀疏/线性注意力、量化等标准加速技术的情况下,实现14B模型的实时推理;(3) 在不使用并行化或分片框架的情况下,以图像扩散模型级别的batch size训练14B模型。

与已有工作不同的是,本文的独特切入点在于对漂移现象的系统性分析和针对性解决。作者首次明确识别了长视频漂移的三种典型表现:位置漂移(Position Shift)、颜色漂移(Color Shift)和恢复漂移(Restoration Shift),并提出了简单而有效的训练策略来显式模拟漂移。更重要的是,作者发现传统的抗漂移策略(如Self-Forcing)本质上是在训练时模拟推理时的误差累积,但这种模拟是隐式的且与rollout长度耦合。Helios通过Frame-Aware Corrupt显式地在训练时注入各种扰动(曝光调整、噪声、下采样等),使模型学会处理不完美的历史上下文,从而解耦了漂移鲁棒性与训练时长。此外,作者发现RoPE的周期性与多头注意力的交互是重复运动的根源,通过Relative RoPE从源头解决这一问题。

核心方法

Helios的方法可以类比为一个视频续写专家:它不是从头生成整个视频,而是基于已有的历史片段续写未来帧,就像一个画家看着前面的画作继续创作。这种续写范式的关键优势在于,无论目标视频多长,模型处理的上下文长度始终固定,从而避免了计算复杂度随视频长度爆炸性增长。技术路线上,Helios采用三阶段渐进式训练:第一阶段(Base)将双向预训练模型转换为自回归生成器;第二阶段(Mid)引入多尺度采样进一步压缩计算;第三阶段(Distilled)通过对抗层次蒸馏将采样步数从50步减少到3步。整个系统通过Unified History Injection统一处理文本到视频(T2V)、图像到视频(I2V)和视频到视频(V2V)任务,通过Deep Compression Flow将计算成本压缩到与1.3B模型相当甚至更低的水平。

Helios最本质的创新在于显式漂移模拟而非隐式漂移修复。现有方法(如Self-Forcing)试图在训练时模拟推理过程,通过长rollout让模型看到自己生成的不完美输出,但这种方式计算开销巨大且效果与rollout长度绑定。Helios的核心洞察是:与其让模型自己犯错再学习修复,不如在训练时直接注入各种类型的扰动,让模型提前学会处理不完美的历史。具体来说,Frame-Aware Corrupt对每个历史帧独立地施加四种扰动之一(曝光调整、噪声注入、下采样模糊或保持干净),概率分别为 $p_c$、$p_a$、$p_b$、$p_d$,且 $p_a + p_b + p_c + p_d = 1$。这种设计的关键在于扰动是逐帧独立采样的,$T_{Hist}$ 个历史帧产生 $T_{Hist}$ 个独立的损坏决策,这对于长视频稳定性至关重要。另一个本质创新是Deep Compression Flow,它将流匹配从全分辨率噪声到全分辨率数据的单一轨迹重构为多个低分辨率噪声到多分辨率数据的轨迹,通过金字塔式多尺度采样将计算量大幅降低。

方法步骤详情

Helios的完整流程包含以下关键步骤:(1) 统一历史注入:将输入表示为历史上下文 $X_{Hist} \in \mathbb{R}^{B \times C \times T_{Hist} \times H \times W}$ 和噪声上下文 $X_{Noisy} \in \mathbb{R}^{B \times C \times T_{Noisy} \times H \times W}$ 的拼接,其中 $T_{Hist} \gg T_{Noisy}$。通过Representation Control,根据 $X_{Hist}$ 的表示自动切换任务:全零执行T2V,仅最后一帧非零执行I2V,否则执行V2V。(2) 引导注意力:在自注意力层中,对噪声上下文和历史上下文分别计算Q、K、V,并引入逐头放大token $amp$ 来调制历史key:$X_{Self} = Attention([Q_{Noisy}, Q_{Hist}], [K_{Noisy}, K_{Hist} \cdot amp], [V_{Noisy}, V_{Hist}])$。交叉注意力仅应用于噪声上下文,避免冗余语义注入。(3) 多术语记忆补丁化:将历史上下文分为短期(T1=16帧)、中期(T2=2帧)和长期(T3=2帧)三部分,分别使用不同大小的卷积核进行压缩,压缩比随时间距离增加:$p^{(1)}_t < p^{(2)}_t < p^{(3)}_t$。这使得总token数量 $L_{total}$ 保持恒定,无论目标视频长度。(4) 金字塔统一预测校正器:将采样分为K=3个阶段,在早期阶段使用低分辨率潜空间,逐步过渡到全分辨率。每个阶段学习多尺度速度场,通过线性插值路径 $x^k_t = (1-\lambda_t)x^k + \lambda_t Up(x^{k-1})$ 定义连续过渡。(5) 对抗层次蒸馏:基于DMD框架,采用纯教师强迫(仅使用真实数据作为历史上下文)、分阶段反向模拟(将反向模拟分解为K个阶段)、粗到细学习(通过课程式策略逐步增加任务难度)和对抗后训练(引入GAN目标突破教师容量限制)。

技术新颖性

Helios的技术新颖性体现在多个层面。首先,与CausVid和Self-Forcing使用因果掩码不同,Helios保持双向注意力并通过Guidance Attention显式分离历史和噪声上下文的处理。这避免了因果掩码对表示能力的限制,同时保留了预训练模型的推理机制。其次,现有的位置编码方案(如绝对RoPE)在长视频中会导致位置漂移,因为模型会回溯到训练时见过的位置索引。Helios的Relative RoPE将历史上下文的索引固定在 $0:T_{Hist}$,噪声上下文在 $T_{Hist}:T_{Hist}+T_{Noisy}$,无论目标视频长度如何,索引范围始终一致。第三,Multi-Term Memory Patchification不同于简单的下采样或关键帧选择,它根据时间距离采用层级式压缩:短期记忆保留细节(压缩比小),长期记忆保留全局上下文(压缩比大),在固定token预算下支持更长的历史。第四,Pyramid Unified Predictor Corrector是UniPC的多尺度变体,通过在早期采样阶段处理更少的token来降低计算量,同时保持生成质量。最后,Adversarial Hierarchical Distillation采用纯教师强迫策略,避免了Self-Forcing的长rollout开销,同时通过对抗后训练突破教师模型的性能上限。

Helios架构图
Figure 4: Helios架构图
多术语记忆补丁化减少开销的效果
Figure 7: 多术语记忆补丁化减少开销的效果
对抗层次蒸馏流程图
Figure 9: 对抗层次蒸馏流程图

实验结果

Helios在多个维度上取得了突破性成果。首先,在速度方面,Helios-Distilled在单个H100 GPU上达到19.53 FPS的端到端推理速度,这比同规模的FastVideo(5.37 FPS)快3.6倍,比TurboDiffusion(10.15 FPS)快1.9倍,甚至比参数量小7倍的SANA Video Long(13.24 FPS)还快1.5倍。更惊人的是,Helios实现了128倍的加速同时保持了可比的质量。在短视频生成(81帧)方面,Helios-Distilled获得总分6.00,在所有蒸馏模型中最高,与基础模型如Wan 2.1 14B(6.15)和HV Video(6.00)相当。特别值得注意的是Semantic(5分)和Naturalness(5分)指标,这两个指标最能反映人类感知质量,Helios在这些指标上匹配甚至超越了更大规模的基础模型。在长视频生成(120-1440帧)方面,Helios-Distilled获得总分7.08,超越了最强基线Reward Forcing(6.88)。在漂移指标上,Helios在Aesthetic(7分)、Semantic(7分)和Naturalness(7分)的漂移分数都表现出色,表明模型能更好地保持内容一致性和场景布局。用户研究进一步证实了这些结果:在200份有效问卷中,Helios在长视频和短视频生成上都一致性地超越了现有方法。消融实验揭示了各组件的重要性:移除First Frame Anchor导致总分从6.47降至5.51;移除Frame-Aware Corrupt导致总分降至4.70;使用因果掩码会导致训练不稳定。

各种视频生成模型在单个H100上的端到端吞吐量(FPS)
Figure 1: 各种视频生成模型在单个H100上的端到端吞吐量(FPS)
Helios及其对比模型的基准测试性能
Figure 2: Helios及其对比模型的基准测试性能
查看结构化数据
任务指标本文基线提升
短视频生成(81帧) 总分(加权平均) Helios-Distilled: 6.00 Wan 2.1 14B: 6.15, Krea: 5.95 超越所有蒸馏模型,匹配基础模型
长视频生成(120-1440帧) 总分(加权平均) Helios-Distilled: 7.08 Reward Forcing: 6.88, Rolling Forcing: 6.86 提升2.9%
推理速度 FPS(端到端) Helios-Distilled: 19.53 FastVideo: 5.37, TurboDiffusion: 10.15, Krea: 6.74 比FastVideo快3.6倍,比TurboDiffusion快1.9倍
漂移鲁棒性 Drifting Naturalness(10分制) Helios-Distilled: 7 Self-Forcing: 3, Krea: 1 比Self-Forcing提升133%

局限与改进

论文存在几个值得关注的局限性。首先,作者承认现有评估指标(如Aesthetic和Smoothness)与人类感知的相关性较弱,Helios生成的视频虽然更自然真实,但这些指标只能反映边际差异。这意味着目前的自动化评估可能无法充分展示Helios的优势。其次,与现有自回归模型类似,Helios在拼接边界处可能出现闪烁伪影,尽管作者声称该方法consistently outperforms previous approaches,但这一问题仍未完全解决。第三,由于资源限制,实验仅在384x640分辨率下进行,更高分辨率的设置尚未探索。这限制了我们对模型在高清场景下表现的理解。此外,作者提到缺乏专门设计用于长视频记忆的机制,这可能限制了模型在需要长期一致性的应用场景中的表现。从我的观察来看,Helios的训练需要128个H100 GPU(第三阶段),这对大多数研究团队来说是难以企及的算力门槛,可能限制该方法的广泛采用。

独立分析的弱点

基于对论文的深入分析,我识别出几个需要改进的方面。首先,分辨率限制是一个显著弱点:所有实验仅在384x640分辨率下进行,而实际应用(如视频编辑、内容创作)通常需要720p或1080p。改进方向包括探索渐进式分辨率训练策略,或在蒸馏阶段引入超分辨率模块。其次,拼接边界的闪烁问题仍然存在:虽然Helios优于现有方法,但在交互式生成中,用户修改提示时的过渡可能仍不平滑。可以通过引入时序一致性损失或优化prompt插值策略来改进。第三,评估体系的不完善:现有指标与人类感知对齐度差,Semantic和Naturalness指标的区分度有限。建议开发基于大型视觉-语言模型的评估方法,或构建更细粒度的人类评估数据集。第四,训练效率问题:第三阶段需要128个H100 GPU,总训练步数超过6000步。可以通过知识蒸馏的课程设计优化,或探索更高效的GAN训练策略来降低算力需求。最后,长视频记忆机制缺失:当前模型没有专门的长期记忆模块,可能在需要跨分钟级一致性的场景(如叙事视频)中表现不佳。

未来方向

基于Helios的成果,未来研究可以沿几个方向展开。首先,更高分辨率的实时生成:当前的Deep Compression Flow框架可以扩展到支持多分辨率输出,通过调整金字塔采样的阶段数和分辨率配置来适应不同应用场景。其次,交互式世界模型:Helios的Interactive Interpolation展示了实时控制的潜力,未来可以探索更精细的控制机制(如姿态控制、风格迁移),使其成为真正的交互式世界模型。第三,更高效的蒸馏方法:Adversarial Hierarchical Distillation可以进一步与一致性模型结合,探索2步甚至1步生成的可能性。第四,长视频记忆架构:受Transformer-XL和记忆网络启发,可以设计专门的长期记忆模块,存储和检索跨分钟级的视觉信息。第五,多模态扩展:当前框架专注于视觉生成,可以扩展到音频-视频联合生成,支持更丰富的多媒体内容创作。最后,开源生态建设:作者计划发布代码、基础模型和蒸馏模型,这将促进社区在实时长视频生成领域的进一步研究。

复现评估

从复现角度来看,Helios的复现难度中等偏高。有利因素包括:(1) 作者计划开源代码、基础模型和蒸馏模型;(2) 论文提供了详细的训练超参数(Tables 1和2),包括学习率、batch size、LoRA配置等;(3) 核心组件(如Guidance Attention、Frame-Aware Corrupt)的实现相对直接。不利因素包括:(1) 第三阶段训练需要128个H100 GPU,总训练时间未明确说明,但根据步数和batch size估算可能需要数周;(2) 训练数据(0.8M clips,时长小于10秒)的具体来源和预处理流程未详细说明;(3) 一些技术细节(如Flash Normalization和Flash RoPE的Triton实现)需要额外的GPU编程经验。对于资源有限的研究团队,建议从Helios-Base或Helios-Mid开始,在较小规模上验证核心思想,或使用作者提供的预训练模型进行下游任务适配。