Artificial intelligence

Flow matching

Definition

Flow matching is a generative-model training method that learns a vector field for transforming a simple probability distribution into a data distribution. In robot learning, the generated samples can be continuous action sequences conditioned on observations and instructions.

Also known as: FM

Updated

Learn how a sample should move

Flow Matching for Generative Modeling trains a model by regressing vector fields along chosen probability paths between noise and data. At generation time, a numerical solver follows the learned field to transform an initial sample into a data-like sample.

Here, flow refers to movement through a mathematical sample space. It does not mean fluid flow around a robot or optical flow between camera frames.

Continuous robot actions are one application

The pi0 model combines a pretrained vision-language model with a flow-matching action architecture. Visual and language information condition the generation of robot actions. Its authors evaluate manipulation tasks including laundry folding and box assembly.

An action sample can represent several future commands, making this approach compatible with action chunking.

Flow matching and diffusion overlap

The original flow-matching formulation can use diffusion probability paths as well as other paths. The terms therefore describe related families of generative methods rather than completely separate ideas.

A chosen path, solver, and number of integration steps affect generation. Results from one model or solver configuration should not be treated as evidence that every flow-matching policy is faster or more accurate than every diffusion policy.

Sources