Predictive modeling for stroke using routine clinical and socio-demographic data in Uganda

Stroke is a leading cause of mortality and disability in Uganda, with a prevalence of approximately 2.5% and a 25-30% 30-day mortality rate, exacerbated by limited access to predictive tools and advanced diagnostics. Machine learning (ML) offers potential for early stroke risk stratification, but its application in resource-constrained settings like Uganda requires tailored models and rigorous performance evaluation, as well as assessment of misclassification patterns. This study aimed to develop and internally validate an ML-based predictive model for stroke using clinical and socio-demographic characteristics among patients at Mulago National Referral Hospital (MNRH) and residents of Wakiso District, Uganda, identify key predictors and assess misclassification patterns to enhance targeted prevention. A retrospective case-control study utilized secondary data from 1,016 participants (254 stroke cases, 762 controls) across three prior studies conducted between 2006 and 2015. Eight ML algorithms were evaluated using 5-fold cross-validation, with performance metrics including accuracy, sensitivity, positive predictive value, F1-score, and area under receiver operator characteristics curve (AU-ROC). McNemars test compared model classification performance of the top 2 models. Feature importance identified key predictors, and misclassification patterns were analyzed using classification matrix and line graphs. The Random Forest (RF) model achieved the highest performance (95.18% accuracy, 90.10% F1-score, 97% AU-ROC). McNemars test confirmed RFs superior classification (p=0.01). Age, systolic blood pressure, total cholesterol, and waist-hip ratio were the top drivers of RF stroke predictions. RF misclassified 8.3% of stroke cases and 3.3% of non-stroke cases, due to overlaps in risk profiles (e.g., lower cholesterol, systolic blood pressure and age in false negatives). The RF model demonstrates strong potential for stroke risk prediction in Uganda, leveraging routine data to prioritize high-risk individuals. Misclassification analysis highlights areas for model refinement, supporting its potential for integration into clinical workflows for early intervention and stroke prevention in resource-limited settings. Future external validation across diverse populations is recommended.