Gradient boosting: A computationally efficient alternative to Markov chain Monte Carlo sampling for fitting large Bayesian spatio-temporal binomial regression models
Disease forecasting and surveillance often involve fitting models to a tremendous volume of historical testing data collected over space and time. Bayesian spatio-temporal regression models fit with Markov chain Monte Carlo (MCMC) methods are commonly used for such data. When the spatio-temporal support of the model is large, implementing an MCMC algorithm becomes a significant computational burden. This research proposes a computationally efficient gradient boosting algorithm for fitting a Bayesian spatio-temporal mixed effects binomial regression model. We demonstrate our method on a disease forecasting model and compare it to a computationally optimized MCMC approach. Both methods are used to produce monthly forecasts for Lyme disease, anaplasmosis, ehrlichiosis, and heartworm disease in domestic dogs for the contiguous United States. The data have a spatial support of 3108 counties and a temporal support of 108-138 months with 71-135 million test results. The proposed estimation approach is several orders of magnitude faster than the optimized MCMC algorithm, with a similar mean absolute prediction error.
Authors
- Stella Self (ORCID: https://orcid.org/0000-0002-5956-5220)
- Rongjie Huang
- Christopher S. McMahan (ORCID: https://orcid.org/0000-0001-5056-9615)
- Alexander C. McLain (ORCID: https://orcid.org/0000-0002-5475-0670)
- Brian H. Herrin
- Bo Cai (ORCID: https://orcid.org/0000-0001-9386-8762)
Institutions
- Statistical and Applied Mathematical Sciences Institute (US)
- University of South Carolina (US)
- Kansas State University (US)
- Clemson University (US)
Publication Details
- Journal
- Infectious Disease Modelling
- Published
- 2024-10-04
- DOI
- https://doi.org/10.1016/j.idm.2024.09.008
- Citations
- 1
- Primary Topic
- Bayesian Methods and Mixture Models
- Type
- article
- Field-Weighted Citation Impact
- 0.35
Funders
- National Institute of General Medical Sciences