FlowPolicy: Enabling Fast and Robust 3D Flow-Based Policy via Consistency Flow Matching for Robot Manipulation

Authors

  • Qinglun Zhang University of Electronic Science and Technology of China Megvii Technology
  • Zhen Liu University of Electronic Science and Technology of China Megvii Technology
  • Haoqiang Fan Megvii Technology
  • Guanghui Liu University of Electronic Science and Technology of China
  • Bing Zeng University of Electronic Science and Technology of China
  • Shuaicheng Liu University of Electronic Science and Technology of China Megvii Technology

DOI:

https://doi.org/10.1609/aaai.v39i14.33617

Abstract

Robots can acquire complex manipulation skills by learning policies from expert demonstrations, which is often known as vision-based imitation learning. Generating policies based on diffusion and flow matching models has been shown to be effective, particularly in robotic manipulation tasks. However, recursion-based approaches are inference inefficient in working from noise distributions to policy distributions, posing a challenging trade-off between efficiency and quality. This motivates us to propose FlowPolicy, a novel framework for fast policy generation based on consistency flow matching and 3D vision. Our approach refines the flow dynamics by normalizing the self-consistency of the velocity field, enabling the model to derive task execution policies in a single inference step. Specifically, FlowPolicy conditions on the observed 3D point cloud, where consistency flow matching directly defines straight-line flows from different time states to the same action space, while simultaneously constraining their velocity values, that is, we approximate the trajectories from noise to robot actions by normalizing the self-consistency of the velocity field within the action space, thus improving the inference efficiency. We validate the effectiveness of FlowPolicy in Adroit and Metaworld, demonstrating a 7× increase in inference speed while maintaining competitive average success rates compared to state-of-the-art methods.

Published

2025-04-11

How to Cite

Zhang, Q., Liu, Z., Fan, H., Liu, G., Zeng, B., & Liu, S. (2025). FlowPolicy: Enabling Fast and Robust 3D Flow-Based Policy via Consistency Flow Matching for Robot Manipulation. Proceedings of the AAAI Conference on Artificial Intelligence, 39(14), 14754–14762. https://doi.org/10.1609/aaai.v39i14.33617

Issue

Section

AAAI Technical Track on Intelligent Robots