Yiming Guan, Rui Yu, John Zhang, Lu Wang, Chaoyun Zhang, Liqun Li, Bo Qiao, Si Qin, He Huang, Fangkai Yang, Pu Zhao, Lukas Wutschitz, Samuel Kessler, Huseyin A. Inan, Robert Sim, Saravan Rajmohan, Qingwei Lin, Dongmei Zhang
4 min
Abstract
Agents operating in complex software environments benefit from reasoning about the consequences of their actions, as even a single incorrect user interface (UI) operation can derail long, artifact-preserving workflows. This challenge is particularly acute for computer-using scenarios, where real execution does not support counterfactual exploration, making large-scale trial-and-error learning and planning impractical despite the environment being fully digital and deterministic. We introduce the Computer-Using World Model (CUWM), a world model for desktop software that predicts the next UI state given the current state and a candidate action. CUWM adopts a two-stage factorization of UI dynamics: it first predicts a textual description of agent-relevant state changes, and then realizes these changes visually to synthesize the next screenshot. CUWM is trained on offline UI transitions collected from agents interacting with real Microsoft Office applications, and further refined with a lightweight reinforcement learning stage that aligns textual transition predictions with the structural requirements of computer-using environments. We evaluate CUWM via test-time action search, where a frozen agent uses the world model to simulate and compare candidate actions before execution. Across a range of Office tasks, world-model-guided test-time scaling improves decision quality and execution robustness.
Alex: Huh, so the words guide decisions better, and the images look convincing where agents need details. But the real test is agents using these simulations to do tasks better, right?
Sam: Precisely. Agents simulate several action options ahead of time, pick the best based on previews, then execute. In Office tasks, this boosts success rates by 4 to 8 percentage points across different models—meaning agents complete goals more reliably, especially spotting structural shifts like new menus.
Alex: So not perfect visuals, but the key info gets through... and it helps avoid dead ends. Interesting limitation though—why does adding both words and images sometimes hurt?
Sam: The paper notes combining them can confuse agents—words might clash with image details, or errors add up. They suggest current models aren't great at blending those signals yet, so image previews alone work best here. Still, it's a meaningful step for safe planning in apps where one wrong click sticks.
Alex: Makes sense why one wrong menu risks everything. Overall, this seems like a grounded step forward for agents handling sticky software workflows.
Sam: The paper suggests this factorization meaningfully advances desktop simulations, enabling reliable test-time looks ahead.
Alex: Thanks for joining this look at safer desktop AI planning on ResearchPod.