Predicting Cardiovascular Disease Risk Using Tree-Based Gradient Boosting Machine Learning Techniques
摘要
Heart disease prediction is a significant challenge in medical science, necessitating robust and interpretable machine learning models. This study utilized tree-based boosting techniques, specifically XGBoost, CatBoost, and LightGBM (LGBM), to develop predictive models for heart disease detection using a cardiovascular dataset. Preprocessing steps included missing value imputation using K-nearest neighbors (kNN) and normalization for optimal model performance. Recursive feature elimination (RFE) was employed for feature selection, enhancing model efficiency. Model performance was assessed through training and testing splits with 5-fold cross-validation. The results indicated that LGBM outperformed the other model, achieving a testing accuracy of 96.55%, precision of 96.73%, recall of 96.82%, and F1-score of 96.68%. These findings suggest that tree-based boosting models are effective for heart disease prediction, enhancing clinical decision-making.