Rethinking UMM Visual Generation: Masked Modeling for Efficient Image-Only Pre-training
Organizations: Zhejiang University · Westlake University · Shanghai Innovation Institute
Abstract
Unified Multimodal Models (UMMs) are often constrained by the pre-training of their , which typically relies on inefficient paradigms and scarce, high-quality text-image paired data. In this paper, we systematically analyze pre-training recipes for and identify these two issues as the major bottlenecks. To address them, we propose , a data-efficient two-stage training framework. The first stage pre-trains the visual generative component using abundant unlabeled image-only data, thereby removing the dependency on paired data . 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 H800 GPU hours (with the vast majority, hours, dedicated to the efficient ). It achieves on GenEval and on WISE--surpassing strong baselines such as BAGEL-7B (0.82 & 0.55) and BLIP3-o-4B (0.84 & 0.50). Code is available https://github.com/LINs-lab/IOMM.
Figures & tables
| METHOD | GenEval | DPGBench ( ) | WISE ( ) | ||||||
|---|---|---|---|---|---|---|---|---|---|
| Single Obj. | Two Obj. | Counting | Colors | Position | Color Attri. | Overall ( ) | |||
| Gen. Only | |||||||||
| SDv1.5 [ 42 ] | 0.97 | 0.38 | 0.35 | 0.76 | 0.04 | 0.06 | 0.43 | 63.18 | 0.32 |
| SDv2.1 [ 42 ] | 0.98 | 0.51 | 0.44 | 0.85 | 0.07 | 0.17 | 0.50 | - | 0.32 |
| SD3-Medium [ 14 ] | 0.99 | 0.94 | 0.72 | 0.89 | 0.33 | 0.60 | 0.74 | 84.08 | 0.42 |
| SDXL [ 40 ] | 0.98 | 0.74 | 0.39 | 0.85 | 0.15 | 0.23 | 0.55 | 74.65 | 0.43 |
| METHOD | Res. | NFE | GenEval | WISE ( ) | ||||||
|---|---|---|---|---|---|---|---|---|---|---|
| Single Obj. | Two Obj. | Counting | Colors | Position | Color Attri. | Overall ( ) | ||||
| OpenUni-L [ 50 ] | 512 | 20 2 | 0.99 | 0.91 | 0.77 | 0.90 | 0.75 | 0.76 | 0.85 | 0.52 |
| \boldsymbol{\color[rgb]{0.7,0,0}\oplus} Image finetuning | 512 | 20 2 | 1.00 ↑0.01 | 0.98 ↑0.07 | 0.22 ↓0.55 | 0.91 ↑0.01 | 0.60 ↓0.15 | 0.77 ↑0.01 | 0.74 ↓0.11 | 0.49 ↓0.03 |
| \boldsymbol{\color[rgb]{0.7,0,0}\oplus} Pair finetuning | 512 | 20 2 | 0.99 | 0.94 ↑0.03 | 0.82 ↑0.05 | 0.91 ↑0.01 | 0.85 ↑0.10 | 0.76 | 0.88 ↑0.03 | 0.62 ↑0.10 |
| \boldsymbol{\color[rgb]{0.7,0,0}\oplus} Mix finetuning | 512 | 20 2 | 0.99 | 0.91 | 0.78 ↑0.01 | 0.93 ↑0.03 | 0.87 ↑0.12 | 0.78 ↑0.02 | 0.88 ↑0.03 | 0.59 ↑0.07 |
| Qwen-Image [ 48 ] | 512 | 50 2 | 0.99 | 0.91 | 0.87 | 0.88 | 0.73 | 0.74 | 0.85 | - |
| METHOD | ImgEdit-Bench | |||||||||
|---|---|---|---|---|---|---|---|---|---|---|
| Add | Adjust | Extract | Replace | Remove | Background | Style | Hybrid | Action | Overall ( ) | |
| Trained with editing data | ||||||||||
| MagicBrush [ 63 ] | 2.84 | 1.58 | 1.51 | 1.97 | 1.58 | 1.75 | 2.38 | 1.62 | 1.22 | 1.90 |
| Instruct-Pix2Pix [ 3 ] | 2.45 | 1.83 | 1.44 | 2.01 | 1.50 | 1.44 | 3.55 | 1.20 | 1.46 | 1.88 |
| AnyEdit [ 61 ] | 3.18 | 2.95 | 1.88 | 2.47 | 2.23 | 2.24 | 2.85 | 1.56 | 2.65 | 2.45 |
| UltraEdit [ 65 ] | 3.44 | 2.81 | 2.13 | 2.96 | 1.45 | 2.83 | 3.76 | 1.91 | 2.98 | 2.70 |
Appendix figures & tables10 assets
Supplementary material from the paper’s appendix.
Appendix
| METHOD | IOMM -B | IOMM -L | IOMM -XL |
|---|---|---|---|
| Optimization | |||
| Optimizer | AdamW | Muon | |
| Learning rate | 1e-4 | 1e-4 | |
| Max gradient norm | 1.0 | 1.0 | |
| Weight decay | 0.0 | 0.0 | |
| METHOD | IOMM -B | IOMM -L | IOMM -XL | ||
| Resolution | 512 | 1024 | 512 | 1024 | 512 |
| Optimization | |||||
| Optimizer | AdamW | AdamW | Muon | ||
| Learning rate | 1e-4 | 1e-4 | 1e-4 | ||
| Max gradient norm | 1.0 | 1.0 | 1.0 | ||
| METHOD | OpenUni-L | Qwen-Image-20B |
|---|---|---|
| Optimization | ||
| Optimizer | AdamW | AdamW |
| (0.9,0.95) | (0.9,0.95) | |
| Learning rate | 1e-4 | 1e-4 |
| Max gradient norm | 1.0 | 1.0 |
| Weight decay | 0.0 | 0.0 |
| METHOD | Global | Entity | Attribute | Relation | Other | Overall |
|---|---|---|---|---|---|---|
| Gen. Only | ||||||
| SDv1.5 [ 42 ] | 74.63 | 74.23 | 75.39 | 73.49 | 67.81 | 63.18 |
| SD3-Medium [ 14 ] | 87.90 | 91.01 | 88.83 | 80.70 | 88.68 | 84.08 |
| SDXL [ 40 ] | 83.27 | 82.43 | 80.91 | 86.76 | 80.41 | 74.65 |
| PixArt- [ 9 ] | 74.97 | 79.32 | 78.60 | 82.57 | 76.96 | 71.11 |
| FLUX.1-dev [ 2 ] | 74.35 | 90.00 | 88.96 | 90.87 | 88.33 | 83.84 |
| METHOD | Cultural | Time | Space | Biology | Physics | Chemistry | Overall |
|---|---|---|---|---|---|---|---|
| Gen. Only | |||||||
| SDv1.5 [ 42 ] | 0.34 | 0.35 | 0.32 | 0.28 | 0.29 | 0.21 | 0.32 |
| SDv2.1 [ 42 ] | 0.30 | 0.38 | 0.35 | 0.33 | 0.34 | 0.21 | 0.32 |
| SD3-Medium [ 14 ] | 0.42 | 0.44 | 0.48 | 0.39 | 0.47 | 0.29 | 0.42 |
| SDXL [ 40 ] | 0.43 | 0.48 | 0.47 | 0.44 | 0.45 | 0.27 | 0.43 |
| SD3.5-Large [ 14 ] | 0.44 | 0.50 | 0.58 | 0.44 | 0.52 | 0.31 | 0.46 |
| Finetuning Recipe | Single Obj. | Two Obj. | Counting | Colors | Position | Color Attri. | Overall ( ) |
|---|---|---|---|---|---|---|---|
| Pre-trained with Text-Image Pair Data | |||||||
| Image | 1.00 | 0.95 | 0.63 | 0.87 | 0.50 | 0.72 | 0.78 |
| Pair | 0.99 | 0.92 | 0.76 | 0.91 | 0.87 | 0.69 | 0.86 |
| Mix | 0.99 | 0.91 | 0.80 | 0.92 | 0.90 | 0.75 | 0.88 |
| Pre-trained with Image-Only Data | |||||||
| Image | 0.99 | 0.84 | 0.24 | 0.75 | 0.37 | 0.45 | 0.61 |
| METHOD | Res. | NFEs | Cultural | Time | Space | Biology | Physics | Chemistry | Overall |
|---|---|---|---|---|---|---|---|---|---|
| OpenUni-L [ 50 ] | 512 | 20 2 | 0.51 | 0.45 | 0.58 | 0.39 | 0.50 | 0.30 | 0.52 |
| \boldsymbol{\color[rgb]{0.7,0,0}\oplus} Image finetuning | 512 | 20 2 | 0.46 | 0.52 | 0.66 | 0.49 | 0.51 | 0.29 | 0.49 |
| \boldsymbol{\color[rgb]{0.7,0,0}\oplus} Pair finetuning | 512 | 20 2 | 0.63 | 0.58 | 0.74 | 0.57 | 0.71 | 0.44 | 0.62 |
| \boldsymbol{\color[rgb]{0.7,0,0}\oplus} Mix finetuning | 512 | 20 2 | 0.60 | 0.58 | 0.70 | 0.51 | 0.64 | 0.46 | 0.59 |
| Qwen-Image [ 48 ] | 512 | 50 2 | - | - | - | - | - | - | - |
| \boldsymbol{\color[rgb]{0.7,0,0}\oplus} Image finetuning | 512 | 50 2 | 0.39 | 0.42 | 0.56 | 0.32 | 0.50 | 0.28 | 0.41 |
Unified Multimodal Models
Unified multimodal models combine multiple data types in shared systems for cross-modal understanding, generation, reasoning, and interaction.
See all 290 papers