AREX: Affine-Residual Exponential Integrator
What it is
AREX is a training-free sampling method for pretrained flow matching models that decomposes the learned dynamics into an analytically tractable affine component and a neural residual. It integrates the affine part using an explicit matrix-valued propagator to improve sample fidelity in few-step regimes without retraining.
The problem it solves
Standard flow matching samplers require many integration steps to achieve high fidelity, making inference slow and computationally expensive.
What you could use it for
- Accelerating inference for pretrained flow matching models
- Improving image generation quality with fewer sampling steps
- Enhancing text-to-image generation fidelity without model retraining
Read the source ↗Back to the library →
Summary written by an AI model from the source; check the source before you rely on it.