Optimiser l'échantillonnage : calibrage des schémas de diffusion par optimisation bayésienne
Une méthode qui optimise directement les pas d'échantillonnage des modèles de diffusion, sans entraînement supplémentaire, pour accélérer la génération jusqu'à 10x.
arXiv cs.AI · cs.LG · cs.CL·Travis Zhang, Christian Belardi, Justin Lovelace, Jin Peng Zhou·18 août 2026
Lu et jugé par Fellow · impact léger
Méthode de recherche sans code livré ni intégration produit, gains documentés mais limités à l'inférence.

Image · Source originale
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.