CVJun 16, 2024

STAR: Scale-wise Text-conditioned AutoRegressive image generation

arXiv:2406.10797v424 citations
Originality Highly original
AI Analysis

This addresses the problem of efficient and high-quality text-to-image synthesis for applications in creative and media industries, representing a strong specific gain rather than a foundational breakthrough.

The paper tackles high-resolution text-to-image generation by introducing STAR, a scale-wise autoregressive model that achieves 1024x1024 image synthesis with improved fidelity, text-image consistency, and aesthetic quality, generating images in 2.21 seconds on an A100 GPU.

We introduce STAR, a text-to-image model that employs a scale-wise auto-regressive paradigm. Unlike VAR, which is constrained to class-conditioned synthesis for images up to 256$\times$256, STAR enables text-driven image generation up to 1024$\times$1024 through three key designs. First, we introduce a pre-trained text encoder to extract and adopt representations for textual constraints, enhancing details and generalizability. Second, given the inherent structural correlation across different scales, we leverage 2D Rotary Positional Encoding (RoPE) and tweak it into a normalized version, ensuring consistent interpretation of relative positions across token maps and stabilizing the training process. Third, we observe that simultaneously sampling all tokens within a single scale can disrupt inter-token relationships, leading to structural instability, particularly in high-resolution generation. To address this, we propose a novel stable sampling method that incorporates causal relationships into the sampling process, ensuring both rich details and stable structures. Compared to previous diffusion models and auto-regressive models, STAR surpasses existing benchmarks in fidelity, text-image consistency, and aesthetic quality, requiring just 2.21s for 1024$\times$1024 images on A100. This highlights the potential of auto-regressive methods in high-quality image synthesis, offering new directions for the text-to-image generation.

Code Implementations1 repo
Foundations

The foundational work for this paper's niche, ranked by how specifically the neighbourhood builds on it — not by global fame.

Your Notes