Planning with Unified Multimodal Models
About
With the powerful reasoning capabilities of large language models (LLMs) and vision-language models (VLMs), many recent works have explored using them for decision-making. However, most of these approaches rely solely on language-based reasoning, which limits their ability to reason and make informed decisions. Recently, a promising new direction has emerged with unified multimodal models (UMMs), which support both multimodal inputs and outputs. We believe such models have greater potential for decision-making by enabling reasoning through generated visual content. To this end, we propose Uni-Plan, a planning framework built on UMMs. Within this framework, a single model simultaneously serves as the policy, dynamics model, and value function. In addition, to avoid hallucinations in dynamics predictions, we present a novel approach self-discriminated filtering, where the generative model serves as a self-discriminator to filter out invalid dynamics predictions. Experiments on embodied decision-making tasks show that Uni-Plan substantially improves success rates compared to VLM-based methods, while also showing strong data scalability, requiring no expert demonstrations and achieving better performance under the same training-data size. This work lays a foundation for future research in reasoning and decision-making with UMMs.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| 2D Navigation | Mini-BEHAVIOR OOD | Success Rate83 | 11 | |
| 2D Navigation | FrozenLake OOD | Success Rate95 | 11 | |
| 3D Navigation | World-in-World Active Recognition OOD | Success Rate61 | 11 | |
| 3D Navigation | World-in-World Image-Goal Navigation OOD | Success Rate43 | 11 | |
| Manipulation | Language Table OOD | Success Rate73 | 11 | |
| Manipulation | Real World OOD | Success Rate63 | 11 | |
| Active Inference Planning | Frozen Lake | Success Rate95 | 7 | |
| Planning | Mini-BEHAVIOR | Success Rate83 | 2 | |
| Planning | Real-world | Success Rate63 | 2 |