Track papers across labs, topics, and trends.
Search papers, labs, and topics across Lattice.
This paper introduces Velocity Scheduled Flow Matching (VSFM), which enhances flow matching by allowing the sample velocity to vary along the trajectory instead of moving at a constant speed. By employing nonnegative velocity profiles, the method significantly improves the quality of generated samples, achieving up to a 19.8% reduction in Fréchet Inception Distance (FID) on CIFAR-10 without requiring retraining. Additionally, training from scratch with a braking profile yields a further 17.4% reduction in FID at only 4 network evaluations, demonstrating the effectiveness of the approach in optimizing sampling efficiency and quality.
Sampling quality improves dramatically with variable velocity profiles, achieving a 19.8% reduction in FID on CIFAR-10 without retraining.
Flow matching trains a neural network to regress the conditional velocity along a linear interpolant between noise and data, and the number of network evaluations~(NFE) sets the cost of sampling. The straight-line interpolant carries an implicit choice: the sample moves at constant speed throughout the trajectory. We relax this choice and introduce Velocity Scheduled Flow Matching~(VSFM), which replaces the conditional target $x_1 - x_0$ with $v(t)(x_1 - x_0)$ for any nonnegative profile $v:[0,1]\to\mathbb{R}_{\geq 0}$ satisfying $\int_0^1 v\,dt = 1$. We study six polynomial profiles drawn from motion planning. The first use of VSFM is at inference time: a pretrained linear flow-matching model can be sampled under any admissible profile by integrating its ODE on a non-uniform $τ$-schedule, with no retraining and no additional computation; on CIFAR-10 this lowers FID by up to $19.8\%$. Training from scratch under a braking profile gives a further reduction of $17.4\%$ at $4$~NFE. Both gains follow from the local truncation error of the Euler integrator on the induced grid.