CAB: Accelerating Flow and Diffusion Sampling via Rectification and Corrected Adams-Bashforth
For practitioners of flow and diffusion models, CAB offers a simple, training-free acceleration method that improves sample quality at low NFE budgets without degrading performance at higher step counts.
CAB is a training-free sampler that accelerates flow and diffusion models by transforming sampling dynamics to a rectified coordinate system and applying a corrected Adams-Bashforth predictor, achieving improved quality-NFE trade-offs in the low-step regime (6-20 NFEs) across various benchmarks.
Flow and diffusion models achieve high-fidelity, high-resolution image synthesis, but often require many function evaluations (NFEs) at sampling time. Existing acceleration methods either require additional training through distillation or rely on training-free high-order solvers, and both can degrade sample quality at low NFE budgets. We propose CAB (Corrected Adams-Bashforth), a training-free sampler that accelerates both flow and diffusion models. CAB first transforms the sampling dynamics to a common rectified coordinate system, and then applies a multistep Adams-Bashforth predictor augmented with a simple correction term based on past velocity evaluations and therefore incurs no additional NFEs. The resulting method is simple, has the same algorithmic form across model classes, and has at least third-order local truncation error and second-order global error. Experiments on pretrained flow and diffusion models, including class-conditional and large-scale text-to-image benchmarks, show that CAB improves quality-NFE trade-offs in the low-step regime of 6-20 NFEs. It also remains competitive with strong training-free samplers at higher step counts across most tested models. The official implementation is available at https://github.com/Anuska-Roy/CAB.