Heart Stroke Prediction Using Machine Learning

Stroke remains one of the leading causes of mortality and long-term disability worldwide, posing a significant burden on healthcare systems and society. Early identification of individuals at high risk of stroke is crucial for implementing preventive strategies and reducing fatal outcomes. This research proposes an intelligent stroke prediction system based on advanced machine learning techniques that analyse clinical and demographic data to assess stroke risk with high accuracy.

The proposed framework utilizes a structured healthcare dataset comprising key attributes such as age, hypertension, heart disease status, body mass index (BMI), average glucose level, smoking habits, and lifestyle factors. A comprehensive data preprocessing pipeline is implemented, including missing value imputation, categorical encoding, feature scaling, and class imbalance handling using resampling techniques. Multiple supervised learning algorithms, including Logistic Regression, Decision Tree, and Random Forest, are employed and comparatively evaluated to identify the most effective predictive model.

Experimental results demonstrate that ensemble-based models, particularly Random Forest, outperform other classifiers in terms of accuracy, precision, recall, and F1-score, achieving robust and reliable predictions. The model also incorporates feature importance analysis to interpret the contribution of critical risk factors, thereby enhancing transparency and clinical relevance.

The proposed system offers a scalable, cost-effective, and efficient solution for early stroke risk detection and can be integrated into modern healthcare infrastructures, including electronic health record systems and mobile health applications. Furthermore, this study highlights the potential of machine learning-driven predictive analytics in transforming preventive healthcare by enabling data-driven decision-making and personalized risk assessment.

Machine Learning, AI, Heart Stroke

1. S. Dritsas and M. Trigka, “Stroke risk prediction using machine learning techniques,” Applied Sciences, vol. 12, no. 3, pp. 1–15, 2022. [Google Scholar] [Crossref]

2. A. Kanwal, M. Aamir, and S. A. Khan, “An optimized machine learning framework for stroke prediction using feature extraction and SMOTE,” Procedia Computer Science, vol. 225, pp. 210–220, 2025. [Google Scholar] [Crossref]

3. R. K. Gupta and P. Sharma, “Machine learning approaches for healthcare analytics: A survey,” IEEE Access, vol. 9, pp. 157–170, 2021. [Google Scholar] [Crossref]

4. J. Chen, Y. Li, and H. Wang, “Stroke prediction using deep neural networks,” Expert Systems with Applications, vol. 168, pp. 114–123, 2021. [Google Scholar] [Crossref]

5. S. K. Mohapatra and B. Panda, “Comparative analysis of machine learning algorithms for stroke prediction,” International Journal of Medical Informatics, vol. 149, pp. 104–115, 2021. [Google Scholar] [Crossref]

6. World Health Organization, “Stroke, cerebrovascular accident,” WHO Report, 2021. [Google Scholar] [Crossref]

7. M. S. Rahman, M. Islam, and A. Hossain, “An intelligent stroke prediction system using machine learning,” IEEE Access, vol. 8, pp. 213–225, 2020. [Google Scholar] [Crossref]

8. T. Brown and L. Smith, “Healthcare prediction systems using artificial intelligence,” Journal of Biomedical Informatics, vol. 115, pp. 103–112, 2021. [Google Scholar] [Crossref]

9. N. Verma and S. Singh, “Machine learning-based predictive analytics in healthcare,” IEEE Transactions on Industrial Informatics, vol. 17, no. 9, pp. 624–633, 2021. [Google Scholar] [Crossref]

10. H. Patel and A. Roy, “Data preprocessing techniques in medical datasets,” Procedia Computer Science, vol. 173, pp. 63–70, 2020. [Google Scholar] [Crossref]

11. P. Kumar, R. Singh, and A. Sharma, “Random forest-based stroke prediction model,” International Journal of Advanced Computer Science and Applications, vol. 12, no. 4, pp. 145–152, 2021. [Google Scholar] [Crossref]

12. Y. Zhou, X. Liu, and Z. Chen, “Deep learning for healthcare data analysis,” Artificial Intelligence in Medicine, vol. 120, pp. 102–110, 2022. [Google Scholar] [Crossref]

13. S. Reddy and K. Nair, “AI-based medical diagnosis systems,” IEEE Reviews in Biomedical Engineering, vol. 14, pp. 456–468, 2021. [Google Scholar] [Crossref]

14. A. Sharma and K. Gupta, “Handling imbalanced datasets using SMOTE in healthcare applications,” Procedia Computer Science, vol. 218, pp. 256–263, 2023. [Google Scholar] [Crossref]

15. M. Khan and S. Das, “IoT-enabled healthcare monitoring and prediction systems,” IEEE Internet of Things Journal, vol. 8, no. 6, pp. 432–440, 2021. [Google Scholar] [Crossref]

16. D. Lee and K. Park, “Feature selection techniques in medical data mining,” ACM Computing Surveys, vol. 54, no. 3, pp. 1–35, 2021. [Google Scholar] [Crossref]

17. F. Zhao, L. Chen, and M. Xu, “Machine learning in predictive healthcare analytics: A review,” IEEE Access, vol. 10, pp. 56789–56800, 2022. [Google Scholar] [Crossref]

18. J. Wang, H. Zhang, and Y. Sun, “Explainable AI for medical decision support systems,” IEEE Transactions on Artificial Intelligence, vol. 4, no. 2, pp. 210–220, 2023. [Google Scholar] [Crossref]

19. R. Thomas and J. Mathew, “Cloud-based healthcare systems and predictive analytics,” IEEE Cloud Computing, vol. 8, no. 4, pp. 56–65, 2021. [Google Scholar] [Crossref]

20. A. Joshi and R. Kulkarni, “AI-driven healthcare systems for disease prediction,” International Journal of Healthcare Technology, vol. 15, no. 1, pp. 67–80, 2024. [Google Scholar] [Crossref]

• Performance Analysis of Silicon PV Array Using Infrared Thermography and Detecting Temperature Non-Uniformities
• Dynamic Capabilities and Digital Transformation Performance: The Role of Organizational Agility in the Public Sector
• Comparative Effectiveness of Neural Mobilization Versus Positional Release Technique in Cervical Radiculopathy – An Experimental Study
• Soil Moisture Mapping of Solano Nueva Vizcaya: A Comparison and Review
• Silent Assistance for Emergencies (SAFE): A Mobile-Based Emergency Reporting Application for the Deaf and Mute Community