Normalizing Trajectory Models : génération d'images en 4 étapes avec vraisemblance exacte

Les Normalizing Trajectory Models (NTM) modélisent chaque étape inverse d'un processus de diffusion comme un flux conditionnel normalisé, permettant un entraînement par vraisemblance exacte. L'architecture associe des blocs inversibles peu profonds à un prédicteur parallèle profond sur la trajectoire complète. La vraisemblance exacte permet une auto-distillation : un débruiteur léger entraîné sur les scores du modèle produit des échantillons de haute qualité en quatre étapes. Sur des benchmarks texte-image, NTM égale ou surpasse les baselines en seulement 4 pas d'échantillonnage.