Machine learning algorithm as a prognostic tool for venous thromboembolism in allogeneic transplant patients

As a serious complication after allogenic hematopoietic stem cell transplantation (allo-HSCT), venous thromboembolism (VTE) is significantly related to increased nonrelapse mortality. Therefore distinguishing patients at high risk of death who should receive specific therapeutic management is key to improving survival. This study aimed to establish a machine learning-based prognostic model for the identification of post-transplantation VTE patients who have a high risk of death. We retrospectively evaluated 256 consecutive VTE patients who underwent allo-HSCT at our center between 2008 and 2019. These patients were further randomly divided into (1) a derivation (80%) cohort of 205 patients and (2) a test (20%) cohort of 51 patients. The least absolute shrinkage and selection operator (LASSO) approach was used to choose the potential predictors from the primary dataset. Eight machine learning classifiers were used to produce 8 candidate models. A 10-fold cross-validation procedure was used to internally evaluate the models and to select the best-performing model for external assessment using the test cohort. In total, 256 of 7238 patients were diagnosed with VTE after transplantation. Among them, 118 patients (46.1%) had catheter-related venous thrombosis, 107 (41.8%) had isolated deep-vein thrombosis (DVT), 20 (7.8%) had isolated pulmonary embolism (PE), and 11 (4.3%) had concomitant DVT and PE. The 2-year overall survival (OS) rate of patients with VTE was 68.8%. Using LASSO regression, 8 potential features were selected from the 54 candidate variables. The best-performing algorithm based on the 10-fold cross-validation runs was a logistic regression classifier. Therefore a prognostic model named BRIDGE was then established to predict the 2-year OS rate. The areas under the curves of the BRIDGE model were 0.883, 0.871, and 0.858 for the training, validation, and test cohorts, respectively. The Hosmer-Lemeshow goodness-of-fit test showed a high agreement between the predicted and observed outcomes. Decision curve analysis indicated that VTE patients could benefit from the clinical application of the prognostic model. A BRIDGE risk score calculator for predicting the study result is available online (47.94.162.105:8080/bridge/). We established the BRIDGE model to precisely predict the risk for all-cause death in VTE patients after allo-HSCT. Identifying VTE patients who have a high risk of death can help physicians treat these patients in advance, which will improve patient survival.