# FlowMPC: Improving Flow Matching policies with World Models

Preprint · Chandon Hamel

- Archive id: arxiv:2606.16286
- Published: 2026-06-15
- arXiv: https://arxiv.org/abs/2606.16286
- Citations: 0
- Source URL: https://arxiv.org/abs/2606.16286
- PDF: https://arxiv.org/pdf/2606.16286
- Provenance: arxiv

## Abstract

Flow Matching (FM) is a powerful approach for behavior cloning in multimodal action spaces \[Jiang et al., 2025\], but because it is not trained to directly maximize expected return, there is still room to improve how FM policies act at test time. This work investigates whether a learned world model can improve FM policies by enabling Model Predictive Path Integral (MPPI) planning over candidate action sequences proposed by the policy. Building on TD-MPC2 \[Hansen et al., 2024\], I introduce FlowMPC, a framework that combines an imitation-learned FM policy with a learned world model for test-time planning in ManiSkill manipulation tasks \[Tao et al., 2025\]. Across PickCube and PickSingleYCB, adding the world model improved performance over the FM policy alone, with especially clear gains in end-of-episode success. These results suggest that world-model-based planning can effectively complement flow-based imitation policies without modifying the FM training objective.

_Abstract reproduced under CC0-1.0 (arXiv descriptive metadata)._

Topics: [Generative & Multimodal Models](https://space-kitty.com/topics/generative-multimodal)

---
Space Kitty Archive · https://space-kitty.com · API: https://space-kitty.com/api/v1 · MCP: https://space-kitty.com/mcp · Premium: $0.20/request via x402 (Base Sepolia (testnet)). Metadata: arXiv (CC0), OpenAlex (CC0), Europe PMC (CC BY/CC0 works), ClinicalTrials.gov.