Search papers, labs, and topics across Lattice.
This paper investigates a novel approach for calibrating long-horizon language agents' world models by introducing a budgeted probing mechanism that allows agents to query the environment before taking action. The authors demonstrate that targeted checks for procedural beliefs and structural cues for spatial beliefs can effectively reduce world-model errors, highlighting the importance of strategic resource allocation in agent decision-making. Controlled experiments reveal that utilizing mid-planning environment evidence significantly enhances the accuracy of the agents' world models, ultimately improving their performance in complex tasks.
Mid-planning environment queries can dramatically reduce world-model errors, transforming how agents interact with their surroundings.
Long-horizon language agents do not only choose actions; they carry a private model of the world from one decision to the next. When that model drifts, a later failure can be decided before the failing action is ever taken. We study a direct repair mechanism: before committing to the next task action, an agent may ask the environment about one belief field and write the answer back into its world model. This makes environment interaction a scarce calibration resource, not merely a way to advance the task. We introduce \method, a budgeted probing operator for structured belief tables. The useful probes are not the same everywhere. Procedural beliefs, such as tool dependencies, can often be repaired by targeted checks, but those checks spend steps that the task may need. Spatial beliefs, such as object locations and graph edges, rely more on structural cues; the agent's own confidence can be a poor guide when the world changes off-screen. A type-stratified analysis formalizes this probe-action frontier, and controlled experiments show that mid-planning environment evidence reduces terminal world-model error when the probe policy follows the structure of the task.