Bag of Design Choices for Inference of High-Resolution Masked Generative Transformer
On the map: Diffusion models → Discrete and text diffusion
Abstract (arXiv)
Text-to-image diffusion models (DMs) develop at an unprecedented pace, supported by thorough theoretical exploration and empirical analysis. Unfortunately, the discrepancy between DMs and autoregressive models (ARMs) complicates the path toward achieving the goal of unified vision and language generation. Recently, the masked generative Transformer (MGT) serves as a promising intermediary between DM and ARM by predicting randomly masked image tokens (i.e., masked image modeling), combining the efficiency of DM with the discrete token nature of ARM. However, we find that the comprehensive analyses regarding the inference for MGT are virtually non-existent, and thus we aim to present positive design choices to fill this gap. We propose and redesign a set of enhanced inference techniques tailored for MGT, providing a detailed analysis of their performance. Additionally, we explore several DM-based approaches aimed at accelerating the sampling process on MGT. Extensive experiments and empirical analyses on the recent SOTA MGT, such as MaskGIT and Meissonic lead to concrete and effective design choices, and these design choices can be merged to achieve further performance gains. For instance, in terms of enhanced inference, we achieve winning rates of approximately 70% compared to vanilla sampling on HPS v2 with Meissonic-1024x1024.
Key facts. Cited by 7 papers in the map (data through 14 Oct 2026). Placed in Discrete and text diffusion, one of 14 directions in Diffusion models. Builds on 44 of its 50 references that are in the map. First posted on arXiv 16 Nov 2024.
Builds on
44 of 50 references.
References · 44
- 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,744
- 2023GPT-4 Technical Reportcited by 18,396
- 2014Microsoft COCO: Common Objects in Contextcited by 17,608
- 2023LLaMA: Open and Efficient Foundation Language Modelscited by 14,132
- 2020Denoising Diffusion Implicit Modelscited by 9,781
- 2020Score-Based Generative Modeling through Stochastic Differential Equationscited by 9,122
- 2022Scalable Diffusion Models with Transformerscited by 6,054
- 2022Classifier-Free Diffusion Guidancecited by 5,847
- 2017Neural Discrete Representation Learningcited by 4,712
- 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,703
- 2023SDXL: Improving Latent Diffusion Models for High-Resolution Image Synthesiscited by 4,356
- 2013Auto-Encoding Variational Bayescited by 4,053
32 more
- 2022Flow Straight and Fast: Learning to Generate and Transfer Data with Rectified Flowcited by 3,700
- 2022Elucidating the Design Space of Diffusion-Based Generative Modelscited by 3,045
- 2016Improved Techniques for Training GANscited by 2,970
- 2015Neural Machine Translation of Rare Words with Subword Unitscited by 2,918
- 2021CLIPScore: A Reference-free Evaluation Metric for Image Captioningcited by 2,424
- 2022DPM-Solver: A Fast ODE Solver for Diffusion Probabilistic Model Sampling in Around 10 Stepscited by 1,979
- 2023Consistency Modelscited by 1,822
- 2017Quantization and Training of Neural Networks for Efficient Integer-Arithmetic-Only Inferencecited by 1,593
- 2022Null-text Inversion for Editing Real Images using Guided Diffusion Modelscited by 1,179
- 2022MaskGIT: Masked Generative Image Transformercited by 1,099
- 2023GenEval: An Object-Focused Framework for Evaluating Text-to-Image Alignmentcited by 972
- 2022DPM-Solver++: Fast Solver for Guided Sampling of Diffusion Probabilistic Modelscited by 938
- 2024Gemma: Open Models Based on Gemini Research and Technologycited by 916
- 2023Human Preference Score v2: A Solid Benchmark for Evaluating Human Preferences of Text-to-Image Synthesiscited by 890
- 2024Autoregressive Model Beats Diffusion: Llama for Scalable Image Generationcited by 777
- 2022On Distillation of Guided Diffusion Modelscited by 725
- 2023Muse: Text-To-Image Generation via Masked Generative Transformerscited by 648
- 2022Fast Sampling of Diffusion Models with Exponential Integratorcited by 532
- 2020Exploring Self-attention for Image Recognitioncited by 313
- 2023T2I-CompBench++: An Enhanced and Comprehensive Benchmark for Compositional Text-to-image Generationcited by 260
- 2023Token Merging for Fast Stable Diffusioncited by 224
- 2023DPM-Solver-v3: Improved Diffusion ODE Solver with Empirical Model Statisticscited by 161
- 2024Lumina-mGPT: Illuminate Flexible Photorealistic Text-to-Image Generation with Multimodal Generative Pretrainingcited by 150
- 2024Fluid: Scaling Autoregressive Text-to-image Generative Models with Continuous Tokenscited by 147
- 2024Q-DiT: Accurate Post-Training Quantization for Diffusion Transformerscited by 94
- 2023Minimizing Trajectory Curvature of ODE-based Generative Modelscited by 89
- 2024Not All Noises Are Created Equally:Diffusion Noise Selection and Optimizationcited by 71
- 2024Accelerating Auto-regressive Text-to-Image Generation with Training-free Speculative Jacobi Decodingcited by 66
- 2024Meissonic: Revitalizing Masked Generative Transformers for Efficient High-Resolution Text-to-Image Synthesiscited by 56
- 2024Alignment of Diffusion Models: Fundamentals, Challenges, and Futurecited by 35
- 2023Catch-Up Distillation: You Only Need to Train Once for Accelerating Samplingcited by 13
- 2023DiffuseExpand: Expanding dataset for 2D medical image segmentation using diffusion modelscited by 6
Built on by
Citing papers per year: 2024: 1; 2025: 2; 2026: 4.
7 papers.
- 2024Zigzag Diffusion Sampling: Diffusion Models Can Self-Improve via Self-Reflectioncited by 41
- 2025Effective and Efficient Masked Image Generation Modelscited by 13
- 2025EditMGT: Unleashing Potentials of Masked Generative Transformers in Image Editingcited by 12
- 2026Guidance Matters: Rethinking the Evaluation Pitfall for Text-to-Image Generationcited by 4
- 2026CRAFT: Aligning Diffusion Models with Fine-Tuning Is Easier Than You Thinkcited by 4
- 2026Accelerating Masked Image Generation by Learning Controlled Latent Dynamicscited by 0
- 2026Reflective Flow Sampling Enhancementcited by 0
- 2026CRAFT: Aligning Diffusion Models with Fine-Tuning Is Easier Than You Think19 Mar 2026
- 2026Reflective Flow Sampling Enhancement6 Mar 2026
- 2026Accelerating Masked Image Generation by Learning Controlled Latent Dynamics27 Feb 2026
- 2026Guidance Matters: Rethinking the Evaluation Pitfall for Text-to-Image Generation26 Feb 2026
- 2025EditMGT: Unleashing Potentials of Masked Generative Transformers in Image Editing12 Dec 2025
- 2025Effective and Efficient Masked Image Generation Models10 Mar 2025
- 2024Zigzag Diffusion Sampling: Diffusion Models Can Self-Improve via Self-Reflection14 Dec 2024
Similar, not cited
Close in wording, no citation either way
- 2023MAGVLT: Masked Generative Vision-and-Language Transformer
- 2025Muddit: Liberating Generation Beyond Text-to-Image with a Unified Discrete Diffusion Model
- 2024Simplified and Generalized Masked Diffusion for Discrete Data
- 2024Simple and Effective Masked Diffusion Language Models
- 2026Nemotron-Labs-Diffusion-Image: Advancing Masked Discrete Diffusion for High-Resolution Image Synthesis
- 2025UNCAGE: Contrastive Attention Guidance for Masked Generative Transformers in Text-to-Image Generation
- 2026Training-Free Self-Correction for Multimodal Masked Diffusion Models
- 2024MaskMamba: A Hybrid Mamba-Transformer Model for Masked Image Generation