MADFormer: Mixed Autoregressive and Diffusion Transformers for Continuous Image Generation

Junhao Chen, Yulia Tsvetkov, Xiaochuang Han
6/9/2025
cs.CVcs.LG

No official code found. Best implementation: Junhaoo-Chen/MADFormer (7 stars, Apache-2.0, last commit 2025-06-24).

Abstract

Recent progress in multimodal generation has increasingly combined autoregressive (AR) and diffusion-based approaches, leveraging their complementary strengths: AR models capture long-range dependencies and produce fluent, context-aware outputs, while diffusion models operate in continuous latent spaces to refine high-fidelity visual details. However, existing hybrids often lack systematic guidance on how and why to allocate model capacity between these paradigms. In this work, we introduce MADFormer, a Mixed Autoregressive and Diffusion Transformer that serves as a testbed for analyzing AR-diffusion trade-offs. MADFormer partitions image generation into spatial blocks, using AR layers for one-pass global conditioning across blocks and diffusion layers for iterative local refinement within each block. Through controlled experiments on FFHQ-1024 and ImageNet, we identify two key insights: (1) block-wise partitioning significantly improves performance on high-resolution images, and (2) vertically mixing AR and diffusion layers yields better quality-efficiency balances--improving FID by up to 75% under constrained inference compute. Our findings offer practical design principles for future hybrid generative models.

View on arXivView PDF

Code Implementations(1)

Apache-2.0

Official implementation of MADFormer, including training, sampling, and evaluation scripts as described in "MADFormer: Mixed Autoregressive and Diffusion Transformers for Continuous Image Generation".

70PythonJun 21, 20251 years agoApache-2.0

Similar Papers

Cite this paper

@article{chen2025madformer,
  title  = {MADFormer: Mixed Autoregressive and Diffusion Transformers for Continuous Image Generation},
  author = {Junhao Chen and Yulia Tsvetkov and Xiaochuang Han},
  year   = {2025},
  eprint = {2506.07999},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2506.07999v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2506.07999.svg)](https://codeofpaper.com/paper/2506.07999)

Discussion