arXiv subjects

Prayag Chatha

Publications and source records attributed to Prayag Chatha.

3 recordsLinked to original sources

arXiv2026

Stochastic Epidemic Model Criticism with Neural Evidence Estimation

Neural techniques for simulation-based inference (SBI) have become popular in scientific domains featuring complex simulation models. These methods train neural networks on simulated data to infer unknown parameters; their reliability depends on the simulator faithfully representing reality. In infectious disease modeling, multiple models encoding different hypotheses often provide plausible explanations of observed data. Model selection is a difficult problem when likelihoods are unavailable, as is the case with many common stochastic transmission models. We propose a model criticism methodology designed for SBI, Neural Evidence Estimation (NEE), that integrates model selection and model misspecification detection. NEE learns a model's marginal likelihood (a.k.a. evidence) directly from simulations using normalizing flows, thereby avoiding noisy or complex Monte Carlo estimators. Through synthetic experiments, we demonstrate that NEE reliably selects the correct data generating process among multiple models and flags poorly specified models. We apply NEE to a real-world modeling problem concerning competing mechanisms for influenza reinfection in a multiple wave epidemic. This work argues for the importance of principled model criticism in SBI and infectious disease modeling.

stat.ME↗

arXiv2026

Neural Posterior Estimation for Stochastic Epidemic Modeling

Stochastic infectious disease models capture uncertainty in public health outcomes and have become increasingly popular in epidemiological practice. However, calibrating these models to observed data is challenging with existing methods for parameter estimation. Stochastic epidemic models are nonlinear dynamical systems with potentially large latent state spaces, resulting in computationally intractable likelihood densities. We develop an approach to calibrating complex epidemiological models to high-dimensional data using Neural Posterior Estimation, a novel technique for simulation-based inference. In NPE, a neural conditional density estimator trained on simulated data learns to "invert" a stochastic simulator, returning a parametric approximation to the posterior distribution. We introduce a stochastic, discrete-time Susceptible Infected (SI) model with heterogeneous transmission for healthcare-associated infections (HAIs). HAIs are a major burden on healthcare systems. They exhibit high rates of asymptotic carriage, making it difficult to estimate infection rates. Through extensive simulation experiments, we show that NPE produces accurate posterior estimates of infection rates with greater sample efficiency compared to Approximate Bayesian Computation (ABC). We then use NPE to fit our SI model to an outbreak of carbapenem-resistant Klebsiella pneumoniae in a long-term acute care facility, finding evidence of location-based heterogeneity in patient-to-patient transmission risk. We argue that our methodology can be fruitfully applied to a wide range of mechanistic transmission models and problems in the epidemiology of infectious disease.

stat.ME↗

arXiv2022

Dynamic Survival Transformers for Causal Inference with Electronic Health Records

In medicine, researchers often seek to infer the effects of a given treatment on patients' outcomes. However, the standard methods for causal survival analysis make simplistic assumptions about the data-generating process and cannot capture complex interactions among patient covariates. We introduce the Dynamic Survival Transformer (DynST), a deep survival model that trains on electronic health records (EHRs). Unlike previous transformers used in survival analysis, DynST can make use of time-varying information to predict evolving survival probabilities. We derive a semi-synthetic EHR dataset from MIMIC-III to show that DynST can accurately estimate the causal effect of a treatment intervention on restricted mean survival time (RMST). We demonstrate that DynST achieves better predictive and causal estimation than two alternative models.

cs.LG↗