Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Thinking by Doing: Building Efficient World Model Reasoning in LLMs via Multi-turn Interaction

About

Developing robust world model reasoning is crucial for large language model (LLM) agents to plan and interact in complex environments. While multi-turn interaction offers a superior understanding of environmental dynamics via authentic feedback, current approaches often impose a rigid reasoning process, which constrains the model's active learning, ultimately hindering efficient world model reasoning. To address these issues, we explore world-model internalization through efficient interaction and active reasoning (WMAct), which liberates the model from structured reasoning, allowing the model to shape thinking directly through its doing, and achieves effective and efficient world model reasoning with two key mechanisms: (1) a reward rescaling mechanism adjusting outcome reward based on action efficacy to incentivize redundancy reduction and purposeful interaction; (2) an interaction frequency annealing strategy to progressively reduce the maximum allowed interaction turns, which compels the model to condense its learning and internalize environmental dynamics rather than over-relying on environmental cues. Our experiments on Sokoban, Maze, and Taxi show that WMAct yields effective world model reasoning capable of resolving tasks in a single turn that previously required multiple interactions and fosters strong transferability to complex environments, improving performance on a suite of reasoning benchmarks.

Bao Shu, Yan Cai, Jianjian Sun, Chunrui Han, En Yu, Liang Zhao, Jingcheng Hu, Yinmin Zhang, Haoran Lv, Yuang Peng, Zheng Ge, Xiangyu Zhang, Daxin Jiang, Xiangyu Yue• 2025

Related benchmarks

TaskDatasetResultRank
SokobanSokoban (Standard)
Accuracy78.57
14
SokobanSokoban (Hard-1)
Accuracy52.68
14
SokobanSokoban (Hard-2)
Accuracy49.9
14
Taxi DomainTaxi Standard
Accuracy62.16
14
Maze NavigationMaze (Standard)
Accuracy0.8814
14
Maze NavigationMaze Hard
Accuracy50.59
14
Showing 6 of 6 rows

Other info

Follow for update