Wei Xinyi, Shi Boyu
Universiti Malaya, Institute for Advanced Studies, Universiti Malaya, Kuala Lumpur, Wilayah Persekutuan Kuala Lumpur, Malaysia.
Universiti Malaya, Academy of Islamic Studies, Universiti Malaya, Kuala Lumpur, Wilayah Persekutuan Kuala Lumpur, Malaysia.
PLoS One. 2025 Aug 7;20(8):e0327569. doi: 10.1371/journal.pone.0327569. eCollection 2025.
Coronary heart disease (CHD) is a major cardiovascular disorder that poses significant threats to global health and is increasingly affecting younger populations. Its treatment and prevention face challenges such as high costs, prolonged recovery periods, and limited efficacy of traditional methods. Additionally, the complexity of diagnostic indicators and the global shortage of medical professionals further complicate accurate diagnosis. This study employs machine learning techniques to analyze CHD-related pathogenic factors and proposes an efficient diagnostic and predictive framework. To address the data imbalance issue, SMOTE-ENN is utilized, and five machine learning algorithms-Decision Trees, KNN, SVM, XGBoost, and Random Forest-are applied for classification tasks. Principal Component Analysis (PCA) and Grid Search are used to optimize the models, with evaluation metrics including accuracy, precision, recall, F1-score, and AUC. According to the random forest model's optimization experiment, the initial unbalanced data's accuracy was 85.26%, and the F1-score was 12.58%. The accuracy increased to 92.16% and the F1-score reached 93.85% after using SMOTE-ENN for data balancing, which is an increase of 6.90% and 81.27%, respectively; the model accuracy increased to 97.91% and the F1-score increased to 97.88% after adding PCA feature dimensionality reduction processing, which is an increase of 5.75% and 4.03%, respectively, compared with the SMOTE-ENN stage. This indicates that combining data balancing and feature dimensionality reduction techniques significantly improves model accuracy and makes the random forest model the best model. This study provides an efficient diagnostic tool for CHD, alleviates the challenges posed by limited medical resources, and offers a scientific foundation for precise prevention and intervention strategies.
冠心病(CHD)是一种主要的心血管疾病,对全球健康构成重大威胁,并且越来越多地影响着年轻人群。其治疗和预防面临着诸如成本高昂、恢复期延长以及传统方法疗效有限等挑战。此外,诊断指标的复杂性以及全球医疗专业人员的短缺进一步使准确诊断变得复杂。本研究采用机器学习技术来分析与冠心病相关的致病因素,并提出了一个高效的诊断和预测框架。为了解决数据不平衡问题,使用了SMOTE-ENN,并且应用了五种机器学习算法——决策树、KNN、支持向量机、XGBoost和随机森林——来进行分类任务。主成分分析(PCA)和网格搜索用于优化模型,评估指标包括准确率、精确率、召回率、F1分数和AUC。根据随机森林模型的优化实验,初始不平衡数据的准确率为85.26%,F1分数为12.58%。在使用SMOTE-ENN进行数据平衡后,准确率提高到92.16%,F1分数达到93.85%,分别提高了6.90%和81.27%;在添加PCA特征降维处理后,模型准确率提高到97.91%,F1分数提高到97.88%,与SMOTE-ENN阶段相比,分别提高了5.75%和4.03%。这表明结合数据平衡和特征降维技术显著提高了模型准确率,并使随机森林模型成为最佳模型。本研究为冠心病提供了一种高效的诊断工具,缓解了医疗资源有限带来的挑战,并为精确的预防和干预策略提供了科学依据。