English

Efficient Training of Diffusion Mixture-of-Experts Models: A Practical Recipe

Machine Learning 2025-12-02 v1 Computer Vision and Pattern Recognition

Abstract

Recent efforts on Diffusion Mixture-of-Experts (MoE) models have primarily focused on developing more sophisticated routing mechanisms. However, we observe that the underlying architectural configuration space remains markedly under-explored. Inspired by the MoE design paradigms established in large language models (LLMs), we identify a set of crucial architectural factors for building effective Diffusion MoE models--including DeepSeek-style expert modules, alternative intermediate widths, varying expert counts, and enhanced attention positional encodings. Our systematic study reveals that carefully tuning these configurations is essential for unlocking the full potential of Diffusion MoE models, often yielding gains that exceed those achieved by routing innovations alone. Through extensive experiments, we present novel architectures that can be efficiently applied to both latent and pixel-space diffusion frameworks, which provide a practical and efficient training recipe that enables Diffusion MoE models to surpass strong baselines while using equal or fewer activated parameters. All code and models are publicly available at: https://github.com/yhlleo/EfficientMoE.

Keywords

Cite

@article{arxiv.2512.01252,
  title  = {Efficient Training of Diffusion Mixture-of-Experts Models: A Practical Recipe},
  author = {Yahui Liu and Yang Yue and Jingyuan Zhang and Chenxi Sun and Yang Zhou and Wencong Zeng and Ruiming Tang and Guorui Zhou},
  journal= {arXiv preprint arXiv:2512.01252},
  year   = {2025}
}

Comments

9 pages, 7 figures

R2 v1 2026-07-01T08:02:58.354Z