arXiv:2506.10915cs.CVcs.AI2025-06被引 13

用Mamba架构实现高效图文生成,速度提升45%。

M4V: Multimodal Mamba for Efficient Text-to-Video Generation

  • 设计多模态Mamba块,融合文本与视觉信息
  • 生成768x1280视频时降低45%计算量
  • 适合追求高效生成的开发者和研究者

图文生成已极大丰富内容创作,有望发展为强大的世界模拟器。然而,建模庞大的时空空间仍计算成本高昂,尤其在使用Transformer时,其序列处理具有二次复杂度,限制了实际应用。近期线性时间序列建模(如Mamba架构)提供了更高效的替代方案。但其原始设计难以直接应用于多模态与时空视频生成任务。为此,我们提出M4V,一种用于高效图文生成的多模态Mamba框架。具体地,在框架中设计了多模态扩散Mamba(MM-DiM)块,实现多模态信息与时空建模的无缝融合。通过新颖的双向多模态标记重组机制与视觉寄存器,增强时空一致性。实验显示,相较于基于注意力的模型,M4V在生成768x1280分辨率视频时减少45%浮点运算量。此外,本工作探索多种训练策略,以更好理解仅使用公开数据集训练图文生成模型的方法。大量基准测试表明,M4V在显著降低计算成本的同时,仍能生成高质量视频。

原文摘要 · Abstract (English)

Text-to-video generation has significantly enriched content creation and holds the potential to evolve into powerful world simulators. However, modeling the vast spatiotemporal space remains computationally demanding, particularly when employing Transformers, which incur quadratic complexity in sequence processing and thus limit practical applications. Recent advancements in linear-time sequence modeling, particularly the Mamba architecture, offer a more efficient alternative. Nevertheless, its plain design limits its direct applicability to multimodal and spatiotemporal video generation tasks. To address these challenges, we introduce M4V, a multimodal Mamba framework for efficient text-to-video generation. Specifically, a MultiModal diffusion Mamba (MM-DiM) block is designed within the framework to enable seamless integration of multimodal information and spatiotemporal modeling. In detail, we introduce a novel multimodal token re-composition design, which employs a bidirectional scheme for multimodal information integration through simple token arrangement, along with visual registers to enhance spatialtemporal consistency. As a result, the MM-DiM blocks in M4V reduce FLOPs by 45% compared with the attention-based alternative when generating videos at 768x1280 resolution. Additionally, several training strategies are explored in this work to provide a better understanding of training text-to-video models using only publicly available datasets. Extensive experiments on text-to-video benchmarks demonstrate M4V's ability to produce high-quality videos while significantly lowering computational costs. Project page: https://huangjch526.github.io/M4V_project/.

图文生成Mamba效率优化

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