Mortality Prediction through Machine Learning: Insights from XGBoost and Random Forest Model
This study explores the use of machine learning algorithms to predict deaths caused by different risk factors in various countries. Using a large dataset that covers many years and regions, we applied models like XGBoost and Random Forest to see how well they could predict death rates based on health and environmental factors. The XGBoost model was very accurate with an $\text{R}^{2}$ of $\mathbf{0. 9 9 5 6}$, showing it could explain most of the variability in death rates. The Random Forest model also performed well with an $\text{R}^{2}$ of 0.9926. These findings show that machine learning can be very useful in health analysis, helping to understand the risk factors that lead to higher death rates. The study discusses how these results can inform public health policies and guide future research.
Authors
- Praveen Kumar Misra
- Vinay K. Tiwari (ORCID: https://orcid.org/0000-0001-5919-4573)
- Rahul Gupta
- Barkha Gupta
Institutions
- Dr. Shakuntala Misra National Rehabilitation University (IN)
Publication Details
- Published
- 2024-09-12
- DOI
- https://doi.org/10.1109/sparc61891.2024.10829275
- Primary Topic
- Artificial Intelligence in Healthcare
- Type
- article
- Field-Weighted Citation Impact
- 0.00