arXiv · 2511.06816
Controllable Flow Matching for Online Reinforcement Learning
Abstract
Model-based reinforcement learning (MBRL) typically relies on modeling environment dynamics for data efficiency. However, due to the accumulation of model errors over long-horizon rollouts, such methods often face challenges in maintaining modeling stability. To address this, we propose CtrlFlow, a trajectory-level synthetic method using conditional flow matching (CFM), which directly modeling the distribution of trajectories from initial states to high-return terminal states without explicitly modeling the environment transition function. Our method ensures optimal trajectory sampling by minimizing the control energy governed by the non-linear Controllability Gramian Matrix, while the generated diverse trajectory data significantly enhances the robustness and cross-task generalization of policy learning. In online settings, CtrlFlow demonstrates the better performance on common MuJoCo benchmark tasks than dynamics models and achieves superior sample efficiency compared to standard MBRL methods.
Explore related subjects
Keep this discovery
Explore connections, maps & timelines
Bin Wang, Boxiang Tao, Haifeng Jing, Hongbo Dou, Zijian Wang. 2025-11-10. Controllable Flow Matching for Online Reinforcement Learning. https://arxiv.org/abs/2511.06816
Cite the original work for its findings. Save a collection to share your selection of sources.