Align Your Stepsは、拡散モデル（Diffusion Models, DMs）の標本化スケジュールを最適化するための手法です。この手法は確率微積分を用いて、様々なソルバー、訓練済みのDM、データセットに対して最適な標本化スケジュールを特定します。KLUB項を最小化することで時間離散化（つまり、標本化スケジュール）を最適化し、同じ計算予算で出力品質を向上させます。この手法は、画像、ビデオ、および2D玩具データ合成のベンチマークテストで優れた性能を示し、最適化された標本化スケジュールは、ほぼ全てのテストにおいて従来の手動設定スケジュールを上回りました。