mlsanityData to 14 Oct 2026

Rethinking UMM Visual Generation: Masked Modeling for Efficient Image-Only Pre-training

arXiv 2603.16139 ↗First posted 17 Mar 2026cs.CVcited by 2

Not on the map. Why?

Abstract (arXiv)

Unified Multimodal Models (UMMs) are often constrained by the pre-training of their visual generation components, which typically relies on inefficient paradigms and scarce, high-quality text-image paired data. In this paper, we systematically analyze pre-training recipes for UMM visual generation and identify these two issues as the major bottlenecks. To address them, we propose Image-Only Training for UMMs (IOMM), a data-efficient two-stage training framework. The first stage pre-trains the visual generative component exclusively using abundant unlabeled image-only data, thereby removing the dependency on paired data for this costly phase. The second stage fine-tunes the model using a mixture of unlabeled images and a small curated set of text-image pairs, leading to improved instruction alignment and generative quality. Extensive experiments show that IOMM not only improves training efficiency but also achieves state-of-the-art (SOTA) performance. For example, our IOMM-B (3.6B) model was trained from scratch using only ~ 1050 H800 GPU hours (with the vast majority, 1000 hours, dedicated to the efficient image-only pre-training stage). It achieves 0.89 on GenEval and 0.55 on WISE--surpassing strong baselines such as BAGEL-7B (0.82 & 0.55) and BLIP3-o-4B (0.84 & 0.50). Code is available hrefhttps://github.com/LINs-lab/IOMMhttps://github.com/LINs-lab/IOMM.

Key facts. Cited by 2 papers in the map (data through 14 Oct 2026). Builds on 52 of its 71 references that are in the map. First posted on arXiv 17 Mar 2026.

Builds on

52 of 71 references.

References · 52

  1. 2017Attention Is All You Needcited by 60,361
  2. 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,305
  3. 2020Score-Based Generative Modeling through Stochastic Differential Equationscited by 9,122
  4. 2021Masked Autoencoders Are Scalable Vision Learnerscited by 7,403
  5. 2022Hierarchical Text-Conditional Image Generation with CLIP Latentscited by 6,585
  6. 2022Flow Matching for Generative Modelingcited by 5,702
  7. 2023SDXL: Improving Latent Diffusion Models for High-Resolution Image Synthesiscited by 4,356
  8. 2024Scaling Rectified Flow Transformers for High-Resolution Image Synthesiscited by 4,181
  9. 2022InstructPix2Pix: Learning to Follow Image Editing Instructionscited by 2,611
  10. 2025InternVL3: Exploring Advanced Training and Test-Time Recipes for Open-Source Multimodal Modelscited by 1,834
  11. 2022MaskGIT: Masked Generative Image Transformercited by 1,099
  12. 2025FLUX.1 Kontext: Flow Matching for In-Context Image Generation and Editing in Latent Spacecited by 1,006
38 more
  1. 2023GenEval: An Object-Focused Framework for Evaluating Text-to-Image Alignmentcited by 972
  2. 2024Chameleon: Mixed-Modal Early-Fusion Foundation Modelscited by 927
  3. 2025Emerging Properties in Unified Multimodal Pretrainingcited by 897
  4. 2023PixArt-α: Fast Training of Diffusion Transformer for Photorealistic Text-to-Image Synthesiscited by 864
  5. 2025Janus-Pro: Unified Multimodal Understanding and Generation with Data and Model Scalingcited by 812
  6. 2024Show-o: One Single Transformer to Unify Multimodal Understanding and Generationcited by 752
  7. 2024SiT: Exploring Flow and Diffusion-based Generative Models with Scalable Interpolant Transformerscited by 737
  8. 2024Representation Alignment for Generation: Training Diffusion Transformers Is Easier Than You Thinkcited by 703
  9. 2023MagicBrush: A Manually Annotated Dataset for Instruction-Guided Image Editingcited by 624
  10. 2024ELLA: Equip Diffusion Models with LLM for Enhanced Semantic Alignmentcited by 475
  11. 2024Janus: Decoupling Visual Encoding for Unified Multimodal Understanding and Generationcited by 457
  12. 2025Step1X-Edit: A Practical Framework for General Image Editingcited by 429
  13. 2024OmniGen: Unified Image Generationcited by 394
  14. 2025BLIP3-o: A Family of Fully Open Unified Multimodal Models-Architecture, Training and Datasetcited by 376
  15. 2024SANA: Efficient High-Resolution Image Synthesis with Linear Diffusion Transformerscited by 340
  16. 2024PixArt-Sigma: Weak-to-Strong Training of Diffusion Transformer for 4K Text-to-Image Generationcited by 329
  17. 2023DreamLLM: Synergistic Multimodal Comprehension and Creationcited by 297
  18. 2024VILA-U: a Unified Foundation Model Integrating Visual Understanding and Generationcited by 295
  19. 2025Z-Image: An Efficient Image Generation Foundation Model with Single-Stream Diffusion Transformercited by 277
  20. 2024UltraEdit: Instruction-based Fine-Grained Image Editing at Scalecited by 268
  21. 2024Playground v2.5: Three Insights towards Enhancing Aesthetic Quality in Text-to-Image Generationcited by 256
  22. 2025Show-o2: Improved Native Unified Multimodal Modelscited by 250
  23. 2024AnyEdit: Mastering Unified High-Quality Image Editing for Any Ideacited by 241
  24. 2025Transfer between Modalities with MetaQueriescited by 236
  25. 2025ImgEdit: A Unified Image Editing Dataset and Benchmarkcited by 231
  26. 2025WISE: A World Knowledge-Informed Semantic Evaluation for Text-to-Image Generationcited by 204
  27. 2025In-Context Edit: Enabling Instructional Image Editing with In-Context Generation in Large Scale Diffusion Transformercited by 204
  28. 2024JanusFlow: Harmonizing Autoregression and Rectified Flow for Unified Multimodal Understanding and Generationcited by 163
  29. 2025Echo-4o: Harnessing the Power of GPT-4o Synthetic Images for Improved Image Generationcited by 100
  30. 2025ShareGPT-4o-Image: Aligning Multimodal Models with GPT-4o-Level Image Generationcited by 97
  31. 2025SANA-Sprint: One-Step Diffusion with Continuous-Time Consistency Distillationcited by 87
  32. 2025Harmonizing Visual Representations for Unified Multimodal Understanding and Generationcited by 74
  33. 2025UniWorld-V1: High-Resolution Semantic Encoders for Unified Visual Understanding and Generationcited by 66
  34. 2025Reconstruction Alignment Improves Unified Multimodal Modelscited by 49
  35. 2025OpenUni: A Simple Baseline for Unified Multimodal Understanding and Generationcited by 46
  36. 2025Unified Continuous Generative Modelscited by 19
  37. 2023MaskDiffusion: Boosting Text-to-Image Consistency with Conditional Maskcited by 17
  38. 2024Towards Efficient Diffusion-Based Image Editing with Instant Attention Maskscited by 16

50 most cited of 52 All 52 (JSON)

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 2.

2 papers.