English

Reconstruction Alignment Improves Unified Multimodal Models

Computer Vision and Pattern Recognition 2025-10-28 v3 Artificial Intelligence Machine Learning

Abstract

Unified multimodal models (UMMs) unify visual understanding and generation within a single architecture. However, conventional training relies on image-text pairs (or sequences) whose captions are typically sparse and miss fine-grained visual details--even when they use hundreds of words to describe a simple image. We introduce Reconstruction Alignment (RecA), a resource-efficient post-training method that leverages visual understanding encoder embeddings as dense "text prompts," providing rich supervision without captions. Concretely, RecA conditions a UMM on its own visual understanding embeddings and optimizes it to reconstruct the input image with a self-supervised reconstruction loss, thereby realigning understanding and generation. Despite its simplicity, RecA is broadly applicable: across autoregressive, masked-autoregressive, and diffusion-based UMMs, it consistently improves generation and editing fidelity. With only 27 GPU-hours, post-training with RecA substantially improves image generation performance on GenEval (0.73\rightarrow0.90) and DPGBench (80.93\rightarrow88.15), while also boosting editing benchmarks (ImgEdit 3.38\rightarrow3.75, GEdit 6.94\rightarrow7.25). Notably, RecA surpasses much larger open-source models and applies broadly across diverse UMM architectures, establishing it as an efficient and general post-training alignment strategy for UMMs

Keywords

Cite

@article{arxiv.2509.07295,
  title  = {Reconstruction Alignment Improves Unified Multimodal Models},
  author = {Ji Xie and Trevor Darrell and Luke Zettlemoyer and XuDong Wang},
  journal= {arXiv preprint arXiv:2509.07295},
  year   = {2025}
}

Comments

34 pages, 28 figures and 11 tables; Update ablation study

R2 v1 2026-07-01T05:27:36.045Z