Lyu Y, Wu HM, Yan HX, Guo R, Xiong YJ, Chen R, Huang WY, Hong J, Lyu R, Wang YQ, Xu J. Classification of coronary artery disease using radial artery pulse wave analysis via machine learning.
BMC Med Inform Decis Mak 2024;
24:256. [PMID:
39285363 PMCID:
PMC11403788 DOI:
10.1186/s12911-024-02666-1]
[Citation(s) in RCA: 0] [Impact Index Per Article: 0] [Reference Citation Analysis] [Abstract] [Key Words] [MESH Headings] [Grants] [Download PDF] [Figures] [Journal Information] [Subscribe] [Scholar Register] [Received: 07/02/2024] [Accepted: 09/05/2024] [Indexed: 09/22/2024] Open
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.
Collapse