Autoregressives Modell schlägt Diffusion: Llama für skalierbare BildgenerierungAutoregressive Model Beats Diffusion: Llama for Scalable Image
Generation
Wir stellen LlamaGen vor, eine neue Familie von Bildgenerierungsmodellen, die das originale „next-token prediction“-Paradigma großer Sprachmodelle auf den visuellen Generierungsbereich anwenden. Es ist eine bejahende Antwort darauf, ob einfache autoregressive Modelle, z. B. Llama, ohne induktive Voreingenommenheiten gegenüber visuellen Signalen Spitzenleistungen bei der Bildgenerierung erzielen können, wenn sie ordnungsgemäß skaliert werden. Wir überprüfen erneut die Designräume von Bild-Tokenizern, die Skalierungseigenschaften von Bildgenerierungsmodellen und die Qualität ihrer Trainingsdaten. Das Ergebnis dieser Untersuchung besteht aus: (1) Einem Bild-Tokenizer mit einer Downsample-Rate von 16, einer Rekonstruktionsqualität von 0,94 rFID und einer Codebuchnutzung von 97% im ImageNet-Benchmark. (2) Eine Reihe von klassenkonditionalen Bildgenerierungsmodellen mit 111M bis 3,1B Parametern, die auf ImageNet 256x256-Benchmarks 2,18 FID erreichen und die beliebten Diffusionsmodelle wie LDM, DiT übertreffen. (3) Ein textkonditionales Bildgenerierungsmodell mit 775M Parametern, das durch zweistufiges Training an LAION-COCO und hochästhetischen Bildern die Wettbewerbsfähigkeit bei visueller Qualität und Textausrichtung demonstriert. (4) Wir bestätigen die Wirksamkeit von LLM-Service-Frameworks bei der Optimierung der Inferenzgeschwindigkeit von Bildgenerierungsmodellen und erzielen eine Beschleunigung von 326% - 414%. Wir veröffentlichen alle Modelle und Codes, um die Open-Source-Community der visuellen Generierung und multimodalen Grundlagenmodelle zu unterstützen.