Stacked Inverse Probability of Censoring Weighted Bagging: A Case Study In the InfCareHIV Register. Issue 1 (20th January 2021)

• Record Type: Journal Article
• Authors: Gonzalez Ginestet, Pablo; Kotalik, Ales; Vock, David M.; Wolfson, Julian; Gabriel, Erin E.
• Is Part Of: Journal of the Royal Statistical Society. Volume 70:Issue 1(2021)
• Page Start: 51
• Page End: 65
• Publication Date: 2021-01-20
• Subjects: competing risk -- cumulative incidence -- ensemble learning -- HIV treatment -- inverse probability of censoring weighting -- machine learning -- stacking; Statistics -- Periodicals; 519.5
• DOI: 10.1111/rssc.12448
• Languages: English
• ISSNs: 0035-9254

Abstract

Abstract: We propose an inverse probability of censoring weighted (IPCW) bagging (bootstrap aggregation) pre-processing that enables the application of any machine learning procedure for classification to be used to predict the cause-specific cumulative incidence, properly accounting for right-censored observations and competing risks. We consider the IPCW area under the time-dependent ROC curve (IPCW-AUC) as a performance evaluation metric. We also suggest a procedure to optimally stack predictions from any set of IPCW bagged methods. We illustrate our proposed method in the Swedish InfCareHIV register by predicting individuals for whom treatment will not maintain an undetectable viral load for at least 2 years following initial suppression. The R package stackBagg that implements our proposed method is available on Github.