Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

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.

Yihao Sun, Zhilong Zhang, Yang Yu, Pierre-Luc Bacon• 2025

Related benchmarks

TaskDatasetResultRank
2D NavigationMini-BEHAVIOR OOD
Success Rate83
11
2D NavigationFrozenLake OOD
Success Rate95
11
3D NavigationWorld-in-World Active Recognition OOD
Success Rate61
11
3D NavigationWorld-in-World Image-Goal Navigation OOD
Success Rate43
11
ManipulationLanguage Table OOD
Success Rate73
11
ManipulationReal World OOD
Success Rate63
11
Active Inference PlanningFrozen Lake
Success Rate95
7
PlanningMini-BEHAVIOR
Success Rate83
2
PlanningReal-world
Success Rate63
2
Showing 9 of 9 rows

Other info

Follow for update