arXiv cs.LG精选83

Video DeltaNet:面向视频原生的混合注意力机制,用于直播视频生成

Video DeltaNet: A Video-Native Hybrid Attention for Livestream Video Generation

· 9天前

精选理由

提出视频原生混合注意力,兼顾长程时空建模与生成质量,实现14.5倍加速

AI 摘要

arXiv:2609.20744v3 公告类型:替换 摘要:视频扩散模型在去噪过程中会反复处理长时空令牌序列,使注意力机制成为主要的计算瓶颈。线性注意力提供了颇具吸引力的替代方案,并已广泛应用于近期的大型语言模型中,但将其直接应用于视频模型,往往无法保留高质量生成所需的细粒度交互。我们提出 Video DeltaNet(VDN),将局部 Softmax 注意力与双向线性记忆相结合,以建模长程视频上下文。其线性分支引入了 Video Delta Attention(VDA),通过联合整合每一帧的空间令牌,每帧更新一次记忆。独立的输出投影和可学习门控机制用于校准两个分支,同时,分阶段的教师对齐方案逐步将这一新路径引入预训练模型。我们在 MiniMax H3 上实例化了 VDN,将这种混合机制应用于视频到视频的交互,同时保留 Softmax 以处理涉及文本或音频的交互。借助八步蒸馏和优化后的 SGLang 服务栈,VDN-H3 在八块 NVIDIA B200 GPU 上,仅用 6.70 秒即可完成 14.3 秒、768p 视频的 DiT 去噪;相比相同 GPU 数量下采用 50 步密集 H3 基线,实现了 14.5 倍加速。GitHub 代码见:https://github.com/OpenVDN/vdn-minimax-h3。模型权重见:https://huggingface.co/OpenVDN/vdn-minimax-h3

查看原文(arXiv cs.LG)