reAR: Rethinking Visual Autoregressive Models via Token-Wise Consistency Regularization

Abstract

Visual autoregressive (AR) generation offers a promising path toward unifying vision and language models, yet its performance remains suboptimal against diffusion models. Prior work often attributes this gap to tokenizer limitations and rasterization ordering. In this work, we identify a core bottleneck from the perspective of generator-tokenizer inconsistency, i.e., the AR-generated tokens may not be well-decoded by the tokenizer. To address this, we propose reAR, a simple training strategy introducing a token-wise regularization objective: when predicting the next token, the causal transformer is also trained to recover the visual embedding of the current token and predict the embedding of the target token under a noisy context. It requires no changes to the tokenizer, generation order, inference pipeline, or external models. Despite its simplicity, reAR substantially improves performance. On ImageNet, it reduces gFID from 3.02 to 1.86 and improves IS to 316.9 using a standard rasterization-based tokenizer. When applied to advanced tokenizers, it achieves a gFID of 1.42 with only 177M parameters, matching the performance with larger state-of-the-art diffusion models (675M).

Cite

Text

He et al. "reAR: Rethinking Visual Autoregressive Models via Token-Wise Consistency Regularization." International Conference on Learning Representations, 2026.

Markdown

[He et al. "reAR: Rethinking Visual Autoregressive Models via Token-Wise Consistency Regularization." International Conference on Learning Representations, 2026.](https://mlanthology.org/iclr/2026/he2026iclr-rear/)

BibTeX

@inproceedings{he2026iclr-rear,
  title     = {{reAR: Rethinking Visual Autoregressive Models via Token-Wise Consistency Regularization}},
  author    = {He, Qiyuan and Li, Yicong and Ye, Haotian and Wang, Jinghao and Liao, Xinyao and Heng, Pheng-Ann and Ermon, Stefano and Zou, James and Yao, Angela},
  booktitle = {International Conference on Learning Representations},
  year      = {2026},
  url       = {https://mlanthology.org/iclr/2026/he2026iclr-rear/}
}