mlsanityData to 7 Oct 2026

Error as Signal: Stiffness-Aware Diffusion Sampling via Embedded Runge-Kutta Guidance

arXiv 2603.03692 ↗First posted 4 Mar 2026cs.CV cs.AIcited by 0

On the map: Diffusion models → Diffusion foundations and sampling

Abstract (arXiv)

Classifier-Free Guidance (CFG) has established the foundation for guidance mechanisms in diffusion models, showing that well-designed guidance proxies significantly improve conditional generation and sample quality. Autoguidance (AG) has extended this idea, but it relies on an auxiliary network and leaves solver-induced errors unaddressed. In stiff regions, the ODE trajectory changes sharply, where local truncation error (LTE) becomes a critical factor that deteriorates sample quality. Our key observation is that these errors align with the dominant eigenvector, motivating us to leverage the solver-induced error as a guidance signal. We propose Embedded Runge-Kutta Guidance (ERK-Guid), which exploits detected stiffness to reduce LTE and stabilize sampling. We theoretically and empirically analyze stiffness and eigenvector estimators with solver errors to motivate the design of ERK-Guid. Our experiments on both synthetic datasets and the popular benchmark dataset, ImageNet, demonstrate that ERK-Guid consistently outperforms state-of-the-art methods. Code is available at https://github.com/mlvlab/ERK-Guid.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Placed in Diffusion foundations and sampling, one of 14 directions in Diffusion models. Builds on 40 of its 49 references that are in the map. First posted on arXiv 4 Mar 2026.

Builds on

40 of 49 references.

Closest ancestors · 23

Same direction, newest first

  1. 2025Differentiable Solver Search for Fast Diffusion Sampling27 May
  2. 2024Dynamic Diffusion Transformer4 Oct
  3. 2024Eliminating Oversaturation and Artifacts of High Guidance Scales in Diffusion Models3 Oct
  4. 2024Classifier-Free Guidance is a Predictor-Corrector16 Aug
  5. 2024Guiding a Diffusion Model with a Bad Version of Itself4 Jun
  6. 2024Applying Guidance in a Limited Interval Improves Sample and Distribution Quality in Diffusion Models11 Apr
  7. 2024Accelerating Diffusion Sampling with Optimized Time Steps27 Feb
  8. 2024DDMI: Domain-Agnostic Latent Diffusion Models for Synthesizing High-Quality Implicit Neural Representations23 Jan
  9. 2023Characteristic Guidance: Non-linear Correction for Diffusion Model at Large Guidance Scale11 Dec
  10. 2023Analyzing and Improving the Training Dynamics of Diffusion Models5 Dec
  11. 2023DPM-Solver-v3: Improved Diffusion ODE Solver with Empirical Model Statistics20 Oct
  12. 2023PixArt-α: Fast Training of Diffusion Transformer for Photorealistic Text-to-Image Synthesis30 Sep
11 more
  1. 2023UniPC: A Unified Predictor-Corrector Framework for Fast Sampling of Diffusion Models9 Feb
  2. 2022Scalable Diffusion Models with Transformers19 Dec
  3. 2022Classifier-Free Diffusion Guidance26 Jul
  4. 2022DPM-Solver: A Fast ODE Solver for Diffusion Probabilistic Model Sampling in Around 10 Steps2 Jun
  5. 2022Elucidating the Design Space of Diffusion-Based Generative Models1 Jun
  6. 2022Fast Sampling of Diffusion Models with Exponential Integrator29 Apr
  7. 2022Pseudo Numerical Methods for Diffusion Models on Manifolds20 Feb
  8. 2021High-Resolution Image Synthesis with Latent Diffusion Models20 Dec
  9. 2020Score-Based Generative Modeling through Stochastic Differential Equations26 Nov
  10. 2020Denoising Diffusion Implicit Models6 Oct
  11. 2020Denoising Diffusion Probabilistic Models19 Jun

Other references · 17

  1. 2018A Style-Based Generator Architecture for Generative Adversarial Networkscited by 5,986
  2. 2022Generative Adversarial Networkscited by 5,603
  3. 2023Adding Conditional Control to Text-to-Image Diffusion Modelscited by 5,466
  4. 2022Flow Matching for Generative Modelingcited by 5,426
  5. 2016Improved Techniques for Training GANscited by 2,938
  6. 2018Large Scale GAN Training for High Fidelity Natural Image Synthesiscited by 2,609
  7. 2022InstructPix2Pix: Learning to Follow Image Editing Instructionscited by 2,591
  8. 2015Variational Inference with Normalizing Flowscited by 2,161
  9. 2022Imagic: Text-Based Real Image Editing with Diffusion Modelscited by 1,172
  10. 2019Improved Precision and Recall Metric for Assessing Generative Modelscited by 925
  11. 2023Photorealistic Video Generation with Diffusion Modelscited by 304
  12. 2023Exposing flaws of generative model evaluation metrics and their unfair treatment of diffusion modelscited by 199
5 more
  1. 2023Bespoke Solvers for Generative Flow Modelscited by 36
  2. 2024Diffusion Prior-Based Amortized Variational Inference for Noisy Inverse Problemscited by 13
  3. 2024Constant Acceleration Flowcited by 11
  4. 2024Stochastic Conditional Diffusion Models for Robust Semantic Image Synthesiscited by 10
  5. 2025Blockwise Flow Matching: Improving Flow Matching Models For Efficient High-Quality Generationcited by 3

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2025Tortoise and Hare Guidance: Accelerating Diffusion Model Inference with Multirate Integration
  2. 2026On the Robustness of Distribution Support under Diffusion Guidance
  3. 2025Guided Path Sampling: Steering Diffusion Models Back on Track with Principled Path Guidance
  4. 2025CFG-OEC: Classifier Free Guidance with Orthogonal Error Correction
  5. 2025In-situ Autoguidance: Eliciting Self-Correction in Diffusion Models
  6. 2026Improving Diffusion Generalization with Weak-to-Strong Segmented Guidance
  7. 2023Elucidating The Design Space of Classifier-Guided Diffusion Generation
  8. 2025Improving Classifier-Free Guidance in Masked Diffusion: Low-Dim Theoretical Insights with High-Dim Impact