Autoregressive Diffusion World Models for Off-Policy Evaluation of LLM Agents
TLDR
ADWM uses a latent diffusion world model to simulate environment responses for off-policy evaluation of LLM agents without online interaction.
Reasoning
The paper introduces a novel combination of autoregressive and diffusion models to address compounding errors in world modeling for LLM agent evaluation. Strengths include a principled approach to handling discrete text actions and policy-conditioned score functions. Weaknesses are the lack of detailed empirical comparisons and potential limitations in scalability, though the abstract claims strong empirical results.
Read-first score
Read-first score 58.5, weighted from topical fit, citation, graph, method, reproducibility, and recency signals. Original total remains 48.
Field roles
Rank sensitivity
Stability: volatile; rank range: 462.
Keyword Scores
Deep Analysis
Innovations
- Proposes ADWM, a latent diffusion world model for off-policy evaluation of LLM agents without real environment interaction.
- Models each transition as an independent denoising process to avoid compounding errors inherent in autoregressive world models.
- Introduces a policy-conditioned score function that allows the evaluation policy to directly guide diffusion generation at each step.
Methodology
ADWM learns a latent diffusion world model from pre-collected trajectories. It performs step-by-step rollouts where the world model and the LLM agent alternate in causal order, with each transition modeled as an independent denoising process. The evaluation policy directly guides the diffusion generation via a policy-conditioned score function, ensuring simulated trajectories reflect the agent's decision-making patterns.
Key Results
ADWM achieves accurate value estimates and evaluation reliability across diverse multi-turn agent tasks, demonstrating its promise as a practical framework for offline LLM agent evaluation.
Limitations
- Relies on the availability and coverage of pre-collected trajectories, which may limit applicability in environments with sparse or biased offline data.