Transfusion : Prédire le Prochain Token et Diffuser les Images avec un Modèle Multi-ModalTransfusion: Predict the Next Token and Diffuse Images with One
Multi-Modal Model
Nous présentons Transfusion, une recette pour entraîner un modèle multi-modal sur des données discrètes et continues. Transfusion combine la perte de modélisation du langage (prédiction du prochain jeton) avec la diffusion pour entraîner un seul transformateur sur des séquences de modalités mixtes. Nous pré-entraînons plusieurs modèles Transfusion jusqu'à 7 milliards de paramètres à partir de zéro sur un mélange de données textuelles et d'images, établissant des lois d'échelle par rapport à une variété de références uni- et cross-modales. Nos expériences montrent que Transfusion évolue de manière significativement meilleure que la quantification des images et l'entraînement d'un modèle de langage sur des jetons d'image discrets. En introduisant des couches d'encodage et de décodage spécifiques à la modalité, nous pouvons améliorer encore les performances des modèles Transfusion, et même compresser chaque image en seulement 16 patchs. Nous démontrons en outre que l'évolution de notre recette Transfusion à 7 milliards de paramètres et 2 billions de jetons multi-modaux produit un modèle capable de générer des images et du texte au niveau des modèles de diffusion de taille similaire et des modèles de langage, récoltant les avantages des deux mondes.