From Prediction to Perfection: Introducing Refinement to Autoregressive Image Generation

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Cheng, Cheng, Song, Lin, An, Di, Xiao, Yicheng, Zhang, Xuchong, Sun, Hongbin, Shan, Ying
Format: Preprint
Publié: 2025
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866910002692751360
author Cheng, Cheng
Song, Lin
An, Di
Xiao, Yicheng
Zhang, Xuchong
Sun, Hongbin
Shan, Ying
author_facet Cheng, Cheng
Song, Lin
An, Di
Xiao, Yicheng
Zhang, Xuchong
Sun, Hongbin
Shan, Ying
contents Autoregressive (AR) image generators offer a language-model-friendly approach to image generation by predicting discrete image tokens in a causal sequence. However, unlike diffusion models, AR models lack a mechanism to refine previous predictions, limiting their generation quality. In this paper, we introduce TensorAR, a new AR paradigm that reformulates image generation from next-token prediction to next-tensor prediction. By generating overlapping windows of image patches (tensors) in a sliding fashion, TensorAR enables iterative refinement of previously generated content. To prevent information leakage during training, we propose a discrete tensor noising scheme, which perturbs input tokens via codebook-indexed noise. TensorAR is implemented as a plug-and-play module compatible with existing AR models. Extensive experiments on LlamaGEN, Open-MAGVIT2, and RAR demonstrate that TensorAR significantly improves the generation performance of autoregressive models.
format Preprint
id arxiv_https___arxiv_org_abs_2505_16324
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle From Prediction to Perfection: Introducing Refinement to Autoregressive Image Generation
Cheng, Cheng
Song, Lin
An, Di
Xiao, Yicheng
Zhang, Xuchong
Sun, Hongbin
Shan, Ying
Computer Vision and Pattern Recognition
Autoregressive (AR) image generators offer a language-model-friendly approach to image generation by predicting discrete image tokens in a causal sequence. However, unlike diffusion models, AR models lack a mechanism to refine previous predictions, limiting their generation quality. In this paper, we introduce TensorAR, a new AR paradigm that reformulates image generation from next-token prediction to next-tensor prediction. By generating overlapping windows of image patches (tensors) in a sliding fashion, TensorAR enables iterative refinement of previously generated content. To prevent information leakage during training, we propose a discrete tensor noising scheme, which perturbs input tokens via codebook-indexed noise. TensorAR is implemented as a plug-and-play module compatible with existing AR models. Extensive experiments on LlamaGEN, Open-MAGVIT2, and RAR demonstrate that TensorAR significantly improves the generation performance of autoregressive models.
title From Prediction to Perfection: Introducing Refinement to Autoregressive Image Generation
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2505.16324