mlsanityData to 7 Oct 2026

Learning to Solve Generative ODEs Beyond the Linear Span

arXiv 2606.08672 ↗First posted 7 Jun 2026cs.CV cs.LGcited by 0

Not on the map. Why?

Abstract (arXiv)

Diffusion and flow generative models sample by integrating a learned ODE, but high quality still requires many sequential model evaluations. Solver learning reduces this cost by adapting scalar coefficients, timesteps, or both, while keeping the backbone model fixed. In this work, we identify a structural bottleneck in this update family: each step remains span-limited. Since the scalar-coefficient update lies in the span of buffered velocity evaluations, it can fit only the in-span component while leaving any out-of-span residual unreachable by scalar recombination alone. We propose SpanLift, a lightweight neural solver that augments scalar-coefficient updates with a spatial residual operator. SpanLift keeps a fixed base solver as an in-span prior and learns a spatial residual operator over the state and velocity buffer. The operator is trained by endpoint teacher matching, preserves the pretrained backbone, and adds no model NFEs. Empirically, the learned correction transfers across base solvers and is predominantly out-of-span. Across pixel-space diffusion, latent flow matching, and precipitation nowcasting, SpanLift achieves state-of-the-art few-step sampling. With only 3 NFE, it improves CIFAR-10 FID from 8.16 to 5.69 and ImageNet FID from 17.37 to 11.83.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 37 of its 45 references that are in the map. First posted on arXiv 7 Jun 2026.

Builds on

37 of 45 references.

References · 37

  1. 2020Denoising Diffusion Probabilistic Modelscited by 20,851
  2. 2020Denoising Diffusion Implicit Modelscited by 9,655
  3. 2014ImageNet Large Scale Visual Recognition Challengecited by 8,441
  4. 2018A Style-Based Generator Architecture for Generative Adversarial Networkscited by 5,986
  5. 2014Going Deeper with Convolutionscited by 4,231
  6. 2022Elucidating the Design Space of Diffusion-Based Generative Modelscited by 2,994
  7. 2025Wan: Open and Advanced Large-Scale Video Generative Modelscited by 2,536
  8. 2022DPM-Solver: A Fast ODE Solver for Diffusion Probabilistic Model Sampling in Around 10 Stepscited by 1,961
  9. 2023Consistency Modelscited by 1,778
  10. 2019StarGAN v2: Diverse Image Synthesis for Multiple Domainscited by 1,095
  11. 2023ImageReward: Learning and Evaluating Human Preferences for Text-to-Image Generationcited by 1,010
  12. 2022DPM-Solver++: Fast Solver for Guided Sampling of Diffusion Probabilistic Modelscited by 924
25 more
  1. 2022Pseudo Numerical Methods for Diffusion Models on Manifoldscited by 695
  2. 2022Fast Sampling of Diffusion Models with Exponential Integratorcited by 526
  3. 2023UniPC: A Unified Predictor-Corrector Framework for Fast Sampling of Diffusion Modelscited by 441
  4. 2023Improved Techniques for Training Consistency Modelscited by 430
  5. 2024SANA: Efficient High-Resolution Image Synthesis with Linear Diffusion Transformerscited by 331
  6. 2020hp-VPINNs: Variational Physics-Informed Neural Networks With Domain Decompositioncited by 236
  7. 2022Learning Fast Samplers for Diffusion Models by Differentiating Through Sample Qualitycited by 181
  8. 2023SE(3)-Stochastic Flow Matching for Protein Backbone Generationcited by 147
  9. 2023PreDiff: Precipitation Nowcasting with Latent Diffusion Modelscited by 96
  10. 2024Align Your Steps: Optimizing Sampling Schedules in Diffusion Modelscited by 91
  11. 2023Fast ODE-based Sampling for Diffusion Models in Around 5 Stepscited by 83
  12. 2024Accelerating Diffusion Sampling with Optimized Time Stepscited by 67
  13. 2024On the Trajectory Regularity of ODE-based Diffusion Samplingcited by 51
  14. 2024Learning to Discretize Denoising Diffusion ODEscited by 39
  15. 2023Bespoke Solvers for Generative Flow Modelscited by 36
  16. 2025Distilling Parallel Gradients for Fast ODE Solvers of Diffusion Modelscited by 13
  17. 2023Towards More Accurate Diffusion Model Acceleration with A Timestep Tunercited by 12
  18. 2023Distilling ODE Solvers of Diffusion Models into Smaller Stepscited by 11
  19. 2025S4S: Solving for a Diffusion Model Solvercited by 10
  20. 2024Bespoke Non-Stationary Solvers for Fast Sampling of Diffusion and Flow Modelscited by 9
  21. 2025Differentiable Solver Search for Fast Diffusion Samplingcited by 5
  22. 2025B'ezierFlow: Learning B'ezier Stochastic Interpolant Schedulers for Few-Step Generationcited by 5
  23. 2025STORK: Faster Diffusion And Flow Matching Sampling By Resolving Both Stiffness And Structure-Dependencecited by 4
  24. 2026DyWeight: Dynamic Gradient Weighting for Few-Step Diffusion Samplingcited by 4
  25. 2026Dual-Solver: A Generalized ODE Solver for Diffusion Models with Dual Predictioncited by 1

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2026From Euler to Dormand-Prince: ODE Solvers for Flow Matching Generative Models
  2. 2026Formalizing the Sampling Design Space of Diffusion-Based Generative Models via Adaptive Solvers and Wasserstein-Bounded Timesteps
  3. 2025ODEt(ODEl): Shortcutting the Time and the Length in Diffusion and Flow Models for Faster Sampling
  4. 2025SADA: Stability-guided Adaptive Diffusion Acceleration
  5. 2026Bi-Anchor Interpolation Solver for Accelerating Generative Modeling
  6. 2024Sequential Flow Straightening for Generative Modeling
  7. 2025On Approaches to Building Surrogate ODE Models for Diffusion Bridges
  8. 2025A-FloPS: Accelerating Diffusion Models via Adaptive Flow Path Sampler