arXiv:2607.15901: DSWorld World Model Speeds Up Data Science Agents Up to 14× in Training and Inference

What is DSWorld, and why do data science agents need a world model?

A world model is a component that internally simulates the outcomes of actions before they are actually carried out, instead of the agent waiting each time for the real result of an operation. Zherui Yang, Fan Liu, and Hao Liu introduce DSWorld, a world model designed specifically for data science agents — systems that automatically write and execute code for data analysis, model training, and evaluating results. The problem they solve is computational inefficiency: every attempted operation would otherwise require actual execution, often on resource-intensive tasks such as model training or processing large datasets.

DSWorld combines structured state, routing, and simulation

The architecture combines three elements: structured state that concisely describes the current task context, intelligent routing that decides when it is worth calling a real simulation versus a fast estimate, and an LLM-based simulator that predicts the outcome of resource-intensive operations without actually executing them. The system is further paired with a technique called “Reflective World Model Optimization” — an error-aware reinforcement learning (RL) approach that learns from its own prediction mistakes — and a dataset of 8,000 recorded trajectories that serve as the training foundation.

DSWorld: 14× faster training and 3-6× faster inference

The results show substantial speedups: DSWorld achieves about 14× faster RL agent training compared to approaches without a world model, and 3 to 6 times faster search-based inference, where the agent explores multiple possible actions before choosing the best one. In addition, the system outperforms the strongest tested LLM baseline by 35.6% on transition-prediction tasks — predicting how a task’s state will change after a given operation — while maintaining competitive performance on overall data science tasks.

Speedups change the economics of agentic work

The combination of 14× faster training and 3-6× faster inference means data science agents can try significantly more strategies within the same time and computational budget. For tasks that otherwise require expensive, repeated code execution — such as hyperparameter search or model selection — this translates directly into lower costs and faster iteration when developing automated data science pipelines.