arrow
Return

Adaptive Mean Flow for Responsive Closed-Loop Robot Control

delete2026-09-17
delete0
PRE
AI
A
Aksel Vaaler
M
Marco Job
C
Christian Holden
O
Olav Egeland
DOI:10.1109/lra.2026.3734902delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
Diffusion- and flow-based robot policies have recently become widespread in robotic Imitation Learning (IL) due to their high performance and ability to model continuous and multimodal distributions. However, the iterative denoising procedure used by these models introduces significant prediction latency, hindering high-frequency closed-loop robot control and leading to jittery, unstable motion when frequent updates to the robot's action predictions are used. Therefore, it is common practice to train models to predict chunks of actions that can be executed sequentially without feedback, even when this reduces responsiveness and may mean the most recent state information is not used. In this article, we present Adaptive Mean Flow (AMF), a flow-based IL method that enables smooth and responsive, fully closed-loop robot control. AMF uses Mean Flow, which is an accelerated form of Flow Matching (FM), to minimize prediction latency. To ensure smoothness and consistency across predictions, AMF uses a corrupted version of the trajectory from the previous step when predicting new robot actions, with the signal-to-noise ratio increasing over the time parameter of the trajectory. This discourages large changes in the prediction from one step to the next, while allowing freedom to adapt the predictions for future steps. We evaluate AMF across a wide range of simulated and real robot tasks and demonstrate significantly improved performance compared with baselines.
Keywords:
Fluid flow
Modeling
Sequences
Sequential analysis
Robots
Training
Noise reduction
Trajectory
Frequency modulation
Noise

Journal

I
IEEE Robotics and Automation Letters
IF:
5.3
Papers:
1.9K
Citations:
3.9W

Organization

N
Norwegian University of Science and Technology
Scholars:
479
Papers: 212
Citations: 0
Cited Papers

Cited Papers

No cited papers available