Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

EditAR: Unified Conditional Generation with Autoregressive Models

About

Recent progress in controllable image generation and editing is largely driven by diffusion-based methods. Although diffusion models perform exceptionally well in specific tasks with tailored designs, establishing a unified model is still challenging. In contrast, autoregressive models inherently feature a unified tokenized representation, which simplifies the creation of a single foundational model for various tasks. In this work, we propose EditAR, a single unified autoregressive framework for a variety of conditional image generation tasks, e.g., image editing, depth-to-image, edge-to-image, segmentation-to-image. The model takes both images and instructions as inputs, and predicts the edited images tokens in a vanilla next-token paradigm. To enhance the text-to-image alignment, we further propose to distill the knowledge from foundation models into the autoregressive modeling process. We evaluate its effectiveness across diverse tasks on established benchmarks, showing competitive performance to various state-of-the-art task-specific methods. Project page: https://jitengmu.github.io/EditAR/

Jiteng Mu, Nuno Vasconcelos, Xiaolong Wang• 2025

Related benchmarks

TaskDatasetResultRank
Image EditingPIE-Bench
PSNR21.32
116
Instructive image editingMagicBrush (test)
CLIP Image0.867
20
MRI Contrast EnhancementBreast DCE-MRI (test)
PSNR19.85
8
MRI Contrast EnhancementAbdominal DCE-MRI (test)
PSNR22.65
8
MRI Contrast EnhancementAbdominal and Breast DCE-MRI Average (test)
Avg. SSIM0.487
8
Depth-to-image translationMultiGen Depth Map
RMSE34.93
6
Edge-to-image translationMultiGen Canny Edge
SSIM48.11
6
Segmentation-to-image translationCOCOStuff Seg. Mask
mIOU22.62
3
Showing 8 of 8 rows

Other info

Code

Follow for update