Nucleus-Image: Sparse MoE for Image Generation
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
- 2025HiDream-I1: A High-Efficient Image Generative Foundation Model with Sparse Diffusion Transformer28 May
- 2025Diffusion-4K: Ultra-High-Resolution Image Synthesis with Latent Diffusion Models24 Mar
- 2025SANA 1.5: Efficient Scaling of Training-Time and Inference-Time Compute in Linear Diffusion Transformer30 Jan
- 2024Scaling Diffusion Transformers to 16 Billion Parameters16 Jul
- 2024Lumina-Next: Making Lumina-T2X Stronger and Faster with Next-DiT5 Jun
- 2024Unveiling and Mitigating Memorization in Text-to-image Diffusion Models through Cross Attention17 Mar
- 2024PixArt-Sigma: Weak-to-Strong Training of Diffusion Transformer for 4K Text-to-Image Generation7 Mar
- 2024Scaling Rectified Flow Transformers for High-Resolution Image Synthesis5 Mar
- 2023PixArt-α: Fast Training of Diffusion Transformer for Photorealistic Text-to-Image Synthesis30 Sep
- 2023SDXL: Improving Latent Diffusion Models for High-Resolution Image Synthesis4 Jul
- 2022Scalable Diffusion Models with Transformers19 Dec
- 2022Classifier-Free Diffusion Guidance26 Jul
1 more
Other references · 31
- 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,744
- 2025Qwen3 Technical Reportcited by 8,820
- 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Modelcited by 8,065
- 2022Flow Straight and Fast: Learning to Generate and Transfer Data with Rectified Flowcited by 3,700
- 2025Wan: Open and Advanced Large-Scale Video Generative Modelscited by 2,708
- 2020GLU Variants Improve Transformercited by 1,804
- 2019Root Mean Square Layer Normalizationcited by 1,480
- 2023GQA: Training Generalized Multi-Query Transformer Models from Multi-Head Checkpointscited by 1,346
- 2023GenEval: An Object-Focused Framework for Evaluating Text-to-Image Alignmentcited by 972
- 2025Emerging Properties in Unified Multimodal Pretrainingcited by 897
- 2025Janus-Pro: Unified Multimodal Understanding and Generation with Data and Model Scalingcited by 812
- 2024Emu3: Next-Token Prediction is All You Needcited by 781
19 more
- 2024Show-o: One Single Transformer to Unify Multimodal Understanding and Generationcited by 752
- 2023PyTorch FSDP: Experiences on Scaling Fully Sharded Data Parallelcited by 660
- 2022Mixture-of-Experts with Expert Choice Routingcited by 623
- 2024FlashAttention-3: Fast and Accurate Attention with Asynchrony and Low-precisioncited by 519
- 2024ELLA: Equip Diffusion Models with LLM for Enhanced Semantic Alignmentcited by 475
- 2024Janus: Decoupling Visual Encoding for Unified Multimodal Understanding and Generationcited by 457
- 2025BLIP3-o: A Family of Fully Open Unified Multimodal Models-Architecture, Training and Datasetcited by 376
- 2024Hunyuan-DiT: A Powerful Multi-Resolution Diffusion Transformer with Fine-Grained Chinese Understandingcited by 298
- 2024Playground v2.5: Three Insights towards Enhancing Aesthetic Quality in Text-to-Image Generationcited by 256
- 2025OmniGen2: Towards Instruction-Aligned Multimodal Generationcited by 253
- 2025Show-o2: Improved Native Unified Multimodal Modelscited by 250
- 2024JanusFlow: Harmonizing Autoregression and Rectified Flow for Unified Multimodal Understanding and Generationcited by 163
- 2025Seedream 3.0 Technical Reportcited by 154
- 2024Playground v3: Improving Text-to-Image Alignment with Deep-Fusion Large Language Modelscited by 116
- 2025OneIG-Bench: Omni-dimensional Nuanced Evaluation for Image Generationcited by 87
- 2024TorchTitan: One-stop PyTorch native solution for production ready LLM pre-trainingcited by 85
- 2020Improved Techniques for Training Single-Image GANscited by 78
- 2025WSM: Decay-Free Learning Rate Schedule via Checkpoint Merging for LLM Pre-trainingcited by 23
- 2026The Algorithmic Gaze of Image Quality Assessment: An Audit and Trace Ethnography of the LAION-Aesthetics Predictorcited by 1
Built on by
Citing papers per year: 2026: 5.
5 papers.
- 2026Worldscape-MoE: A Unified Mixture-of-Experts World Model for Scalable Heterogeneous Action Controlcited by 5
- 2026Normalizing Trajectory Modelscited by 2
- 2026Complete-muE: Optimal Hyperparameter Transfer and Scaling for MoE Modelscited by 1
- 2026MoECa: Aligning Feature Reuse with Expert Decomposition in Diffusion Transformerscited by 0
- 2026Free-Lunch Augmentation by Revisiting Diffusion-Based Data Generation for Cross-Domain Few-Shot Object Detectioncited by 0
- 2026Free-Lunch Augmentation by Revisiting Diffusion-Based Data Generation for Cross-Domain Few-Shot Object Detection5 Aug 2026
- 2026Worldscape-MoE: A Unified Mixture-of-Experts World Model for Scalable Heterogeneous Action Control4 Jul 2026
- 2026MoECa: Aligning Feature Reuse with Expert Decomposition in Diffusion Transformers14 Jun 2026
- 2026Complete-muE: Optimal Hyperparameter Transfer and Scaling for MoE Models22 May 2026
- 2026Normalizing Trajectory Models8 May 2026
Similar, not cited
Close in wording, no citation either way
- 2024EC-DIT: Scaling Diffusion Transformers with Adaptive Expert-Choice Routing
- 2025Dense2MoE: Restructuring Diffusion Transformer to MoE for Efficient Text-to-Image Generation
- 2023Exploring Sparse MoE in GANs for Text-conditioned Image Synthesis
- 2023Muse: Text-To-Image Generation via Masked Generative Transformers
- 2026The Thinking Pixel: Recursive Sparse Reasoning in Multimodal Diffusion Latents
- 2024Stretching Each Dollar: Diffusion Training from Scratch on a Micro-Budget
- 2025Reusing Computation in Text-to-Image Diffusion for Efficient Generation of Image Sets
- 2026Keep The Essentials: Efficient Reference Conditioned Generation via Token Dropping