Machine learning-based long-term outcome prediction in patients undergoing percutaneous coronary intervention

Abstract

Traditional prognostic risk assessment in patients with coronary artery disease undergoing percutaneous coronary intervention (PCI) is based on a limited selection of clinical and imaging findings. Machine learning (ML) can consider a higher number and complexity of variables and may be useful for characterising cardiovascular risk, predicting outcomes, and identifying biomarkers in large population studies. We prospectively enrolled 9,680 consecutive patients with coronary artery disease who underwent PCI at our institution between January 2013 and December 2013. Clinical features were selected and used to train 6 different ML models (support vector machine, decision tree, random forest, gradient boosting decision tree, neural network, and logistic regression) to predict cardiovascular outcomes, 10-fold cross-validation to evaluate the performance of models. During the 5-year follow-up, 467 (4.82%) patients died. Eighty-seven risk baseline measurements were used to train ML models. Compared with the other models, the random forest model (RF-PCI) exhibited the best performance on predicting all-cause mortality (area under the receiver operating characteristic curve: 0.71±0.04). Calibration plots demonstrated a slight overprediction for patients using the RF-PCI model (Hosmer-Lemeshow test: P>0.05). The top 15 features related to PCI candidates' long-term prognosis, among which 11 were laboratory measures. ML models improved the prediction of long-term all-cause mortality in patients with coronary artery disease before PCI. The performance of the RF model was better than that of the other models, providing a meaningful stratification.