AION
Research paperEfficiency & Inference · Image, Video & 3D Generation · Interpretability1 source · Oct 7, 2026

Koopman Observers for Diffusion Acceleration: Correcting Feature Forecasts with Shallow Measurements

We introduce an observation-corrected Koopman framework for accelerating frozen diffusion models.

Key points

  • Feature caching accelerates diffusion sampling by replacing expensive network evaluations with predictions from previously computed activations.
  • Using calibration trajectories, we identify finite-dimensional, time-dependent Koopman approximations that jointly describe the increments of shallow and deep network features.
  • This formulation enables controlled comparisons of temporal prediction and observation correction.
  • The observer achieves $1.89\times$ and $1.85\times$ measured speedups over DDIM-50, supporting improved reference-sampler fidelity without retraining the denoiser.

Sources (1)

Extractive summary: sentences quoted from the sources.