Effective and Efficient Masked Image Generation Models

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: You, Zebin, Ou, Jingyang, Zhang, Xiaolu, Hu, Jun, Zhou, Jun, Li, Chongxuan
Formato: Preprint
Publicado: 2025
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866908857370935296
author You, Zebin
Ou, Jingyang
Zhang, Xiaolu
Hu, Jun
Zhou, Jun
Li, Chongxuan
author_facet You, Zebin
Ou, Jingyang
Zhang, Xiaolu
Hu, Jun
Zhou, Jun
Li, Chongxuan
contents Although masked image generation models and masked diffusion models are designed with different motivations and objectives, we observe that they can be unified within a single framework. Building upon this insight, we carefully explore the design space of training and sampling, identifying key factors that contribute to both performance and efficiency. Based on the improvements observed during this exploration, we develop our model, referred to as \textbf{eMIGM}. Empirically, eMIGM demonstrates strong performance on ImageNet generation, as measured by Fréchet Inception Distance (FID). In particular, on ImageNet $256\times256$, with similar number of function evaluations (NFEs) and model parameters, eMIGM outperforms the seminal VAR. Moreover, as NFE and model parameters increase, eMIGM achieves performance comparable to the state-of-the-art continuous diffusion model REPA while requiring less than 45\% of the NFE. Additionally, on ImageNet $512\times512$, eMIGM outperforms the strong continuous diffusion model EDM2. Code is available at https://github.com/ML-GSAI/eMIGM.
format Preprint
id arxiv_https___arxiv_org_abs_2503_07197
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Effective and Efficient Masked Image Generation Models
You, Zebin
Ou, Jingyang
Zhang, Xiaolu
Hu, Jun
Zhou, Jun
Li, Chongxuan
Computer Vision and Pattern Recognition
Machine Learning
Although masked image generation models and masked diffusion models are designed with different motivations and objectives, we observe that they can be unified within a single framework. Building upon this insight, we carefully explore the design space of training and sampling, identifying key factors that contribute to both performance and efficiency. Based on the improvements observed during this exploration, we develop our model, referred to as \textbf{eMIGM}. Empirically, eMIGM demonstrates strong performance on ImageNet generation, as measured by Fréchet Inception Distance (FID). In particular, on ImageNet $256\times256$, with similar number of function evaluations (NFEs) and model parameters, eMIGM outperforms the seminal VAR. Moreover, as NFE and model parameters increase, eMIGM achieves performance comparable to the state-of-the-art continuous diffusion model REPA while requiring less than 45\% of the NFE. Additionally, on ImageNet $512\times512$, eMIGM outperforms the strong continuous diffusion model EDM2. Code is available at https://github.com/ML-GSAI/eMIGM.
title Effective and Efficient Masked Image Generation Models
topic Computer Vision and Pattern Recognition
Machine Learning
url https://arxiv.org/abs/2503.07197