mlsanityData to 14 Oct 2026

Nucleus-Image: Sparse MoE for Image Generation

arXiv 2604.12163 ↗First posted 14 Apr 2026cs.CVcited by 5

On the map: Diffusion models → Diffusion foundations and sampling

also in 1 more
Abstract (arXiv)

We present Nucleus-Image, a text-to-image generation model that establishes a new Pareto frontier in quality-versus-efficiency by matching or exceeding leading models on GenEval, DPG-Bench, and OneIG-Bench while activating only approximately 2B parameters per forward pass. Nucleus-Image employs a sparse mixture-of-experts (MoE) diffusion transformer architecture with Expert-Choice Routing that scales total model capacity to 17B parameters across 64 routed experts per layer. We adopt a streamlined architecture optimized for inference efficiency by excluding text tokens from the transformer backbone entirely and using joint attention that enables text KV sharing across timesteps. To improve routing stability when using timestep modulation, we introduce a decoupled routing design that separates timestep-aware expert assignment from timestep-conditioned expert computation. We construct a large-scale training corpus of 1.5B high-quality training pairs spanning 700M unique images through multi-stage filtering, deduplication, aesthetic tiering, and caption curation. Training follows a progressive resolution curriculum (256 to 512 to 1024) with multi-aspect-ratio bucketing at every stage, coupled with progressive sparsification of the expert capacity factor. We adopt the Muon optimizer and share our parameter grouping recipe tailored for diffusion models with timestep modulation. Nucleus-Image demonstrates that sparse MoE scaling is a highly effective path to high-quality image generation, reaching the performance of models with significantly larger active parameter budgets at a fraction of the inference cost. These results are achieved without post-training optimization of any kind: no reinforcement learning, no direct preference optimization, and no human preference tuning. We release the training recipe, making Nucleus-Image the first fully open-source MoE diffusion model at this quality.

Key facts. Cited by 5 papers in the map (data through 14 Oct 2026). Placed in Diffusion foundations and sampling, one of 14 directions in Diffusion models. Builds on 44 of its 53 references that are in the map. First posted on arXiv 14 Apr 2026.

Builds on

44 of 53 references.

Closest ancestors · 13

Same direction, newest first

  1. 2025HiDream-I1: A High-Efficient Image Generative Foundation Model with Sparse Diffusion Transformer28 May
  2. 2025Diffusion-4K: Ultra-High-Resolution Image Synthesis with Latent Diffusion Models24 Mar
  3. 2025SANA 1.5: Efficient Scaling of Training-Time and Inference-Time Compute in Linear Diffusion Transformer30 Jan
  4. 2024Scaling Diffusion Transformers to 16 Billion Parameters16 Jul
  5. 2024Lumina-Next: Making Lumina-T2X Stronger and Faster with Next-DiT5 Jun
  6. 2024Unveiling and Mitigating Memorization in Text-to-image Diffusion Models through Cross Attention17 Mar
  7. 2024PixArt-Sigma: Weak-to-Strong Training of Diffusion Transformer for 4K Text-to-Image Generation7 Mar
  8. 2024Scaling Rectified Flow Transformers for High-Resolution Image Synthesis5 Mar
  9. 2023PixArt-α: Fast Training of Diffusion Transformer for Photorealistic Text-to-Image Synthesis30 Sep
  10. 2023SDXL: Improving Latent Diffusion Models for High-Resolution Image Synthesis4 Jul
  11. 2022Scalable Diffusion Models with Transformers19 Dec
  12. 2022Classifier-Free Diffusion Guidance26 Jul
1 more
  1. 2021High-Resolution Image Synthesis with Latent Diffusion Models20 Dec

Other references · 31

  1. 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,744
  2. 2025Qwen3 Technical Reportcited by 8,820
  3. 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Modelcited by 8,065
  4. 2022Flow Straight and Fast: Learning to Generate and Transfer Data with Rectified Flowcited by 3,700
  5. 2025Wan: Open and Advanced Large-Scale Video Generative Modelscited by 2,708
  6. 2020GLU Variants Improve Transformercited by 1,804
  7. 2019Root Mean Square Layer Normalizationcited by 1,480
  8. 2023GQA: Training Generalized Multi-Query Transformer Models from Multi-Head Checkpointscited by 1,346
  9. 2023GenEval: An Object-Focused Framework for Evaluating Text-to-Image Alignmentcited by 972
  10. 2025Emerging Properties in Unified Multimodal Pretrainingcited by 897
  11. 2025Janus-Pro: Unified Multimodal Understanding and Generation with Data and Model Scalingcited by 812
  12. 2024Emu3: Next-Token Prediction is All You Needcited by 781
19 more
  1. 2024Show-o: One Single Transformer to Unify Multimodal Understanding and Generationcited by 752
  2. 2023PyTorch FSDP: Experiences on Scaling Fully Sharded Data Parallelcited by 660
  3. 2022Mixture-of-Experts with Expert Choice Routingcited by 623
  4. 2024FlashAttention-3: Fast and Accurate Attention with Asynchrony and Low-precisioncited by 519
  5. 2024ELLA: Equip Diffusion Models with LLM for Enhanced Semantic Alignmentcited by 475
  6. 2024Janus: Decoupling Visual Encoding for Unified Multimodal Understanding and Generationcited by 457
  7. 2025BLIP3-o: A Family of Fully Open Unified Multimodal Models-Architecture, Training and Datasetcited by 376
  8. 2024Hunyuan-DiT: A Powerful Multi-Resolution Diffusion Transformer with Fine-Grained Chinese Understandingcited by 298
  9. 2024Playground v2.5: Three Insights towards Enhancing Aesthetic Quality in Text-to-Image Generationcited by 256
  10. 2025OmniGen2: Towards Instruction-Aligned Multimodal Generationcited by 253
  11. 2025Show-o2: Improved Native Unified Multimodal Modelscited by 250
  12. 2024JanusFlow: Harmonizing Autoregression and Rectified Flow for Unified Multimodal Understanding and Generationcited by 163
  13. 2025Seedream 3.0 Technical Reportcited by 154
  14. 2024Playground v3: Improving Text-to-Image Alignment with Deep-Fusion Large Language Modelscited by 116
  15. 2025OneIG-Bench: Omni-dimensional Nuanced Evaluation for Image Generationcited by 87
  16. 2024TorchTitan: One-stop PyTorch native solution for production ready LLM pre-trainingcited by 85
  17. 2020Improved Techniques for Training Single-Image GANscited by 78
  18. 2025WSM: Decay-Free Learning Rate Schedule via Checkpoint Merging for LLM Pre-trainingcited by 23
  19. 2026The Algorithmic Gaze of Image Quality Assessment: An Audit and Trace Ethnography of the LAION-Aesthetics Predictorcited by 1

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 5.

5 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2024EC-DIT: Scaling Diffusion Transformers with Adaptive Expert-Choice Routing
  2. 2025Dense2MoE: Restructuring Diffusion Transformer to MoE for Efficient Text-to-Image Generation
  3. 2023Exploring Sparse MoE in GANs for Text-conditioned Image Synthesis
  4. 2023Muse: Text-To-Image Generation via Masked Generative Transformers
  5. 2026The Thinking Pixel: Recursive Sparse Reasoning in Multimodal Diffusion Latents
  6. 2024Stretching Each Dollar: Diffusion Training from Scratch on a Micro-Budget
  7. 2025Reusing Computation in Text-to-Image Diffusion for Efficient Generation of Image Sets
  8. 2026Keep The Essentials: Efficient Reference Conditioned Generation via Token Dropping