Search papers, labs, and topics across Lattice.
EpiMind introduces a graph-factored recurrent state-space world model coupled with graph-temporal ADMM to optimize multi-region epidemic interventions under mobility spillovers and shared capacity constraints. This addresses a critical gap where existing spatial planning either fails to model coupled action-conditioned dynamics or cannot enforce strict per-period resource feasibility. The framework cuts hospital admission forecasting RMSE by 29% relative to graph-free dynamics and plans policies within 1–5% of the optimal feasible constant policy while strictly guaranteeing shared-budget compliance.
Model-based planning over dynamic spatial graphs cuts epidemic prediction error by 29% while guaranteeing the hard, per-period resource feasibility that standard RL controllers routinely violate.
Epidemic policy planning often requires coordination between geographical regions, taking into account mobility-driven spillovers and how to make use of limited resources. Existing methods either lack action-conditioned models of coupled dynamics or cannot guarantee per-period feasibility. We present EpiMind, a graph world model framework for constrained epidemic policy planning across regions. A graph-factored recurrent state-space model generates joint policy-conditioned rollouts from regional latent beliefs, while graph-temporal ADMM optimizes regional interventions, enforces shared-resource feasibility through projection, and evaluates temporal specifications under the learned model. EpiMind reduces admission RMSE by 29% relative to graph-free dynamics modeling, plans within 1-5% of the best feasible constant policy with guaranteed shared-budget feasibility, and outperforms all deployable baselines across three resource budgets in real-context evaluation. These results demonstrate that graph-structured policy imagination with explicit constrained coordination supports effective epidemic interventions from learned dynamics.