APIUp to 28% cheaper than official pricesTry the API →
Hermes
Back to News

University of Geneva's GAR Speeds Up Robot AI Policies by 1.91x

New paper shows flow robot policies produce near-Gaussian per-step action distributions, with under 2% Wasserstein error. Gaussian Action Readout replaces the flow policy's final layer with a Gaussian head trained on 0.02% of states.

University of Geneva's GAR Speeds Up Robot AI Policies by 1.91x
Source
AlphaSignal
Published
Author
AlphaSignal
Read
2 min read

Takeaways

  • New paper shows flow robot policies produce near-Gaussian per-step action distributions, with under 2% Wasserstein error.
  • Gaussian Action Readout replaces the flow policy's final layer with a Gaussian head trained on 0.02% of states.
  • Matches teacher success on 50 RoboCasa365 tasks: 57.64% vs 58.24% for Xiaomi, 21.92% vs 23.08% for GR00T.
  • Delivers up to 1.91x faster inference by sampling in a single action-network pass instead of iterative flow integration.
  • Provides closed-form log-likelihoods, useful for offline RL, safety filters, and policy scoring.
  • Multimodal behavior still emerges across time via state changes, demonstrated on multi-order packing tasks.

Flow matching powers robot-control policies from Xiaomi Robotics and NVIDIA GR00T. These models can represent several plausible actions for the same state, a distribution that a simple Gaussian often approximates poorly. Researchers at the University of Geneva and University of Tübingen report a narrower result: after conditioning on an observation, action chunks sampled from these two policies closely match full-covariance Gaussians. Their study, Gaussian Distillation of Flow Policies ( paper ), turns that finding into a faster inference method.

Gaussian Action Readout, or GAR, freezes a trained flow policy and adds lightweight readouts for the Gaussian mean and covariance. The readout reduces iterative flow integration to one backbone pass. On 50 simulated manipulation tasks, GAR trails its teachers by 0.60 to 1.16 percentage points while accelerating model inference by 1.57× to 1.91×.

Why flow sampling taxes the control loop

A flow policy predicts an action chunk, a short sequence of future actions, by transforming random noise through a learned velocity field. Sampling requires several transformer evaluations as a numerical solver follows that field toward a clean action. Repeating those evaluations at every control step raises latency and limits control frequency.

Likelihood-based tools incur another cost. Offline reinforcement learning, safety filters, and imitation-learning metrics may need to score an action under the policy. A flow model generally computes that likelihood by integrating along a trajectory. A Gaussian provides the log-probability directly from its mean and covariance.

Reporting is indexed from AlphaSignal. Rights remain with the original publisher and cited sources.

Read original report

Sources

  1. 01arxiv.org
  2. 02twitter.com
  3. 03arxiv.org