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

Randomized Autoregressive Visual Generation

About

This paper presents Randomized AutoRegressive modeling (RAR) for visual generation, which sets a new state-of-the-art performance on the image generation task while maintaining full compatibility with language modeling frameworks. The proposed RAR is simple: during a standard autoregressive training process with a next-token prediction objective, the input sequence-typically ordered in raster form-is randomly permuted into different factorization orders with a probability r, where r starts at 1 and linearly decays to 0 over the course of training. This annealing training strategy enables the model to learn to maximize the expected likelihood over all factorization orders and thus effectively improve the model's capability of modeling bidirectional contexts. Importantly, RAR preserves the integrity of the autoregressive modeling framework, ensuring full compatibility with language modeling while significantly improving performance in image generation. On the ImageNet-256 benchmark, RAR achieves an FID score of 1.48, not only surpassing prior state-of-the-art autoregressive image generators but also outperforming leading diffusion-based and masked transformer-based methods. Code and models will be made available at https://github.com/bytedance/1d-tokenizer

Qihang Yu, Ju He, Xueqing Deng, Xiaohui Shen, Liang-Chieh Chen• 2024

Related benchmarks

TaskDatasetResultRank
Class-conditional Image GenerationImageNet 256x256
Inception Score (IS)326
441
Image GenerationImageNet 256x256 (val)
FID1.48
307
Class-conditional Image GenerationImageNet 256x256 (val)
FID1.48
293
Image ReconstructionImageNet 256x256
rFID2.28
93
Conditional Image GenerationImageNet-1K 256x256 (val)
gFID1.48
86
Image GenerationImageNet-1K 256x256 (val)
Inception Score326
85
Class-conditional Image GenerationImageNet-1k (val)
FID1.48
68
Class-conditional generationImageNet 256 x 256 1k (val)
FID1.48
67
Class-conditional Image GenerationImageNet-1K 256x256 (test)
FID1.5
50
Class-conditional Image GenerationImageNet 256x256 2012 (train val)
FID (w/o G)3.91
30
Showing 10 of 12 rows

Other info

Code

Follow for update