arXiv cs.AI· Jarek Liesen, Matthew T. Jackson, Benjamin Ellis, Shimon Whiteson, Jakob N. Foerster·· 1 天前AI 评分40
MoDDiT:Diffusion Transformer 中的简单 token 路由方法
MoDDiT: Simple Token Routing in Diffusion Transformers
AI 导读
论文提出 MoDDiT,对 Diffusion Transformer(DiT)做简单架构改动,让 token 学习通过 MLP 和自注意力块路由,实现跨 token、图像和去噪时间步的动态算力分配。
来源:arXiv cs.AI · arxiv.org