Abstract:Birth-death models play a key role in phylodynamic analysis for their interpretation in terms of key epidemiological parameters. In particular, models with piecewise-constant rates varying at different epochs in time, to which we refer as episodic birth-death-sampling (EBDS) models, are valuable for their reflection of changing transmission dynamics over time. A challenge, however, that persists with current time-varying model inference procedures is their lack of computational efficiency. This limitation hinders the full utilization of these models in large-scale phylodynamic analyses, especially when dealing with high-dimensional parameter vectors that exhibit strong correlations. We present here a linear-time algorithm to compute the gradient of the birth-death model sampling density with respect to all time-varying parameters, and we implement this algorithm within a gradient-based Hamiltonian Monte Carlo (HMC) sampler to alleviate the computational burden of conducting inference under a wide variety of structures of, as well as priors for, EBDS processes. We assess this approach using three different real world data examples, including the HIV epidemic in Odesa, Ukraine, seasonal influenza A/H3N2 virus dynamics in New York state, America, and Ebola outbreak in West Africa. HMC sampling exhibits a substantial efficiency boost, delivering a 10- to 200-fold increase in minimum effective sample size per unit-time, in comparison to a Metropolis-Hastings-based approach. Additionally, we show the robustness of our implementation in both allowing for flexible prior choices and in modeling the transmission dynamics of various pathogens by accurately capturing the changing trend of viral effective reproductive number. Epidemic control and forecasting relies on accurate quantification of transmission and recovery dynamics. This quantification is achievable through the analysis of phylogenetic relationships among pathogen strains obtained from infected individuals. As a key analytical tool for such inference, we concentrate on the study of the episodic birth-death-sampling (EBDS) models. These models define a probability distribution on time-calibrated phylogenies that enable estimation of time-varying rates of pathogens' transmission, recovery, and sampling. Advances in sequencing technology, however, have led to an increasing amount of genetic data collected from these pathogens. Consequently, the traditional computational methods for analyzing these large-scale data under EBDS models have become inadequate due to their high computational load. We aimed to break this computational bottleneck by developing a new approach, based on the Hamiltonian Monte Carlo sampling, that considerably accelerates inference of all rate parameters compared to the traditional random-walk Metropolis-Hasting method. Our method greatly improves the ability to explore the complex distributions that arise when we want to understand disease dynamics at a more granular temporal resolution. This advancement delivers value to public health as it helps rapid data-driven decision-making during outbreaks and enhances our understanding of the spread of infectious diseases.

Gradient Boosting: A Computationally Efficient Alternative to Markov Chain Monte Carlo Sampling for Fitting Large Bayesian Spatio-Temporal Binomial Regression Models

Efficient History Matching Using the Markov-Chain Monte Carlo Method by Means of the Transformed Adaptive Stochastic Collocation Method

Predictive analytics with gradient boosting in clinical medicine

Extension of the Gradient Boosting Algorithm for Joint Modeling of Longitudinal and Time-to-Event data

Boosting domain-specific models with shrinkage: An application in mortality forecasting

Big problems in spatio-temporal disease mapping: methods and software

Bayesian group testing regression models for spatial data

A Large Scale Spatio-temporal Binomial Regression Model for Estimating Seroprevalence Trends

Multivariate Probabilistic Regression with Natural Gradient Boosting

Scalable gradients enable Hamiltonian Monte Carlo sampling for phylodynamic inference under episodic birth-death-sampling models

Bayesian hierarchical modeling for bivariate multiscale spatial data with application to blood test monitoring

Fast Bayesian estimation of spatial count data models

Modeling temporal gradients in regionally aggregated California asthma hospitalization data

Scalable Bayesian modeling for smoothing disease risks in large spatial data sets

Inference of Stochastic Disease Transmission Models Using Particle-MCMC and a Gradient Based Proposal

Gradient Boosting for Linear Mixed Models

Ensemble learning methods of inference for spatially stratified infectious disease systems

A Framework of Zero-Inflated Bayesian Negative Binomial Regression Models For Spatiotemporal Data

Time series analysis of hemorrhagic fever with renal syndrome in mainland China by using an XGBoost forecasting model

Spike and Slab Regression for Nonstationary Gaussian Linear Mixed Effects Modeling of Rapid Disease Progression

SurvBoost: An R Package for High-Dimensional Variable Selection in the Stratified Proportional Hazards Model via Gradient Boosting