EpiWorld: Grounding LLM Policy Agents in Epidemiological World Models

Zeeshan Memon, Yiqi Su, Kai Shu, Naren Ramakrishnan, Liang Zhao


Abstract
Epidemic intervention policies are textual artefacts that human decision-makers interpret, justify, and revise through natural language, making large language models a natural candidate for epidemic policy reasoning. A naive LLM, however, lacks the epidemic dynamics needed to project intervention consequences, the quantitative surveillance signals required to assess severity, and the institutional constraints that define admissible actions. We present EpiWorld, a closed-loop framework that grounds an LLM policy actor in a learned action-conditioned epidemiological world model and a tiered skill library of public-health protocols, surveillance tools, and adaptive lessons accumulated through after-action analysis. Given a candidate intervention, the world model predicts regional epidemic evolution and enables fast counterfactual rollouts that provide feedback for policy selection and refinement. Outcomes of simulated futures are distilled into reusable lessons while protocol constraints remain fixed, allowing the decision process to improve without sacrificing interpretability or controllability. We evaluate both the world model and the end-to-end framework on retrospective COVID-19 and Influenza datasets: the world model achieves the best out-of-distribution Peak-MAE among all forecasting baselines, and the closed-loop framework reduces cumulative hospitalisation by up to 59% across datasets and by an average of ~16% across six LLM backbones, outperforming reinforcement-learning and optimal-control policy baselines.
Anthology ID:
2026.findings-emnlp.564
Volume:
Findings of the Association for Computational Linguistics: EMNLP 2026
Month:
October
Year:
2026
Address:
Budapest, Hungary
Editors:
Hung-Yi Lee, Sunipa Dev, Jessy Li, Desmond Elliott
Venue:
Findings
SIG:
SIGDAT
Publisher:
Association for Computational Linguistics
Note:
Pages:
11885–11906
Language:
URL:
https://aclanthology.org/2026.findings-emnlp.564/
DOI:
Bibkey:
Cite (ACL):
Zeeshan Memon, Yiqi Su, Kai Shu, Naren Ramakrishnan, and Liang Zhao. 2026. EpiWorld: Grounding LLM Policy Agents in Epidemiological World Models. In Findings of the Association for Computational Linguistics: EMNLP 2026, pages 11885–11906, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
EpiWorld: Grounding LLM Policy Agents in Epidemiological World Models (Memon et al., Findings 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.findings-emnlp.564.pdf