Implicit-explicit and split-explicit super-time-stepping methods

arXiv:2608.02823 2026 Sampling 1 ideas extracted · analyzed Aug 31, 2026

What the math gives to ML

The paper develops a constructive multirate integration mechanism for additively split dynamics in which a stiff diffusion-like component is advanced by super-time-stepping while advection or learned drift remains explicit and local reactions can use localized implicit solves. The transferable asset is solve-decoupled coupling: many stable diffusion substeps can be inserted inside one macrostep without requiring a globally coupled implicit solve. This suggests neural ODE and diffusion-sampling architectures whose vector field is split into a known smoothing operator, a learned nonstiff drift, and an optional pointwise nonlinear term. The first useful test is whether the ExtSTS macrostep permits fewer score-network evaluations or larger stable solver steps at matched sample quality than Euler, Heun, and standard operator splitting.

Ideas from this paper

✓✓ Beats tuned baseline 2026

Solve-Decoupled Super-Time-Stepping Sampler

Split a neural ODE or diffusion-model probability-flow ODE into a stiff known smoothing operator, a learned drift, and an optional local reaction term. Use super-time-stepping stages for the smoothing operator inside a single macrostep, while evaluating the learned drift only at selected coupling stages and treating the local reaction with diagonal or block-local implicit solves. This should allow substantially larger stable macrosteps when the known operator has a large negative spectral…

Useful7/10
Difficulty6/10
Novelty7/10
Paper: Implicit-explicit and split-explicit super-time-stepping methods arXiv:2608.02823