Skip to main navigation Skip to search Skip to main content

CoLA-Flow Policy: Temporally Coherent Imitation Learning via Continuous Latent Action Flow Matching for Robotic Manipulation

  • Songwei Wu
  • , Zhiduo Jiang
  • , Wandong Sun
  • , Guanghu Xie
  • , Yuteng Xie
  • , Rui Zhao
  • , Yang Liu*
  • , Hong Liu
  • *Corresponding author for this work
  • Harbin Institute of Technology
  • The University of Sydney
  • Ltd.

Research output: Contribution to journalArticlepeer-review

Abstract

Learning long-horizon robotic manipulation requires jointly achieving expressive behavior modeling, real-time inference, and stable execution, which remains challenging for existing generative policies. Diffusion-based approaches offer strong modeling capacity but incur high inference latency, while flow matching enables fast, near-single-step generation yet often suffers from unstable execution when operating directly in the raw action space. We propose Continuous Latent Action Flow Policy (CoLA-Flow Policy), a trajectory-level imitation learning framework that performs flow matching in a continuous latent action space. By encoding action sequences into temporally coherent latent trajectories and learning an explicit latent-space flow, CoLA-Flow Policy decouples global motion structure from low-level control noise, enabling smooth and reliable long-horizon execution. The framework further integrates geometry-aware point cloud conditioning and execution-time multimodal modulation, using visual cues as a representative modality to enhance real-world robustness. Experiments in simulation and on real robots show that CoLA-Flow Policy achieves near-single-step inference, improves trajectory smoothness by up to 93.7% and task success by up to 25 percentage points over raw action-space flow baselines, while remaining significantly faster than diffusion-based policies.

Original languageEnglish
Pages (from-to)10329-10336
Number of pages8
JournalIEEE Robotics and Automation Letters
Volume11
Issue number9
DOIs
StatePublished - 2026

Keywords

  • Imitation learning
  • continuous latent action space
  • flow-based policy
  • real-time robot control

Fingerprint

Dive into the research topics of 'CoLA-Flow Policy: Temporally Coherent Imitation Learning via Continuous Latent Action Flow Matching for Robotic Manipulation'. Together they form a unique fingerprint.

Cite this