Abstract:We recently developed a new method random-intercept accelerated failure time model with Bayesian additive regression trees (riAFT-BART) to draw causal inferences about population treatment effect on patient survival from clustered and censored survival data while accounting for the multilevel data structure. The practical utility of this method goes beyond the estimation of population average treatment effect. In this work, we exposit how riAFT-BART can be used to solve two important statistical questions with clustered survival data: estimating the treatment effect heterogeneity and variable selection. Leveraging the likelihood-based machine learning, we describe a way in which we can draw posterior samples of the individual survival treatment effect from riAFT-BART model runs, and use the drawn posterior samples to perform an exploratory treatment effect heterogeneity analysis to identify subpopulations who may experience differential treatment effects than population average effects. There is sparse literature on methods for variable selection among clustered and censored survival data, particularly ones using flexible modeling techniques. We propose a permutation-based approach using the predictor's variable inclusion proportion supplied by the riAFT-BART model for variable selection. To address the missing data issue frequently encountered in health databases, we propose a strategy to combine bootstrap imputation and riAFT-BART for variable selection among incomplete clustered survival data. We conduct an expansive simulation study to examine the practical operating characteristics of our proposed methods, and provide empirical evidence that our proposed methods perform better than several existing methods across a wide range of data scenarios. Finally, we demonstrate the methods via a case study of predictors for in-hospital mortality among severe COVID-19 patients and estimating the heterogeneous treatment effects of three COVID-specific medications. The methods developed in this work are readily available in the R ${\textsf {R}}$ package riAFTBART $\textsf {riAFTBART}$ .

A new method for clustered survival data: Estimation of treatment effect heterogeneity and variable selection

A flexible approach for causal inference with multiple treatments and clustered survival outcomes

A flexible approach for variable selection in large-scale healthcare database studies with missing covariate and outcome data

A Bayesian machine learning approach for estimating heterogeneous survivor causal effects: Applications to a critical care trial

Nonparametric failure time: Time‐to‐event machine learning with heteroskedastic bayesian additive regression trees and low information omnibus dirichlet process mixtures

Targeted Estimation of Heterogeneous Treatment Effect in Observational Survival Analysis

Variable Selection with Random Survival Forest and Bayesian Additive Regression Tree for Survival Data

Estimation of Causal Effects of Multiple Treatments in Observational Studies with a Binary Outcome

Nonparametric competing risks analysis using Bayesian Additive Regression Trees (BART)

Survival Analysis Using a 5-Step Stratified Testing and Amalgamation Routine in Randomized Clinical Trials

Subgroup identification in clinical trials: an overview of available methods and their implementations with R

Leveraging machine learning: Covariate-adjusted Bayesian adaptive randomization and subgroup discovery in multi-arm survival trials

Estimation of causal effects of multiple treatments in healthcare database studies with rare outcomes

Seemingly unrelated Bayesian additive regression trees for cost-effectiveness analyses in healthcare

Individualized Treatment Effects with Censored Data via Fully Nonparametric Bayesian Accelerated Failure Time Models

Incorporating External Data into the Analysis of Clinical Trials via Bayesian Additive Regression Trees

Robust method for optimal treatment decision making based on survival data

Combining Survival Trials Using Aggregate Data Based on Misspecified Models

Dynamic Treatment Regimes Using Bayesian Additive Regression Trees for Censored Outcomes

The application of Bayesian method in estimating heterogeneity of treatment effect

Using Recursive Partitioning to Find and Estimate Heterogenous Treatment Effects In Randomized Clinical Trials