Optimiser l'échantillonnage : calibrage des schémas de diffusion par optimisation bayésienne

Les auteurs proposent OYS, qui traite le choix des timesteps de diffusion comme un problème d'optimisation boîte noire résolu par optimisation bayésienne, plutôt que d'utiliser des substituts théoriques. La méthode surpasse les schémas par défaut et Align Your Steps sur la génération text-to-image, l'inpainting et d'autres tâches. Un schéma OYS à 5 étapes conserve 89-94% de la qualité d'un schéma à 50 étapes tout en divisant le coût d'inférence par 10.