Mediation analysis of community context effects on heart failure using the survival R2D2 prior

TL;DR

This paper introduces a Bayesian variable selection method using a R2D2 prior for survival analysis, improving identification of risk factors for heart failure from large health records.

Contribution

It proposes a novel R2D2 prior approach for Bayesian survival regression that enhances variable selection and computational efficiency.

Findings

The method outperforms existing variable selection techniques in simulations.

Applied to real data, it identified socioeconomic factors influencing heart failure risk.

Higher socioeconomic inequality is associated with increased heart failure risk.

Abstract

Congestive heart failure (CHF) is a leading cause of morbidity, mortality and healthcare costs, impacting $>$23 million individuals worldwide. Large electronic health records data provide an opportunity to improve clinical management of diseases, but statistical inference on large amounts of relevant personal data is still challenging. Thus, accurately identifying influential risk factors is pivotal to reducing information dimensionality. Bayesian variable selection in survival regression is a common approach towards solving this problem. Here, we propose placing a beta prior directly on the model coefficient of determination (Bayesian $R^2$), which induces a prior on the global variance of the predictors and provides shrinkage. Through reparameterization using an auxiliary variable, we are able to update a majority of the parameters with Gibbs sampling, simplifying computation and quickening convergence. Performance gains over competing variable selection methods are showcased through an extensive simulation study. Finally, the method is applied in a mediation analysis to identify community context attributes impacting time to first congestive heart failure diagnosis of patients enrolled in University of North Carolina Cardiovascular Device Surveillance Registry. The model has high predictive performance and we find that factors associated with higher socioeconomic inequality increase risk of heart failure.