aMUSEd: An Open MUSE Reproduction
On the map: Diffusion models → Image generation, text-to-image and editing
Abstract (arXiv)
We present aMUSEd, an open-source, lightweight masked image model (MIM) for text-to-image generation based on MUSE. With 10 percent of MUSE's parameters, aMUSEd is focused on fast image generation. We believe MIM is under-explored compared to latent diffusion, the prevailing approach for text-to-image generation. Compared to latent diffusion, MIM requires fewer inference steps and is more interpretable. Additionally, MIM can be fine-tuned to learn additional styles with only a single image. We hope to encourage further exploration of MIM by demonstrating its effectiveness on large-scale text-to-image generation and releasing reproducible training code. We also release checkpoints for two models which directly produce images at 256x256 and 512x512 resolutions.
Key facts. Cited by 32 papers in the map (data through 7 Oct 2026). Placed in Image generation, text-to-image and editing, one of 14 directions in Diffusion models. Builds on 32 of its 41 references that are in the map. First posted on arXiv 3 Jan 2024.
Builds on
32 of 41 references.
Closest ancestors · 1
Same direction, newest first
Other references · 31
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,204
- 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,456
- 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,604
- 2021High-Resolution Image Synthesis with Latent Diffusion Modelscited by 18,694
- 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,491
- 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,084
- 2020Denoising Diffusion Implicit Modelscited by 9,655
- 2020Score-Based Generative Modeling through Stochastic Differential Equationscited by 8,964
- 2022Scalable Diffusion Models with Transformerscited by 5,931
- 2017On Calibration of Modern Neural Networkscited by 4,860
- 2023SDXL: Improving Latent Diffusion Models for High-Resolution Image Synthesiscited by 4,290
- 2021Zero-Shot Text-to-Image Generationcited by 4,276
19 more
- 2022LAION-5B: An open large-scale dataset for training next generation image-text modelscited by 4,131
- 2020Taming Transformers for High-Resolution Image Synthesiscited by 3,229
- 2022Elucidating the Design Space of Diffusion-Based Generative Modelscited by 2,994
- 2019The Curious Case of Neural Text Degenerationcited by 2,881
- 2017FiLM: Visual Reasoning with a General Conditioning Layercited by 2,286
- 2022DPM-Solver: A Fast ODE Solver for Diffusion Probabilistic Model Sampling in Around 10 Stepscited by 1,961
- 2018Hierarchical Neural Story Generationcited by 1,182
- 2019An Introduction to Variational Autoencoderscited by 997
- 2022DPM-Solver++: Fast Solver for Guided Sampling of Diffusion Probabilistic Modelscited by 924
- 2023PixArt-α: Fast Training of Diffusion Transformer for Photorealistic Text-to-Image Synthesiscited by 854
- 2023Latent Consistency Models: Synthesizing High-Resolution Images with Few-Step Inferencecited by 826
- 2023Language Is Not All You Need: Aligning Perception with Language Modelscited by 543
- 2022Fast Sampling of Diffusion Models with Exponential Integratorcited by 526
- 2023UniPC: A Unified Predictor-Corrector Framework for Fast Sampling of Diffusion Modelscited by 441
- 2020How Can We Know When Language Models Know? On the Calibration of Language Models for Question Answeringcited by 439
- 2023DPOK: Reinforcement Learning for Fine-tuning Text-to-Image Diffusion Modelscited by 431
- 2023LCM-LoRA: A Universal Stable-Diffusion Acceleration Modulecited by 211
- 2023DPM-Solver-v3: Improved Diffusion ODE Solver with Empirical Model Statisticscited by 161
- 2022CM3: A Causal Masked Multimodal Model of the Internetcited by 142
Built on by
Citing papers per year: 2024: 15; 2025: 14; 2026: 3.
32 papers.
Sort:
By area:
- 2024World Model on Million-Length Video And Language With Blockwise RingAttentioncited by 196
- 2024iVideoGPT: Interactive VideoGPTs are Scalable World Modelscited by 136
- 2025MineWorld: a Real-Time and Open-Source Interactive World Model on Minecraftcited by 93
- 2024WorldDreamer: Towards General World Models for Video Generation via Predicting Masked Tokenscited by 87
- 2025Lumina-DiMOO: An Omni Diffusion Large Language Model for Multi-Modal Generation and Understandingcited by 83
- 2025A Survey on Diffusion Language Modelscited by 75
- 2024Continuous, Subject-Specific Attribute Control in T2I Models by Identifying Semantic Directionscited by 39
- 2024VidTok: A Versatile and Open-Source Video Tokenizercited by 35
- 2025Halton Scheduler For Masked Generative Image Transformercited by 30
- 2024HumanEdit: A High-Quality Human-Rewarded Dataset for Instruction-based Image Editingcited by 29
- 2024Contrasting Deepfakes Diffusion via Contrastive Learning and Global-Local Similaritiescited by 27
- 2024Next Token Prediction Towards Multimodal Intelligence: A Comprehensive Surveycited by 27
20 more
- 2025EditMGT: Unleashing Potentials of Masked Generative Transformers in Image Editingcited by 12
- 2025Stay-Positive: A Case for Ignoring Real Image Features in Fake Image Detectioncited by 11
- 2024Video In-context Learning: Autoregressive Transformers are Zero-Shot Video Imitatorscited by 9
- 2025Fast Autoregressive Video Generation with Diagonal Decodingcited by 8
- 2024Generating Coherent Sequences of Visual Illustrations for Real-World Manual Taskscited by 6
- 2025Dimathtt[M]O: Distilling Masked Diffusion Models into One-step Generatorcited by 5
- 2025NEP: Autoregressive Image Editing via Next Editing Token Predictioncited by 5
- 2025Soft-Di[M]O: Improving One-Step Discrete Image Generation with Soft Embeddingscited by 5
- 2024TMT: Tri-Modal Translation between Speech, Image, and Text by Processing Different Modalities as Different Languagescited by 4
- 2025Training Data Provenance Verification: Did Your Model Use Synthetic Data from My Generative Model for Training?cited by 4
- 2025dMLLM-TTS: Self-Verified and Efficient Test-Time Scaling for Diffusion Multi-Modal Large Language Modelscited by 2
- 2024DECOR:Decomposition and Projection of Text Embeddings for Text-to-Image Customizationcited by 1
- 2025Accelerating Inference of Masked Image Generators via Reinforcement Learningcited by 1
- 2024Controllable Image Generation with Composed Parallel Token Predictioncited by 0
- 2024Sustainable techniques to improve Data Quality for training image-based explanatory models for Recommender Systemscited by 0
- 2024Multi-Style Facial Sketch Synthesis through Masked Generative Modelingcited by 0
- 2025Deconstructing Generative Diversity: An Information Bottleneck Analysis of Discrete Latent Generative Modelscited by 0
- 2026Controllable Image Generation with Composed Parallel Token Predictioncited by 0
- 2026Dissect and Prune: Enhancing Robustness in AI-Generated Image Detectioncited by 0
- 2026Persistent Watermarking of Text-to-Image Modelscited by 0
- 2026Persistent Watermarking of Text-to-Image Models30 Sep 2026
- 2026Dissect and Prune: Enhancing Robustness in AI-Generated Image Detection9 Jun 2026
- 2026Controllable Image Generation with Composed Parallel Token Prediction7 Apr 2026
- 2025dMLLM-TTS: Self-Verified and Efficient Test-Time Scaling for Diffusion Multi-Modal Large Language Models22 Dec 2025
- 2025EditMGT: Unleashing Potentials of Masked Generative Transformers in Image Editing12 Dec 2025
- 2025Deconstructing Generative Diversity: An Information Bottleneck Analysis of Discrete Latent Generative Models1 Dec 2025
- 2025Accelerating Inference of Masked Image Generators via Reinforcement Learning30 Nov 2025
- 2025Lumina-DiMOO: An Omni Diffusion Large Language Model for Multi-Modal Generation and Understanding7 Oct 2025
- 2025Soft-Di[M]O: Improving One-Step Discrete Image Generation with Soft Embeddings26 Sep 2025
- 2025A Survey on Diffusion Language Models14 Aug 2025
- 2025NEP: Autoregressive Image Editing via Next Editing Token Prediction8 Aug 2025
- 2025MineWorld: a Real-Time and Open-Source Interactive World Model on Minecraft11 Apr 2025
20 more
- 2025Halton Scheduler For Masked Generative Image Transformer21 Mar 2025
- 2025Dimathtt[M]O: Distilling Masked Diffusion Models into One-step Generator19 Mar 2025
- 2025Fast Autoregressive Video Generation with Diagonal Decoding18 Mar 2025
- 2025Training Data Provenance Verification: Did Your Model Use Synthetic Data from My Generative Model for Training?12 Mar 2025
- 2025Stay-Positive: A Case for Ignoring Real Image Features in Fake Image Detection11 Feb 2025
- 2024VidTok: A Versatile and Open-Source Video Tokenizer17 Dec 2024
- 2024Next Token Prediction Towards Multimodal Intelligence: A Comprehensive Survey16 Dec 2024
- 2024DECOR:Decomposition and Projection of Text Embeddings for Text-to-Image Customization12 Dec 2024
- 2024HumanEdit: A High-Quality Human-Rewarded Dataset for Instruction-based Image Editing5 Dec 2024
- 2024Multi-Style Facial Sketch Synthesis through Masked Generative Modeling22 Aug 2024
- 2024Contrasting Deepfakes Diffusion via Contrastive Learning and Global-Local Similarities29 Jul 2024
- 2024Video In-context Learning: Autoregressive Transformers are Zero-Shot Video Imitators10 Jul 2024
- 2024Sustainable techniques to improve Data Quality for training image-based explanatory models for Recommender Systems9 Jul 2024
- 2024iVideoGPT: Interactive VideoGPTs are Scalable World Models24 May 2024
- 2024Generating Coherent Sequences of Visual Illustrations for Real-World Manual Tasks16 May 2024
- 2024Controllable Image Generation with Composed Parallel Token Prediction10 May 2024
- 2024Continuous, Subject-Specific Attribute Control in T2I Models by Identifying Semantic Directions25 Mar 2024
- 2024TMT: Tri-Modal Translation between Speech, Image, and Text by Processing Different Modalities as Different Languages25 Feb 2024
- 2024World Model on Million-Length Video And Language With Blockwise RingAttention13 Feb 2024
- 2024WorldDreamer: Towards General World Models for Video Generation via Predicting Masked Tokens18 Jan 2024
- 2025Lumina-DiMOO: An Omni Diffusion Large Language Model for Multi-Modal Generation and Understandingcited by 83
- 2025A Survey on Diffusion Language Modelscited by 75
- 2024Continuous, Subject-Specific Attribute Control in T2I Models by Identifying Semantic Directionscited by 39
- 2025EditMGT: Unleashing Potentials of Masked Generative Transformers in Image Editingcited by 12
- 2025Dimathtt[M]O: Distilling Masked Diffusion Models into One-step Generatorcited by 5
- 2025Soft-Di[M]O: Improving One-Step Discrete Image Generation with Soft Embeddingscited by 5
- 2025Training Data Provenance Verification: Did Your Model Use Synthetic Data from My Generative Model for Training?cited by 4
- 2025EditMGT: Unleashing Potentials of Masked Generative Transformers in Image Editing12 Dec 2025
- 2025Lumina-DiMOO: An Omni Diffusion Large Language Model for Multi-Modal Generation and Understanding7 Oct 2025
- 2025Soft-Di[M]O: Improving One-Step Discrete Image Generation with Soft Embeddings26 Sep 2025
- 2025A Survey on Diffusion Language Models14 Aug 2025
- 2025Dimathtt[M]O: Distilling Masked Diffusion Models into One-step Generator19 Mar 2025
- 2025Training Data Provenance Verification: Did Your Model Use Synthetic Data from My Generative Model for Training?12 Mar 2025
- 2024Continuous, Subject-Specific Attribute Control in T2I Models by Identifying Semantic Directions25 Mar 2024
- 2024World Model on Million-Length Video And Language With Blockwise RingAttentioncited by 196
- 2024iVideoGPT: Interactive VideoGPTs are Scalable World Modelscited by 136
- 2025MineWorld: a Real-Time and Open-Source Interactive World Model on Minecraftcited by 93
- 2024WorldDreamer: Towards General World Models for Video Generation via Predicting Masked Tokenscited by 87
- 2025MineWorld: a Real-Time and Open-Source Interactive World Model on Minecraft11 Apr 2025
- 2024iVideoGPT: Interactive VideoGPTs are Scalable World Models24 May 2024
- 2024World Model on Million-Length Video And Language With Blockwise RingAttention13 Feb 2024
- 2024WorldDreamer: Towards General World Models for Video Generation via Predicting Masked Tokens18 Jan 2024
- 2025Fast Autoregressive Video Generation with Diagonal Decoding18 Mar 2025
- 2026Persistent Watermarking of Text-to-Image Modelscited by 0
- 2026Persistent Watermarking of Text-to-Image Models30 Sep 2026
- 2024VidTok: A Versatile and Open-Source Video Tokenizercited by 35
- 2025Halton Scheduler For Masked Generative Image Transformercited by 30
- 2024HumanEdit: A High-Quality Human-Rewarded Dataset for Instruction-based Image Editingcited by 29
- 2024Contrasting Deepfakes Diffusion via Contrastive Learning and Global-Local Similaritiescited by 27
- 2024Next Token Prediction Towards Multimodal Intelligence: A Comprehensive Surveycited by 27
- 2025Stay-Positive: A Case for Ignoring Real Image Features in Fake Image Detectioncited by 11
- 2024Video In-context Learning: Autoregressive Transformers are Zero-Shot Video Imitatorscited by 9
- 2024Generating Coherent Sequences of Visual Illustrations for Real-World Manual Taskscited by 6
- 2025NEP: Autoregressive Image Editing via Next Editing Token Predictioncited by 5
- 2024TMT: Tri-Modal Translation between Speech, Image, and Text by Processing Different Modalities as Different Languagescited by 4
- 2024DECOR:Decomposition and Projection of Text Embeddings for Text-to-Image Customizationcited by 1
- 2025Accelerating Inference of Masked Image Generators via Reinforcement Learningcited by 1
12 most cited of 18 All 18 (JSON)
- 2026Dissect and Prune: Enhancing Robustness in AI-Generated Image Detection9 Jun 2026
- 2026Controllable Image Generation with Composed Parallel Token Prediction7 Apr 2026
- 2025Deconstructing Generative Diversity: An Information Bottleneck Analysis of Discrete Latent Generative Models1 Dec 2025
- 2025Accelerating Inference of Masked Image Generators via Reinforcement Learning30 Nov 2025
- 2025NEP: Autoregressive Image Editing via Next Editing Token Prediction8 Aug 2025
- 2025Halton Scheduler For Masked Generative Image Transformer21 Mar 2025
- 2025Stay-Positive: A Case for Ignoring Real Image Features in Fake Image Detection11 Feb 2025
- 2024VidTok: A Versatile and Open-Source Video Tokenizer17 Dec 2024
- 2024Next Token Prediction Towards Multimodal Intelligence: A Comprehensive Survey16 Dec 2024
- 2024DECOR:Decomposition and Projection of Text Embeddings for Text-to-Image Customization12 Dec 2024
- 2024HumanEdit: A High-Quality Human-Rewarded Dataset for Instruction-based Image Editing5 Dec 2024
- 2024Multi-Style Facial Sketch Synthesis through Masked Generative Modeling22 Aug 2024
12 newest of 18 All 18 (JSON)
Similar, not cited
Close in wording, no citation either way
- 2023Muse: Text-To-Image Generation via Masked Generative Transformers
- 2023RAPHAEL: Text-to-Image Generation via Large Mixture of Diffusion Paths
- 2026i1: A Simple and Fully Open Recipe for Strong Text-to-Image Models
- 2024Image Regeneration: Evaluating Text-to-Image Model via Generating Identical Image with Multimodal Large Language Models
- 2024Training-Free Sketch-Guided Diffusion with Latent Optimization
- 2026Na"ive PAINE: Lightweight Text-to-Image Generation Improvement with Prompt Evaluation
- 2026LGTM: Training-Free Light-Guided Text-to-Image Diffusion Model via Initial Noise Manipulation
- 2023Emu: Enhancing Image Generation Models Using Photogenic Needles in a Haystack