Transfusion: Vorhersage des nächsten Tokens und Diffusion von Bildern mit einem Multi-Modalen ModellTransfusion: Predict the Next Token and Diffuse Images with One
Multi-Modal Model
Wir stellen Transfusion vor, ein Rezept zur Schulung eines multimodalen Modells über diskrete und kontinuierliche Daten. Transfusion kombiniert die Verlustfunktion des Sprachmodellierens (Vorhersage des nächsten Tokens) mit Diffusion, um einen einzigen Transformer über gemischte Modalitätssequenzen zu trainieren. Wir trainieren mehrere Transfusion-Modelle mit bis zu 7 Milliarden Parametern von Grund auf auf einer Mischung aus Text- und Bilddaten vor und etablieren Skalierungsgesetze im Hinblick auf eine Vielzahl von uni- und multimodalen Benchmarks. Unsere Experimente zeigen, dass Transfusion signifikant besser skaliert als die Quantisierung von Bildern und das Training eines Sprachmodells über diskrete Bildtokens. Durch die Einführung von modalitätsspezifischen Codierungs- und Decodierungsschichten können wir die Leistung der Transfusion-Modelle weiter verbessern und jedes Bild sogar auf nur 16 Patches komprimieren. Wir zeigen weiterhin, dass die Skalierung unseres Transfusion-Rezepts auf 7 Milliarden Parameter und 2 Billionen multimodale Tokens ein Modell hervorbringt, das Bilder und Text auf Augenhöhe mit ähnlich skalierten Diffusionsmodellen und Sprachmodellen generieren kann und die Vorteile beider Welten nutzt.