Align Your Steps est une méthode d'optimisation du calendrier d'échantillonnage des modèles de diffusion (Diffusion Models, DMs). Cette méthode utilise le calcul stochastique pour trouver le calendrier d'échantillonnage optimal pour différents solveurs, DMs entraînés et ensembles de données. Elle optimise la discrétisation temporelle, c'est-à-dire le calendrier d'échantillonnage, en minimisant le terme KLUB, améliorant ainsi la qualité de sortie pour un budget de calcul identique. Cette méthode offre d'excellents résultats sur des tests de référence de synthèse d'images, de vidéos et de données jouets 2D ; les calendriers d'échantillonnage optimisés surpassent systématiquement les calendriers manuels précédents dans presque toutes les expériences.