MC-Sparse框架无需训练即提升扩散Transformer生成速度

论文AI 评分 68/100arXiv cs.AI
AI 聚合

本条为 AI 依据下方公开信源自动整理生成的摘要,不构成转载,可能存在偏差,请以原文为准。整理者:贝果科技 AI 资讯助手。

针对扩散Transformer长序列生成耗时问题,现有稀疏注意力机制在高稀疏度下会损害生成质量。研究定位其退化源于词元分组约束、交互选择不准及丢弃词元丢失贡献。由此提出免训练框架MC-Sparse,它挑选独立KV词元并将相似查询组成对齐组以提升GPU效率,同时缓存查询组、基于精确概率选出的KV索引及残差,供后续去噪步复用。在视频与3D生成模型上,该框架比现有基线保真度更高且加速更大。相对稠密注意力,在Minimax-H3-Base上实现1.80倍去噪加速,在3D生成上实现2.32倍加速,均无可见质量损失。

为什么值得看

免训练实现2倍以上去噪加速且无质量损失,对视频及3D生成推理优化极具实用价值。

MiniMax对齐GPU

信源1 家

  1. [1]arXiv cs.AI一手信源MC-Sparse: Deconstructing and Closing the Dense-Sparse Attention Gap in Diffusion Transformers

关键事实

  • MC-Sparse是一个免训练的稀疏注意力框架,旨在加速扩散Transformer的长序列生成[1]

  • 该框架在Minimax-H3-Base上实现1.80倍去噪加速,在3D生成上实现2.32倍加速[1]

  • 研究将现有稀疏注意力质量退化的原因归结为词元分组约束、交互选择不准和丢弃词元贡献丢失[1]

相关 · 论文

本页内容由 AI 自动聚合公开信源生成,仅供了解行业动态参考,不构成任何投资或决策建议。如需引用请以原文出处为准。