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}$ .

Broken Adaptive Ridge Regression for Right-Censored Survival Data

Censored broken adaptive ridge regression in high-dimension

Broken Adaptive Ridge Method for Variable Selection in Generalized Partly Linear Models with Application to the Coronary Artery Disease Data

Penalized Variable Selection with Broken Adaptive Ridge Regression for Semi-competing Risks Data

Copula-based Semiparametric Nonnormal Transformed Linear Model for Survival Data with Dependent Censoring

Right-censored models by the expectile method

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

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

High-dimensional robust inference for censored linear models

Regularized Buckley-James method for right-censored outcomes with block-missing multimodal covariates

Semiparametric Bayesian Inference for Accelerated Failure Time Models with Errors-in-covariates and Doubly Censored Data

On the Maximum Penalized Full Likelihood Approach for Cox Model with Extreme Value for Heavily Censored Survival Data

Regression Analysis of Right-Censored Failure Time Data with Missing Censoring Indicators

On Least-Squares Regression with Censored Data

Rank-based Variable Selection with Censored Data

Optimal subsampling for semi‐parametric accelerated failure time models with massive survival data using a rank‐based approach

Group lasso priors for Bayesian accelerated failure time models with left-truncated and interval-censored data

A fast algorithm for the accelerated failure time model with high-dimensional time-to-event data

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

SurvRegCensCov: Weibull Regression for a Right-Censored Endpoint with a Censored Covariate

Penalized Bayesian forward continuation ratio model with application to high-dimensional data with discrete survival outcomes