अपने कदम मिलाएँ (Align Your Steps) डिफ्यूज़न मॉडल (Diffusion Models, DMs) के सैंपलिंग शेड्यूल को बेहतर बनाने का एक तरीका है। यह तरीका स्टोकेस्टिक कैलकुलस का उपयोग करके विभिन्न सॉल्वर्स, प्रशिक्षित DMs और डेटासेट के लिए विशिष्ट अनुकूलित सैंपलिंग शेड्यूल खोजता है। यह KLUB पद को कम करके समय के विवेचना को बेहतर बनाता है, अर्थात सैंपलिंग शेड्यूल, जिससे समान कम्प्यूटेशनल बजट के साथ आउटपुट गुणवत्ता में सुधार होता है। यह तरीका छवि, वीडियो और 2D टॉय डेटा सिंथेसिस बेंचमार्क टेस्ट में बेहतरीन प्रदर्शन करता है, और अनुकूलित सैंपलिंग शेड्यूल लगभग सभी प्रयोगों में पहले से मौजूद मैन्युअल रूप से निर्धारित शेड्यूल से बेहतर प्रदर्शन करता है।