Saved in:
Bibliographic Details
Main Authors: Wang, Jiepeng, Wang, Zhaoqing, Pan, Hao, Liu, Yuan, Yu, Dongdong, Wang, Changhu, Wang, Wenping
Format: Preprint
Published: 2025
Subjects:
Online Access:https://arxiv.org/abs/2503.20644
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866913759953420288
author Wang, Jiepeng
Wang, Zhaoqing
Pan, Hao
Liu, Yuan
Yu, Dongdong
Wang, Changhu
Wang, Wenping
author_facet Wang, Jiepeng
Wang, Zhaoqing
Pan, Hao
Liu, Yuan
Yu, Dongdong
Wang, Changhu
Wang, Wenping
contents A unified diffusion framework for multi-modal generation and understanding has the transformative potential to achieve seamless and controllable image diffusion and other cross-modal tasks. In this paper, we introduce MMGen, a unified framework that integrates multiple generative tasks into a single diffusion model. This includes: (1) multi-modal category-conditioned generation, where multi-modal outputs are generated simultaneously through a single inference process, given category information; (2) multi-modal visual understanding, which accurately predicts depth, surface normals, and segmentation maps from RGB images; and (3) multi-modal conditioned generation, which produces corresponding RGB images based on specific modality conditions and other aligned modalities. Our approach develops a novel diffusion transformer that flexibly supports multi-modal output, along with a simple modality-decoupling strategy to unify various tasks. Extensive experiments and applications demonstrate the effectiveness and superiority of MMGen across diverse tasks and conditions, highlighting its potential for applications that require simultaneous generation and understanding.
format Preprint
id arxiv_https___arxiv_org_abs_2503_20644
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle MMGen: Unified Multi-modal Image Generation and Understanding in One Go
Wang, Jiepeng
Wang, Zhaoqing
Pan, Hao
Liu, Yuan
Yu, Dongdong
Wang, Changhu
Wang, Wenping
Computer Vision and Pattern Recognition
A unified diffusion framework for multi-modal generation and understanding has the transformative potential to achieve seamless and controllable image diffusion and other cross-modal tasks. In this paper, we introduce MMGen, a unified framework that integrates multiple generative tasks into a single diffusion model. This includes: (1) multi-modal category-conditioned generation, where multi-modal outputs are generated simultaneously through a single inference process, given category information; (2) multi-modal visual understanding, which accurately predicts depth, surface normals, and segmentation maps from RGB images; and (3) multi-modal conditioned generation, which produces corresponding RGB images based on specific modality conditions and other aligned modalities. Our approach develops a novel diffusion transformer that flexibly supports multi-modal output, along with a simple modality-decoupling strategy to unify various tasks. Extensive experiments and applications demonstrate the effectiveness and superiority of MMGen across diverse tasks and conditions, highlighting its potential for applications that require simultaneous generation and understanding.
title MMGen: Unified Multi-modal Image Generation and Understanding in One Go
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2503.20644