MMOE: Modernizing Diffusion Transformers with Efficient Expert Design
On the map: Diffusion models → Discrete and text diffusion
Abstract (arXiv)
Modern large language models scale successfully by pairing capacity growth with efficiency, keeping per-token and deployment costs under control as capacity grows. AIGC Foundation Models (AFMs), especially diffusion-transformer backbones, have begun to adopt sparse experts, but recent efforts mostly enlarge total parameter counts and sparsity ratios without importing the efficiency mechanisms that made LLM scaling practical, so generation quality is seldom balanced against training and deployment cost. This raises a natural question: can the architectural principles behind efficient LLM scaling be adapted to AFMs in a more balanced way? We introduce ModernMOE (MMOE), a modernization of SiT-style diffusion transformers that systematically adapts routed experts, shared and lightweight experts, gate-residual routing, and attention-residual information reuse to AIGC generation. Rather than treating MoE as a single plug-in replacement, MMOE studies how different modern expert components affect convergence, efficiency, and generation quality when composed inside a diffusion transformer. Every experiment in this paper is trained on a single eight-GPU H100 node with batch size 256 for 400k steps, an accessible single-machine budget. Under matched training and sampling protocols and at this budget, MMOE reaches lower FID at every recorded checkpoint, that is, it converges faster per training step, than dense and intermediate sparse-expert baselines, and among the sparse variants it attains the best quality-cost balance. Routing analysis further shows stable expert specialization across depth, substantial use of lightweight routes, and modest step-to-step routing changes during denoising. These results suggest that AFMs can follow the balanced scaling path of LLMs by importing proven efficiency designs, rather than by simply increasing total parameters and sparsity ratios.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Placed in Discrete and text diffusion, one of 14 directions in Diffusion models. Builds on 41 of its 49 references that are in the map. First posted on arXiv 27 Jul 2026.
Builds on
41 of 49 references.
References · 41
- 2020Denoising Diffusion Probabilistic Modelscited by 20,851
- 2021High-Resolution Image Synthesis with Latent Diffusion Modelscited by 18,694
- 2020Score-Based Generative Modeling through Stochastic Differential Equationscited by 8,964
- 2022Scalable Diffusion Models with Transformerscited by 5,931
- 2022Classifier-Free Diffusion Guidancecited by 5,717
- 2022Flow Matching for Generative Modelingcited by 5,426
- 2022A ConvNet for the 2020scited by 3,545
- 2017Outrageously Large Neural Networks: The Sparsely-Gated Mixture-of-Experts Layercited by 3,413
- 2021Switch Transformers: Scaling to Trillion Parameter Models with Simple and Efficient Sparsitycited by 3,014
- 2022Elucidating the Design Space of Diffusion-Based Generative Modelscited by 2,994
- 2020GShard: Scaling Giant Models with Conditional Computation and Automatic Shardingcited by 1,601
- 2023Stochastic Interpolants: A Unifying Framework for Flows and Diffusionscited by 897
29 more
- 2021GLaM: Efficient Scaling of Language Models with Mixture-of-Expertscited by 890
- 2024DeepSeekMoE: Towards Ultimate Expert Specialization in Mixture-of-Experts Language Modelscited by 834
- 2021Scaling Vision with Sparse Mixture of Expertscited by 740
- 2024SiT: Exploring Flow and Diffusion-based Generative Models with Scalable Interpolant Transformerscited by 707
- 2024Representation Alignment for Generation: Training Diffusion Transformers Is Easier Than You Thinkcited by 666
- 2022Mixture-of-Experts with Expert Choice Routingcited by 611
- 2022All are Worth Words: A ViT Backbone for Diffusion Modelscited by 498
- 2022ST-MoE: Designing Stable and Transferable Sparse Expert Modelscited by 391
- 2024Mixture-of-Depths: Dynamically allocating compute in transformer-based language modelscited by 242
- 2023NeRO: Neural Geometry and BRDF Reconstruction of Reflective Objects from Multiview Imagescited by 163
- 2024Scaling Diffusion Transformers to 16 Billion Parameterscited by 65
- 2024Dynamic Diffusion Transformercited by 60
- 2024MoE++: Accelerating Mixture-of-Experts Methods with Zero-Computation Expertscited by 55
- 2025AI Flow: Perspectives, Scenarios, and Approachescited by 44
- 2024AI Flow at the Network Edgecited by 43
- 2024EC-DIT: Scaling Diffusion Transformers with Adaptive Expert-Choice Routingcited by 29
- 2025DiffMoE: Dynamic Token Selection for Scalable Diffusion Transformerscited by 23
- 2025Routing Matters in MoE: Scaling Diffusion Transformers with Explicit Routing Guidancecited by 23
- 2025OmniVDiff: Omni Controllable Video Diffusion for Generation and Understandingcited by 22
- 2025Seeing Sound, Hearing Sight: Uncovering Modality Bias and Conflict of AI models in Sound Localizationcited by 17
- 2025Uni-Retrieval: A Multi-Style Retrieval Framework for STEM's Educationcited by 13
- 2025Expert Race: A Flexible Routing Strategy for Scaling Diffusion Transformer with Mixture of Expertscited by 13
- 2025MMGen: Unified Multi-modal Image Generation and Understanding in One Gocited by 10
- 2025Efficient Training of Diffusion Mixture-of-Experts Models: A Practical Recipecited by 6
- 2025CtrlVDiff: Controllable Video Generation via Unified Multimodal Video Diffusioncited by 5
- 2025From Query to Explanation: Uni-RAG for Multi-Modal Retrieval-Augmented Learning in STEMcited by 2
- 2026FORGE: Fused On-Register Gradient Elimination for Memory-Efficient LLM Trainingcited by 2
- 2026Generative Transmission: Rethinking Computation, Bandwidth, and Memory in Communicationcited by 2
- 2026EduStory: A Unified Framework for Pedagogically-Consistent Multi-Shot STEM Instructional Video Generationcited by 1
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026Expert-Choice Routing Enables Adaptive Computation in Diffusion Language Models
- 2025From Next-Token to Next-Block: A Principled Adaptation Path for Diffusion LLMs
- 2025Scaling Diffusion Transformers Efficiently via μP
- 2026Accelerating Masked Diffusion Large Language Models: A Survey of Efficient Inference Techniques
- 2025LLaDA2.0: Scaling Up Diffusion Language Models to 100B
- 2025On Powerful Ways to Generate: Autoregression, Diffusion, and Beyond
- 2026Top 10 Open Challenges Steering the Future of Diffusion Language Model and Its Variants
- 2026LLaDA MoE v2: Scaling Mixture-of-Experts Diffusion Language Models