R-WoM: Retrieval-augmented World Model For Computer-use Agents
TLDR
Proposes R-WoM, a retrieval-augmented world model that improves LLM-based simulation for computer-use agents by grounding predictions in external tutorials.
Reasoning
Strengths include systematic probing of LLM world model capabilities and a novel retrieval-augmented approach that shows significant improvements on real-world benchmarks. Weaknesses are the narrow focus on digital environments and reliance on tutorial retrieval, which may limit generalization.
Read-first score
Read-first score 59.7, weighted from topical fit, citation, graph, method, reproducibility, and recency signals. Original total remains 42.
Field roles
Rank sensitivity
Stability: volatile; rank range: 180.
Keyword Scores
Deep Analysis
Innovations
- Systematic probing of LLMs' core world modeling capabilities (future state prediction and reward estimation) through three tasks: next-state identification, full-procedure planning alignment, and milestone transition recognition.
- Proposal of Retrieval-augmented World Model (R-WoM) that grounds LLM simulations by incorporating factual, up-to-date knowledge retrieved from external tutorials.
Methodology
The study first probes LLMs' world modeling abilities via three tasks: next-state identification, full-procedure planning alignment, and milestone transition recognition. It then proposes R-WoM, which augments LLM simulations with retrieved external tutorial knowledge. Experiments are conducted on subsets of OSWorld and Webarena, comparing against baselines.
Key Results
R-WoM achieves relative improvements of up to 23.4% and 16.3% on subsets of OSWorld and Webarena, respectively, with particular advantage in longer-horizon simulations.
Limitations
- LLMs' tendency toward hallucination and reliance on static training knowledge.
- Compounding errors that inhibit long-horizon simulations.
- Performance rapidly degrades in full-procedure planning, highlighting limitations in reliably modeling environment dynamics over long horizons.