Customize Your Visual Autoregressive Recipe with Set Autoregressive Modeling
This work addresses the problem of improving autoregressive image generation efficiency and flexibility for AI researchers and practitioners, representing a novel paradigm rather than an incremental improvement.
The authors introduced Set AutoRegressive Modeling (SAR), a new paradigm for autoregressive image generation that generalizes conventional AR by splitting sequences into arbitrary sets of multiple tokens rather than outputting tokens in fixed raster order, achieving photo-realistic image synthesis with a 900M text-to-image model on ImageNet.
We introduce a new paradigm for AutoRegressive (AR) image generation, termed Set AutoRegressive Modeling (SAR). SAR generalizes the conventional AR to the next-set setting, i.e., splitting the sequence into arbitrary sets containing multiple tokens, rather than outputting each token in a fixed raster order. To accommodate SAR, we develop a straightforward architecture termed Fully Masked Transformer. We reveal that existing AR variants correspond to specific design choices of sequence order and output intervals within the SAR framework, with AR and Masked AR (MAR) as two extreme instances. Notably, SAR facilitates a seamless transition from AR to MAR, where intermediate states allow for training a causal model that benefits from both few-step inference and KV cache acceleration, thus leveraging the advantages of both AR and MAR. On the ImageNet benchmark, we carefully explore the properties of SAR by analyzing the impact of sequence order and output intervals on performance, as well as the generalization ability regarding inference order and steps. We further validate the potential of SAR by training a 900M text-to-image model capable of synthesizing photo-realistic images with any resolution. We hope our work may inspire more exploration and application of AR-based modeling across diverse modalities.