Matryoshka Diffusion Models
On the map: Diffusion models → Diffusion foundations and sampling
Abstract (arXiv)
Diffusion models are the de facto approach for generating high-quality images and videos, but learning high-dimensional models remains a formidable task due to computational and optimization challenges. Existing methods often resort to training cascaded models in pixel space or using a downsampled latent space of a separately trained auto-encoder. In this paper, we introduce Matryoshka Diffusion Models(MDM), an end-to-end framework for high-resolution image and video synthesis. We propose a diffusion process that denoises inputs at multiple resolutions jointly and uses a NestedUNet architecture where features and parameters for small-scale inputs are nested within those of large scales. In addition, MDM enables a progressive training schedule from lower to higher resolutions, which leads to significant improvements in optimization for high-resolution generation. We demonstrate the effectiveness of our approach on various benchmarks, including class-conditioned image generation, high-resolution text-to-image, and text-to-video applications. Remarkably, we can train a single pixel-space model at resolutions of up to 1024x1024 pixels, demonstrating strong zero-shot generalization using the CC12M dataset, which contains only 12 million images. Our code is released at https://github.com/apple/ml-mdm
Key facts. Cited by 72 papers in the map (data through 14 Oct 2026). Placed in Diffusion foundations and sampling, one of 14 directions in Diffusion models. Builds on 49 of its 56 references that are in the map. First posted on arXiv 23 Oct 2023.
Builds on
49 of 56 references.
Closest ancestors · 12
Same direction, newest first
- 2023SDXL: Improving Latent Diffusion Models for High-Resolution Image Synthesis4 Jul
- 2023Simple diffusion: End-to-end diffusion for high resolution images26 Jan
- 2023On the Importance of Noise Scheduling for Diffusion Models26 Jan
- 2022Scalable Diffusion Models with Transformers19 Dec
- 2022f-DM: A Multi-stage Diffusion Model via Progressive Signal Transformation10 Oct
- 2022Pyramidal Denoising Diffusion Probabilistic Models3 Aug
- 2021High-Resolution Image Synthesis with Latent Diffusion Models20 Dec
- 2021Cascaded Diffusion Models for High Fidelity Image Generation30 May
- 2021Diffusion Models Beat GANs on Image Synthesis11 May
- 2021Improved Denoising Diffusion Probabilistic Models18 Feb
- 2020Score-Based Generative Modeling through Stochastic Differential Equations26 Nov
- 2020Denoising Diffusion Probabilistic Models19 Jun
Other references · 37
- 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,744
- 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,713
- 2014Microsoft COCO: Common Objects in Contextcited by 17,608
- 2015Deep Unsupervised Learning using Nonequilibrium Thermodynamicscited by 7,058
- 2022Hierarchical Text-Conditional Image Generation with CLIP Latentscited by 6,585
- 2022Photorealistic Text-to-Image Diffusion Models with Deep Language Understandingcited by 6,355
- 2017Neural Discrete Representation Learningcited by 4,712
- 2020Taming Transformers for High-Resolution Image Synthesiscited by 3,253
- 2022DreamFusion: Text-to-3D using 2D Diffusioncited by 3,025
- 2017Progressive Growing of GANs for Improved Quality, Stability, and Variationcited by 2,988
- 2022Video Diffusion Modelscited by 2,305
- 2022Imagen Video: High Definition Video Generation with Diffusion Modelscited by 1,778
25 more
- 2021Image Super-Resolution via Iterative Refinementcited by 1,511
- 2023AnimateDiff: Animate Your Personalized Text-to-Image Diffusion Models without Specific Tuningcited by 1,433
- 2022Scaling Autoregressive Models for Content-Rich Text-to-Image Generationcited by 1,295
- 2021Conceptual 12M: Pushing Web-Scale Image-Text Pre-Training To Recognize Long-Tail Visual Conceptscited by 1,216
- 2021Efficient Geometry-aware 3D Generative Adversarial Networkscited by 1,205
- 2021Frozen in Time: A Joint Video and Image Encoder for End-to-End Retrievalcited by 1,177
- 2022Diffusion-LM Improves Controllable Text Generationcited by 1,027
- 2022eDiff-I: Text-to-Image Diffusion Models with an Ensemble of Expert Denoiserscited by 931
- 2016Conditional Image Generation with PixelCNN Decoderscited by 862
- 2016Pixel Recurrent Neural Networkscited by 816
- 2015Scheduled Sampling for Sequence Prediction with Recurrent Neural Networkscited by 803
- 2023AudioLDM: Text-to-Audio Generation with Latent Diffusion Modelscited by 599
- 2023Scaling up GANs for Text-to-Image Synthesiscited by 492
- 2022Make-A-Scene: Scene-Based Text-to-Image Generation with Human Priorscited by 487
- 2021StyleNeRF: A Style-based 3D-Aware Generator for High-resolution Image Synthesiscited by 477
- 2022Matryoshka Representation Learningcited by 282
- 2015Deep Generative Image Models using a Laplacian Pyramid of Adversarial Networkscited by 214
- 2022Scalable Adaptive Computation for Iterative Generationcited by 195
- 2019Scaling Autoregressive Video Modelscited by 190
- 2023NerfDiff: Single-image View Synthesis with NeRF-guided Distillation from 3D-aware Diffusioncited by 180
- 2022Generative Modelling With Inverse Heat Dissipationcited by 167
- 2023RAPHAEL: Text-to-Image Generation via Large Mixture of Diffusion Pathscited by 142
- 2016Video Pixel Networkscited by 108
- 2023PLANNER: Generating Diversified Paragraph via Latent Language Diffusion Modelcited by 58
- 2023Learning multi-scale local conditional probability models of imagescited by 21
Built on by
Citing papers per year: 2023: 5; 2024: 32; 2025: 25; 2026: 10.
72 papers.
Sort:
By area:
- 2024Lumiere: A Space-Time Diffusion Model for Video Generationcited by 455
- 2024Hunyuan-DiT: A Powerful Multi-Resolution Diffusion Transformer with Fine-Grained Chinese Understandingcited by 298
- 2023Align Your Gaussians: Text-to-4D with Dynamic 3D Gaussians and Composed Diffusion Modelscited by 169
- 2024InitNO: Boosting Text-to-Image Diffusion Models via Initial Noise Optimizationcited by 123
- 2024Snap Video: Scaled Spatiotemporal Transformers for Text-to-Video Synthesiscited by 105
- 2024DistriFusion: Distributed Parallel Inference for High-Resolution Diffusion Modelscited by 98
- 2025PixelFlow: Pixel-Space Generative Models with Flowcited by 97
- 2024Simpler Diffusion (SiD2): 1.5 FID on ImageNet512 with pixel-space diffusioncited by 75
- 2024Make a Cheap Scaling: A Self-Cascade Diffusion Model for Higher-Resolution Adaptationcited by 51
- 2024Steering Rectified Flow Models in the Vector Field for Controlled Image Generationcited by 39
- 2023Exploiting Diffusion Prior for Generalizable Dense Predictioncited by 38
- 2024Alleviating Distortion in Image Generation via Multi-Resolution Diffusion Models and Time-Dependent Layer Normalizationcited by 36
38 more
- 2024RodinHD: High-Fidelity 3D Avatar Generation with Diffusion Modelscited by 36
- 2025Flow to the Mode: Mode-Seeking Diffusion Autoencoders for State-of-the-Art Image Tokenizationcited by 36
- 2024DART: Denoising Autoregressive Transformer for Scalable Text-to-Image Generationcited by 32
- 2025FlashVideo: Flowing Fidelity to Detail for Efficient High-Resolution Video Generationcited by 32
- 2025Beyond Matryoshka: Revisiting Sparse Coding for Adaptive Representationcited by 26
- 2025Latent Space Super-Resolution for Higher-Resolution Image Generation with Diffusion Modelscited by 25
- 2024Hierarchical Patch Diffusion Models for High-Resolution Video Generationcited by 24
- 2024Shielded Diffusion: Generating Novel and Diverse Images using Sparse Repellencycited by 22
- 2024Upsample Guidance: Scale Up Diffusion Models without Trainingcited by 20
- 2024Learning Image Priors through Patch-based Diffusion Models for Solving Inverse Problemscited by 20
- 2023Synthetically Enhanced: Unveiling Synthetic Data's Potential in Medical Imaging Researchcited by 18
- 2024RealCompo: Balancing Realism and Compositionality Improves Text-to-Image Diffusion Modelscited by 18
- 2024Sora Detector: A Unified Hallucination Detection for Large Text-to-Video Modelscited by 17
- 2025Enhancing Intent Understanding for Ambiguous prompt: A Human-Machine Co-Adaption Strategycited by 16
- 2024Public Domain 12M: A Highly Aesthetic Image-Text Dataset with Novel Governance Mechanismscited by 15
- 2024Spectral Image Tokenizercited by 15
- 2024I4VGen: Image as Free Stepping Stone for Text-to-Video Generationcited by 14
- 2024Is One GPU Enough? Pushing Image Generation at Higher-Resolutions with Foundation Modelscited by 14
- 2025Audio Super-Resolution with Latent Bridge Modelscited by 14
- 2024Kaleido Diffusion: Improving Conditional Diffusion Models with Autoregressive Latent Modelingcited by 13
- 2025Scale-wise Distillation of Diffusion Modelscited by 11
- 2025How far can we go with ImageNet for Text-to-Image generation?cited by 10
- 2024∞-Brush: Controllable Large Image Synthesis with Diffusion Models in Infinite Dimensionscited by 8
- 2024Preference Adaptive and Sequential Text-to-Image Generationcited by 8
- 2026Asymmetric Flow Modelscited by 8
- 2024MatMamba: A Matryoshka State Space Modelcited by 7
- 2025Latent Wavelet Diffusion For Ultra-High-Resolution Image Synthesiscited by 7
- 2025Projected Coupled Diffusion for Test-Time Constrained Joint Generationcited by 7
- 2025ImageDoctor: Diagnosing Text-to-Image Generation via Grounded Image Reasoningcited by 7
- 2024E(3)-equivariant models cannot learn chirality: Field-based molecular generationcited by 6
- 2025Improving Progressive Generation with Decomposable Flow Matchingcited by 6
- 2024AsCAN: Asymmetric Convolution-Attention Networks for Efficient Recognition and Generationcited by 5
- 2025ShortFT: Diffusion Model Alignment via Shortcut-based Fine-Tuningcited by 5
- 2025NAMI: Efficient Image Generation via Bridged Progressive Rectified Flow Transformerscited by 4
- 2025Pyramidal Patchification Flow for Visual Generationcited by 4
- 2025Elastic ViTs from Pretrained Models without Retrainingcited by 4
- 2026DDiT: Dynamic Patch Scheduling for Efficient Diffusion Transformerscited by 4
- 2026Scale Space Diffusioncited by 4
50 most cited of 72 All 72 (JSON)
- 2026AReA: Adaptive Receptive-Field Attention for Efficient High-Resolution Diffusion Transformers12 Oct 2026
- 2026APEX: Asynchronous Prefix Denoising for Extreme Compression9 Oct 2026
- 2026Balancing Frequencies and Pixels in Flow Matching2 Sep 2026
- 2026SEGA: Spectral-Energy Guided Attention for Resolution Extrapolation in Diffusion Transformers21 May 2026
- 2026DIVE: Embedding Compression via Self-Limiting Gradient Updates20 May 2026
- 2026Asymmetric Flow Models13 May 2026
- 2026PoM: A Linear-Time Replacement for Attention with the Polynomial Mixer7 Apr 2026
- 2026CTCal: Rethinking Text-to-Image Diffusion Models via Cross-Timestep Self-Calibration21 Mar 2026
- 2026Scale Space Diffusion9 Mar 2026
- 2026DDiT: Dynamic Patch Scheduling for Efficient Diffusion Transformers19 Feb 2026
- 2025MFM-point: Multi-scale Flow Matching for Point Cloud Generation25 Nov 2025
- 2025MIRO: MultI-Reward cOnditioned pretraining improves T2I quality and efficiency29 Oct 2025
38 more
- 2025Eigenfunction Extraction for Ordered Representation Learning28 Oct 2025
- 2025Elastic ViTs from Pretrained Models without Retraining20 Oct 2025
- 2025CanvasMAR: Improving Masked Autoregressive Video Prediction With Canvas15 Oct 2025
- 2025ImageDoctor: Diagnosing Text-to-Image Generation via Grounded Image Reasoning1 Oct 2025
- 2025Audio Super-Resolution with Latent Bridge Models22 Sep 2025
- 2025LowDiff: Efficient Diffusion Sampling with Low-Resolution Condition18 Sep 2025
- 2025Projected Coupled Diffusion for Test-Time Constrained Joint Generation14 Aug 2025
- 2025KnapFormer: An Online Load Balancer for Efficient Diffusion Transformers Training8 Aug 2025
- 2025ShortFT: Diffusion Model Alignment via Shortcut-based Fine-Tuning30 Jul 2025
- 2025APT: Improving Diffusion Models for High Resolution Image Generation with Adaptive Path Tracing29 Jul 2025
- 2025Pyramidal Patchification Flow for Visual Generation30 Jun 2025
- 2025Improving Progressive Generation with Decomposable Flow Matching24 Jun 2025
- 2025Latent Wavelet Diffusion For Ultra-High-Resolution Image Synthesis31 May 2025
- 2025PixelFlow: Pixel-Space Generative Models with Flow10 Apr 2025
- 2025Latent Space Super-Resolution for Higher-Resolution Image Generation with Diffusion Models24 Mar 2025
- 2025Scale-wise Distillation of Diffusion Models20 Mar 2025
- 2025AugGen: Synthetic Augmentation using Diffusion Models Can Improve Recognition14 Mar 2025
- 2025Flow to the Mode: Mode-Seeking Diffusion Autoencoders for State-of-the-Art Image Tokenization14 Mar 2025
- 2025NAMI: Efficient Image Generation via Bridged Progressive Rectified Flow Transformers12 Mar 2025
- 2025Beyond Matryoshka: Revisiting Sparse Coding for Adaptive Representation3 Mar 2025
- 2025How far can we go with ImageNet for Text-to-Image generation?28 Feb 2025
- 2025FlashVideo: Flowing Fidelity to Detail for Efficient High-Resolution Video Generation7 Feb 2025
- 2025Enhancing Intent Understanding for Ambiguous prompt: A Human-Machine Co-Adaption Strategy25 Jan 2025
- 2024Spectral Image Tokenizer12 Dec 2024
- 2024Preference Adaptive and Sequential Text-to-Image Generation10 Dec 2024
- 2024Nested Diffusion Models Using Hierarchical Latent Priors8 Dec 2024
- 2024INRFlow: Flow Matching for INRs in Ambient Space5 Dec 2024
- 2024Steering Rectified Flow Models in the Vector Field for Controlled Image Generation27 Nov 2024
- 2024PoM: Efficient Image and Video Generation with the Polynomial Mixer19 Nov 2024
- 2024AsCAN: Asymmetric Convolution-Attention Networks for Efficient Recognition and Generation7 Nov 2024
- 2024Public Domain 12M: A Highly Aesthetic Image-Text Dataset with Novel Governance Mechanisms30 Oct 2024
- 2024Simpler Diffusion (SiD2): 1.5 FID on ImageNet512 with pixel-space diffusion25 Oct 2024
- 2024DART: Denoising Autoregressive Transformer for Scalable Text-to-Image Generation10 Oct 2024
- 2024MatMamba: A Matryoshka State Space Model9 Oct 2024
- 2024Shielded Diffusion: Generating Novel and Diverse Images using Sparse Repellency8 Oct 2024
- 2024JPEG-LM: LLMs as Image Generators with Canonical Codec Representations15 Aug 2024
- 2024∞-Brush: Controllable Large Image Synthesis with Diffusion Models in Infinite Dimensions20 Jul 2024
- 2024RodinHD: High-Fidelity 3D Avatar Generation with Diffusion Models9 Jul 2024
50 newest of 72 All 72 (JSON)
- 2024Lumiere: A Space-Time Diffusion Model for Video Generationcited by 455
- 2024Hunyuan-DiT: A Powerful Multi-Resolution Diffusion Transformer with Fine-Grained Chinese Understandingcited by 298
- 2023Align Your Gaussians: Text-to-4D with Dynamic 3D Gaussians and Composed Diffusion Modelscited by 169
- 2024InitNO: Boosting Text-to-Image Diffusion Models via Initial Noise Optimizationcited by 123
- 2024DistriFusion: Distributed Parallel Inference for High-Resolution Diffusion Modelscited by 98
- 2024Simpler Diffusion (SiD2): 1.5 FID on ImageNet512 with pixel-space diffusioncited by 75
- 2024Make a Cheap Scaling: A Self-Cascade Diffusion Model for Higher-Resolution Adaptationcited by 51
- 2024Steering Rectified Flow Models in the Vector Field for Controlled Image Generationcited by 39
- 2023Exploiting Diffusion Prior for Generalizable Dense Predictioncited by 38
- 2024Alleviating Distortion in Image Generation via Multi-Resolution Diffusion Models and Time-Dependent Layer Normalizationcited by 36
- 2024RodinHD: High-Fidelity 3D Avatar Generation with Diffusion Modelscited by 36
- 2025Flow to the Mode: Mode-Seeking Diffusion Autoencoders for State-of-the-Art Image Tokenizationcited by 36
12 most cited of 45 All 45 (JSON)
- 2026AReA: Adaptive Receptive-Field Attention for Efficient High-Resolution Diffusion Transformers12 Oct 2026
- 2026Balancing Frequencies and Pixels in Flow Matching2 Sep 2026
- 2026SEGA: Spectral-Energy Guided Attention for Resolution Extrapolation in Diffusion Transformers21 May 2026
- 2026CTCal: Rethinking Text-to-Image Diffusion Models via Cross-Timestep Self-Calibration21 Mar 2026
- 2026Scale Space Diffusion9 Mar 2026
- 2026DDiT: Dynamic Patch Scheduling for Efficient Diffusion Transformers19 Feb 2026
- 2025MFM-point: Multi-scale Flow Matching for Point Cloud Generation25 Nov 2025
- 2025LowDiff: Efficient Diffusion Sampling with Low-Resolution Condition18 Sep 2025
- 2025Projected Coupled Diffusion for Test-Time Constrained Joint Generation14 Aug 2025
- 2025KnapFormer: An Online Load Balancer for Efficient Diffusion Transformers Training8 Aug 2025
- 2025ShortFT: Diffusion Model Alignment via Shortcut-based Fine-Tuning30 Jul 2025
- 2025APT: Improving Diffusion Models for High Resolution Image Generation with Adaptive Path Tracing29 Jul 2025
12 newest of 45 All 45 (JSON)
- 2024Preference Adaptive and Sequential Text-to-Image Generation10 Dec 2024
- 2024Snap Video: Scaled Spatiotemporal Transformers for Text-to-Video Synthesiscited by 105
- 2025PixelFlow: Pixel-Space Generative Models with Flowcited by 97
- 2025FlashVideo: Flowing Fidelity to Detail for Efficient High-Resolution Video Generationcited by 32
- 2025Beyond Matryoshka: Revisiting Sparse Coding for Adaptive Representationcited by 26
- 2024Sora Detector: A Unified Hallucination Detection for Large Text-to-Video Modelscited by 17
- 2025Enhancing Intent Understanding for Ambiguous prompt: A Human-Machine Co-Adaption Strategycited by 16
- 2024Public Domain 12M: A Highly Aesthetic Image-Text Dataset with Novel Governance Mechanismscited by 15
- 2024Spectral Image Tokenizercited by 15
- 2024Is One GPU Enough? Pushing Image Generation at Higher-Resolutions with Foundation Modelscited by 14
- 2025Audio Super-Resolution with Latent Bridge Modelscited by 14
- 2025How far can we go with ImageNet for Text-to-Image generation?cited by 10
- 2026Asymmetric Flow Modelscited by 8
12 most cited of 24 All 24 (JSON)
- 2026APEX: Asynchronous Prefix Denoising for Extreme Compression9 Oct 2026
- 2026DIVE: Embedding Compression via Self-Limiting Gradient Updates20 May 2026
- 2026Asymmetric Flow Models13 May 2026
- 2026PoM: A Linear-Time Replacement for Attention with the Polynomial Mixer7 Apr 2026
- 2025Eigenfunction Extraction for Ordered Representation Learning28 Oct 2025
- 2025Elastic ViTs from Pretrained Models without Retraining20 Oct 2025
- 2025CanvasMAR: Improving Masked Autoregressive Video Prediction With Canvas15 Oct 2025
- 2025Audio Super-Resolution with Latent Bridge Models22 Sep 2025
- 2025Latent Wavelet Diffusion For Ultra-High-Resolution Image Synthesis31 May 2025
- 2025PixelFlow: Pixel-Space Generative Models with Flow10 Apr 2025
- 2025Beyond Matryoshka: Revisiting Sparse Coding for Adaptive Representation3 Mar 2025
- 2025How far can we go with ImageNet for Text-to-Image generation?28 Feb 2025
12 newest of 24 All 24 (JSON)
Similar, not cited
Close in wording, no citation either way
- 2024DiffuseHigh: Training-free Progressive High-Resolution Image Synthesis through Structure Guidance
- 2024MegaFusion: Extend Diffusion Models towards Higher-resolution Image Generation without Further Tuning
- 2022Representation Learning with Diffusion Models
- 2024DiM: Diffusion Mamba for Efficient High-Resolution Image Synthesis
- 2026Efficient and Training-Free Single-Image Diffusion Models
- 2023Diffusion Models Without Attention
- 2023Stable Video Diffusion: Scaling Latent Video Diffusion Models to Large Datasets
- 2022On Distillation of Guided Diffusion Models