Applying Ensemble Machine Learning Model to Explain the Causes of Stroke
摘要
In this study, we used a machine learning model based on ensemble learning algorithms to predict and identify stroke-causing factors using a dataset from Hospital 175 in Ho Chi Minh City, Vietnam. We statistically analyzed the data characteristics and correlated parameters affecting strokes. The training results revealed that the Categorical Boosting (CatBoost), Light Gradient Boosting Machine (LightGBM), and eXtreme Gradient Boosting (XGBoost) models demonstrated outstanding performance, achieving accuracies of 96.59%, 96.92%, and 96.57%, respectively. Additionally, “SHapley Additive exPlanations (SHAP)” were used to quantify each feature's contribution to predictions, helping identify stroke causes. These results can support integrating artificial intelligence into stroke diagnosis and treatment, paving the way for innovative healthcare solutions.