AI agents that run long, multi-step tasks have a habit of doubling down on bad assumptions instead of correcting course. Researchers propose a fix that edits the agent's own reasoning instead of just predicting what happens next.
Most "world models" built for AI agents work by forecasting environment responses, essentially guessing what a tool or webpage will return. The new Agent-Editing World Model (AEWM) skips that and instead tracks how an agent's reasoning and actions move a task forward. An "Action Judge" component sorts each decision into three buckets: critical, exploratory, or noisy. A "State Revision" component then rewrites the noisy ones using the history the agent already has, and a third piece called EditAct applies those rewrites mid-task, using real execution feedback rather than just flagging problems after the fact. The researchers trained and tested the system across three domains: web search, terminal use, and software engineering tasks.
The target problem, which the paper calls "task-state contamination," is familiar to anyone who has watched an agent spiral: one bad guess early in a session quietly corrupts every decision that follows, and nothing in most agent architectures catches it. Editing the reasoning trail directly is a more targeted approach than the usual fix of just adding more memory or more retries.
The paper reports EditAct beating the strongest baseline by 3.2 to 6.7 points across six benchmarks and three agent backbones, plus a 70.5% macro-F1 score on an Action Judge benchmark the researchers built themselves. That last figure, scored on a test of their own design, is worth filing under "claimed" until someone outside the lab reproduces it.