SimpleAR: Pushing the Frontier of Autoregressive Visual Generation

April 22, 2025 ยท View on GitHub

SimpleARย 

SimpleAR: Pushing the Frontier of Autoregressive Visual Generation through Pretraining, SFT, and RL

Junke Wang1, Zhi Tian2, Xun Wang2, Xinyu Zhang2, Weilin Huang2, Zuxuan Wu1, Yu-Gang Jiang1
1Fudan University, 2ByteDance Seed

Introduction

This paper presents SimpleAR, a vanilla autoregressive visual generation model that achieves state-of-the-art text-to-image generation performance. First the first time, we demonstrate that:

  • ๐Ÿ† with only 0.5B parameters, an AR model can generate 1024 resolution images with high fidelity, and achieve competitive results on challenging T2I benchmarks, e.g., 0.59 on GenEval and 79.66 on DPG;
  • ๐Ÿš€ both supervised fine-tuning (SFT) and Group Relative Policy Optimization (GRPO) training could lead to significant improvements on image aesthectics and prompt alignment;
  • โšก๏ธ when deployed with vLLM, the throughput of our model allows for generating 1024 resolution images in 14 seconds, making high-resolution generation practical for real-world applications.

We open-sourced all the training and inference code, hoping to show the potential of autoregressive visual generation and encourage more participation in this research field.

Updates

  • [2025/04/20] Installation instructions and model zoo are updated! Thanks syjmelody, wusize, and micky-li-hd for raising issues.
  • [2025/04/21] Stronger models with better generation quality, and more functionality, e.g., editing and controllable generation, will be released in this repo, please stay tuned!
  • [2025/04/22] We provide a demo code to play with our released models.

Installation

For basic usage (pretraining, SFT, inference without vLLM), you can install the dependencies with:

python3 -m venv env

source env/bin/activate

pip install -e ".[train]"

While for advanced usage, please refer to TRAIN.md (GRPO training) and EVAL.md (inference with vLLM) to setup the environments, respectively.

Models & Scripts

Model Zoo

We provide both SFT and RL checkpoints:

nameGenEvalDPGHF weights ๐Ÿค—
SimpleAR-0.5B-SFT0.5379.34simplear-0.5B-sft
SimpleAR-0.5B-RL0.5979.66simplear-0.5B-grpo
SimpleAR-1.5B-SFT0.6180.11simplear-1.5B-sft
SimpleAR-1.5B-RL0.6381.31simplear-1.5B-grpo

Cosmos is used as our visual tokenizer, you can download and put it under ./checkpoints/:

cd checkpoints

git lfs install

git clone https://huggingface.co/nvidia/Cosmos-1.0-Tokenizer-DV8x16x16

Play with Our Model (Quick Start)

You can directly load SimpleAR with from_pretrained now ๐Ÿค—! We provide the demo code in PLAY.md.

Training

Please find the instructions on data preparation and training here.

Evaluation and Inference

We provide scripts to evaluate our released checkpoints on GenEval and DPG-Bench. Please see EVAL.md for more details.

Also, you can generate images with SimpleAR using generate.py. We implement different acceleration approaches, e.g., vLLM, speculative jacobi decoding. Please refer to EVAL.md.

Visualizations


1024 x 1024 generation results by SimpleAR.

Citation

If you find this repository helpful, please consider citing:

@article{wang2025simplear,
  title={SimpleAR: Pushing the Frontier of Autoregressive Visual Generation through Pretraining, SFT, and RL},
  author={Wang, Junke and Tian, Zhi and Wang, Xun and Zhang, Xinyu and Huang, Weilin and Wu, Zuxuan and Jiang, Yu-Gang},
  journal={arXiv preprint arXiv:2504.11455},
  year={2025}
}

Acknowledgement

We thank Peize Sun, Rui Tian, Feng Li, and Teng Yao for their valuable discussions.