Variable selection with missing data in both covariates and outcomes: Imputation and machine learning

Examining the Trajectory of Health-Related Quality of Life among Coronavirus Disease Patients

BACKGROUND

Recent studies have reported a reduction in health-related quality of life (HR-QoL) among post-coronavirus disease 2019 (COVID-19) patients. However, there remains a gap in research examining the heterogeneity and determinants of HR-QoL trajectory in these patients.

OBJECTIVE

To describe and identify factors explaining the variability in HR-QoL trajectories among a cohort of patients with history of COVID-19.

DESIGN

A prospective study using data from a cohort of COVID-19 patients enrolled into a registry established at a health system in New York City.

PARTICIPANTS

Participants were enrolled from July 2020 to June 2022, and completed a baseline evaluation and two follow-up visits at 6 and 12 months.

METHODS

We assessed HR-QoL with the 29-item Patient Reported Outcomes Measurement Information System instrument, which was summarized into mental and physical health domains. We performed latent class growth and multinomial logistic regression to examine trajectories of HR-QoL and identify factors associated with specific trajectories.

RESULTS

The study included 588 individuals with a median age of 52 years, 65% female, 54% White, 18% Black, and 18% Hispanic. We identified five physical health trajectories and four mental health trajectories. Female gender, having pre-existing hypertension, cardiovascular disease, asthma, and hospitalization for acute COVID-19 were independently associated with lower physical health. In addition, patients with increasing body mass index were more likely to experience lower physical health over time. Female gender, younger age, pre-existing asthma, arthritis and cardiovascular disease were associated with poor mental health.

CONCLUSIONS

We found significant heterogeneity of HR-QoL after COVID-19, with women and patients with specific comorbidities at increased risk of lower HR-QoL. Implementation of targeted psychological and physical interventions is crucial for enhancing the quality of life of this patient population.

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

Clinical trials evaluate the safety and efficacy of treatments for specific diseases. Ensuring these studies are well-powered is crucial for identifying superior treatments. With the rise of personalized medicine, treatment efficacy may vary based on biomarker profiles. However, researchers often lack prior knowledge about which biomarkers are linked to varied treatment effects. Fixed or response-adaptive designs may not sufficiently account for heterogeneous patient characteristics, such as genetic diversity, potentially reducing the chance of selecting the optimal treatment for individuals. Recent advances in Bayesian nonparametric modeling pave the way for innovative trial designs that not only maintain robust power but also offer the flexibility to identify subgroups deriving greater benefits from specific treatments. Building on this inspiration, we introduce a Bayesian adaptive design for multi-arm trials focusing on time-to-event endpoints. We introduce a covariate-adjusted response adaptive randomization, updating treatment allocation probabilities grounded on causal effect estimates using a random intercept accelerated failure time BART model. After the trial concludes, we suggest employing a multi-response decision tree to pinpoint subgroups with varying treatment impacts. The performance of our design is then assessed via comprehensive simulations.

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

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

Using Tree-Based Machine Learning for Health Studies: Literature Review and Case Series

Tree-based machine learning methods have gained traction in the statistical and data science fields. They have been shown to provide better solutions to various research questions than traditional analysis approaches. To encourage the uptake of tree-based methods in health research, we review the methodological fundamentals of three key tree-based machine learning methods: random forests, extreme gradient boosting and Bayesian additive regression trees. We further conduct a series of case studies to illustrate how these methods can be properly used to solve important health research problems in four domains: variable selection, estimation of causal effects, propensity score weighting and missing data. We exposit that the central idea of using ensemble tree methods for these research questions is accurate prediction via flexible modeling. We applied ensemble trees methods to select important predictors for the presence of postoperative respiratory complication among early stage lung cancer patients with resectable tumors. We then demonstrated how to use these methods to estimate the causal effects of popular surgical approaches on postoperative respiratory complications among lung cancer patients. Using the same data, we further implemented the methods to accurately estimate the inverse probability weights for a propensity score analysis of the comparative effectiveness of the surgical approaches. Finally, we demonstrated how random forests can be used to impute missing data using the Study of Women's Health Across the Nation data set. To conclude, the tree-based methods are a flexible tool and should be properly used for health investigations.

A Flexible Approach for Assessing Heterogeneity of Causal Treatment Effects on Patient Survival Using Large Datasets with Clustered Observations

Personalized medicine requires an understanding of treatment effect heterogeneity. Evolving toward causal evidence for scenarios not studied in randomized trials necessitates a methodology using real-world evidence. Herein, we demonstrate a methodology that generates causal effects, assesses the heterogeneity of the effects and adjusts for the clustered nature of the data. This study uses a state-of-the-art machine learning survival model, riAFT-BART, to draw causal inferences about individual survival treatment effects, while accounting for the variability in institutional effects; further, it proposes a data-driven approach to agnostically (as opposed to a priori hypotheses) ascertain which subgroups exhibit an enhanced treatment effect from which intervention, relative to global evidence-average treatment effects measured at the population level. Comprehensive simulations show the advantages of the proposed method in terms of bias, efficiency and precision in estimating heterogeneous causal effects. The empirically validated method was then used to analyze the National Cancer Database.

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

When drawing causal inferences about the effects of multiple treatments on clustered survival outcomes using observational data, we need to address implications of the multilevel data structure, multiple treatments, censoring, and unmeasured confounding for causal analyses. Few off-the-shelf causal inference tools are available to simultaneously tackle these issues. We develop a flexible random-intercept accelerated failure time model, in which we use Bayesian additive regression trees to capture arbitrarily complex relationships between censored survival times and pre-treatment covariates and use the random intercepts to capture cluster-specific main effects. We develop an efficient Markov chain Monte Carlo algorithm to draw posterior inferences about the population survival effects of multiple treatments and examine the variability in cluster-level effects. We further propose an interpretable sensitivity analysis approach to evaluate the sensitivity of drawn causal inferences about treatment effect to the potential magnitude of departure from the causal assumption of no unmeasured confounding. Expansive simulations empirically validate and demonstrate good practical operating characteristics of our proposed methods. Applying the proposed methods to a dataset on older high-risk localized prostate cancer patients drawn from the National Cancer Database, we evaluate the comparative effects of three treatment approaches on patient survival, and assess the ramifications of potential unmeasured confounding. The methods developed in this work are readily available in the R $$ \mathsf{R}\kern.15em $$ package riAFTBART $$ \mathsf{riAFTBART} $$ .

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

Background

Prior work has shown that combining bootstrap imputation with tree-based machine learning variable selection methods can provide good performances achievable on fully observed data when covariate and outcome data are missing at random (MAR). This approach however is computationally expensive, especially on large-scale datasets.

Methods

We propose an inference-based method, called RR-BART, which leverages the likelihood-based Bayesian machine learning technique, Bayesian additive regression trees, and uses Rubin’s rule to combine the estimates and variances of the variable importance measures on multiply imputed datasets for variable selection in the presence of MAR data. We conduct a representative simulation study to investigate the practical operating characteristics of RR-BART, and compare it with the bootstrap imputation based methods. We further demonstrate the methods via a case study of risk factors for 3-year incidence of metabolic syndrome among middle-aged women using data from the Study of Women’s Health Across the Nation (SWAN).

Results

The simulation study suggests that even in complex conditions of nonlinearity and nonadditivity with a large percentage of missingness, RR-BART can reasonably recover both prediction and variable selection performances, achievable on the fully observed data. RR-BART provides the best performance that the bootstrap imputation based methods can achieve with the optimal selection threshold value. In addition, RR-BART demonstrates a substantially stronger ability of detecting discrete predictors. Furthermore, RR-BART offers substantial computational savings. When implemented on the SWAN data, RR-BART adds to the literature by selecting a set of predictors that had been less commonly identified as risk factors but had substantial biological justifications.

Conclusion

The proposed variable selection method for MAR data, RR-BART, offers both computational efficiency and good operating characteristics and is utilitarian in large-scale healthcare database studies.

Supplementary Information

The online version contains supplementary material available at (10.1186/s12874-022-01608-7).