Surrogate methods for parameter inference in transmission dynamic models

Mechanistic models of infectious disease transmission provide a framework for understanding and predicting disease dynamics, but statistical inference is often computationally prohibitive. This challenge is especially pronounced for models with high-dimensional structures and many unknown parameters, where trajectory matching may require thousands of evaluations of an expensive likelihood function. We developed a surrogate-assisted optimization framework that uses Bayesian optimization to reduce the number of evaluations of the underlying mechanistic model required for parameter inference. We evaluated the approach using simulated fitting problems based on age-structured susceptible-exposed-infected-recovered models with 4, 8, and 16 age classes and a 210-compartment age-structured rotavirus transmission model. Applying surrogate-assisted optimization substantially reduced the number of likelihood evaluations required to identify good parameter estimates, with larger improvements as model dimensionality increased. In the rotavirus model, the surrogate Bayesian optimization approach reached the benchmark after approximately 0.7 hours, compared with approximately 9.4 hours using traditional Nelder-Mead optimization. These results demonstrate that surrogate-assisted optimization can substantially reduce the computational burden of trajectory matching for complex mechanistic transmission models, particularly for higher-dimensional inference problems.