arXiv:2602.12271cs.CVcs.LG2026-02被引 4

提出Monarch-RT,让视频生成实时运行更快更准。

MonarchRT: Efficient Attention for Real-Time Video Generation

  • 用结构化矩阵分解注意力,兼顾表达力与效率
  • 在Self-Forcing模型上实现95%注意力稀疏且质量无损
  • 适配实时生成场景,适合追求高效视频生成的研究者

基于扩散变换器的实时视频生成受限于3D自注意力的二次复杂度,尤其在少步数、自回归的实时场景中,错误随时间累积,每步需携带更多信息。此前的稀疏注意力近似在此场景下失效,因视频注意力既含周期性时空位置结构,又有动态语义对应和密集混合,超出经典top-k注意力的表达能力。为此,我们提出Monarch-RT,一种基于蒙德矩阵的视频扩散注意力参数化方法,通过对齐块结构与扩展的分块参数化,在保持高表达力的同时确保计算效率。结合定制Triton内核微调,显著降低参数化开销。验证表明,Monarch-RT优于仅针对双向模型设计的稀疏基线。在SOTA模型Self-Forcing上,其达到95%注意力稀疏度而无质量损失。优化实现超越FlashAttention-2、3、4,在Nvidia RTX 5090、H100、B200上分别获得1.4–11.8倍加速。首次在单张RTX 5090上实现Self-Forcing模型16帧/秒的真正实时视频生成。

原文摘要 · Abstract (English)

Real-time video generation with Diffusion Transformers is bottlenecked by the quadratic cost of 3D self-attention, especially in real-time regimes that are both few-step and autoregressive, where errors compound across time and each denoising step must carry substantially more information. In this setting, we find that prior sparse-attention approximations break down, despite showing strong results for bidirectional, many-step diffusion. Specifically, we observe that video attention is not reliably sparse, but instead combines pronounced periodic structure driven by spatiotemporal position with dynamic, sparse semantic correspondences and dense mixing, exceeding the representational capacity of even oracle top-k attention. Building on this insight, we propose Monarch-RT, a structured attention parameterization for video diffusion models that factorizes attention using Monarch matrices. Through appropriately aligned block structure and our extended tiled Monarch parameterization, we achieve high expressivity while preserving computational efficiency. We further overcome the overhead of parameterization through finetuning, with custom Triton kernels. We first validate the high efficacy of Monarch-RT over existing sparse baselines designed only for bidirectional models. We further observe that Monarch-RT attains up to 95% attention sparsity with no loss in quality when applied to the state-of-the-art model Self-Forcing, making Monarch-RT a pioneering work on highly-capable sparse attention parameterization for real-time video generation. Our optimized implementation outperforms FlashAttention-2, FlashAttention-3, and FlashAttention-4 kernels on Nvidia RTX 5090, H100, and B200 GPUs respectively, providing kernel speedups in the range of 1.4-11.8X. This enables us, for the first time, to achieve true real-time video generation with Self-Forcing at 16 FPS on a single RTX 5090.

视频生成扩散模型注意力机制实时推理

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。