Grounded-Retrieval Adversarial Imitation Loop: Integrating Language, Agent, and World Models

We present GRAIL, a grounded simulation framework that unifies Language, Agent, and World models. GRAIL retrieves candidate real next-actions from large behavior logs, lets an LM reason-and-act (ReAct) to select among them, and learns a world model that predicts counterfactual outcomes. To support short term alignment, a GAIL-style discriminator supplies an adversarial reward that aligns agent trajectories to human occupancy measures while mitigating the homogenization artifacts of “silicon survey” prompting. On YouTube benchmarks, GRAIL improves (i) trajectory fidelity to human behavior and (ii) long-horizon stability versus Language-only and Action-only baselines. We release code, slates, and evaluation scripts to catalyze grounded, agentic world-modeling research.