Classification of coronary artery disease using radial artery pulse wave analysis via machine learning
Yi Lyu,Hai-Mei Wu,Hai-Xia Yan,Rui Guo,Yu-Jie Xiong,Rui Chen,Wen-Yue Huang,Jing Hong,Rong Lyu,Yi-Qin Wang,Jin Xu
DOI: https://doi.org/10.1186/s12911-024-02666-1
2024-09-16
Abstract:Background: Coronary artery disease (CAD) is a major global cardiovascular health threat and the leading cause of death in many countries. The disease has a significant impact in China, where it has become the leading cause of death. There is an urgent need to develop non-invasive, rapid, cost-effective, and reliable techniques for the early detection of CAD using machine learning (ML). Methods: Six hundred eight participants were divided into three groups: healthy, hypertensive, and CAD. The raw data of pulse wave from those participants was collected. The data were de-noised, normalized, and analyzed using several applications. Seven ML classifiers were used to model the processed data, including Decision Tree (DT), Random Forest (RF), Gradient Boosting Decision Tree (GBDT), Extra Trees (ET), Extreme Gradient Boosting (XGBoost), Light Gradient Boosting (LightGBM), and Unbiased Boosting with Categorical Features (CatBoost). Results: The Extra Trees classifier demonstrated the best classification performance. After tunning, the results performance evaluation on test set are: 0.8579 accuracy, 0.9361 AUC, 0.8561 recall, 0.8581 precision, 0.8571 F1 score, 0.7859 kappa coefficient, and 0.7867 MCC. The top 10 feature importances of ET model are w/t1, t3/tmax, tmax, t3/t1, As, hf/3, tf/3/tmax, tf/5, w and tf/3/t1. Conclusion: Radial artery pulse wave can be used to identify healthy, hypertensive and CAD participants by using Extra Trees Classifier. This method provides a potential pathway to recognize CAD patients by using a simple, non-invasive, and cost-effective technique.