Align Your Steps ist eine Methode zur Optimierung des Abtastzeitplans von Diffusionsmodellen (Diffusion Models, DMs). Diese Methode nutzt stochastische Kalkül-Methoden, um für verschiedene Löser, trainierte DMs und Datensätze einen spezifischen optimalen Abtastzeitplan zu finden. Durch Minimierung des KLUB-Terms wird die zeitliche Diskretisierung, also der Abtastplan, optimiert, wodurch bei gleichem Rechenaufwand die Ausgabequalität verbessert wird. Die Methode zeigt hervorragende Ergebnisse bei Benchmarks für die Synthese von Bildern, Videos und 2D-Spielzeugdaten. Der optimierte Abtastzeitplan übertrifft in fast allen Experimenten zuvor manuell festgelegte Zeitpläne.