English

Translating Flow to Policy via Hindsight Online Imitation

Robotics 2026-02-13 v2 Machine Learning

Abstract

Recent advances in hierarchical robot systems leverage a high-level planner to propose task plans and a low-level policy to generate robot actions. This design allows training the planner on action-free or even non-robot data sources (e.g., videos), providing transferable high-level guidance. Nevertheless, grounding these high-level plans into executable actions remains challenging, especially with the limited availability of high-quality robot data. To this end, we propose to improve the low-level policy through online interactions. Specifically, our approach collects online rollouts, retrospectively annotates the corresponding high-level goals from achieved outcomes, and aggregates these hindsight-relabeled experiences to update a goal-conditioned imitation policy. Our method, Hindsight Flow-conditioned Online Imitation (HinFlow), instantiates this idea with 2D point flows as the high-level planner. Across diverse manipulation tasks in both simulation and physical world, our method achieves more than 2×2\times performance improvement over the base policy, significantly outperforming the existing methods. Moreover, our framework enables policy acquisition from planners trained on cross-embodiment video data, demonstrating its potential for scalable and transferable robot learning.

Keywords

Cite

@article{arxiv.2512.19269,
  title  = {Translating Flow to Policy via Hindsight Online Imitation},
  author = {Yitian Zheng and Zhangchen Ye and Weijun Dong and Shengjie Wang and Yuyang Liu and Chongjie Zhang and Chuan Wen and Yang Gao},
  journal= {arXiv preprint arXiv:2512.19269},
  year   = {2026}
}