Predicting Measles Vaccination Defaulting in Uganda and Kenya: Comparative Evaluation of Custom Soft Voting and TabNet Stacked Ensemble Architectures

Background Measles is a major cause of death among infants despite free routine immunization (RI) in low- and middle-income countries (LMICs). Vaccination defaulting contributes to measles outbreak occurrences. Conventional immunization defaulter identification methods are expensive to implement, prone to data falsification, reactive, and overestimate uptake. Several machine learning studies leveraged the use of child and maternal characteristics as predictors of immunization defaulting, overlooking community, d…

Predicting Measles Vaccination Defaulting in Uganda and Kenya: Comparative Evaluation of Custom Soft Voting and TabNet Stacked Ensemble Architectures
United States Naval Medical Bulletin Vol. 6, Nos. 1-4, 1912 by U.S. Navy. Bureau of Medicine and Surgery. Public domain

In brief

Researchers benchmarked nine classical and deep learning classifiers on Demographic and Health Survey data from 2016 to 2022 to predict which infants in Kenya and Uganda would default on measles vaccination. They paired a custom soft voting ensemble and a TabNet stacked ensemble with SHAP explanations, and built a tool using Streamlit and SQLite.

In Kenya the voting ensemble reached an AUC of 0.8847, and in Uganda the stacked hybrid reached 0.8862. The results matter because conventional defaulter identification is costly, reactive and prone to falsification, so a predictive approach could help target outreach. How well the models work in routine practice remains untested in the source.

Main points

  1. Kenya's best model was a TabNet+GB+RF custom soft voting ensemble with an F1-score of 0.7111 and an AUC of 0.8847.
  2. Uganda's best model was a TabNet hybrid ensemble (GB+RF+AB→TabNet) with an AUC of 0.8862 and an F1-score of 0.6756.
  3. SHAP identified child's age and vitamin A1 as key predictors in both countries, while other predictors differed.
  4. The study recommends outreach to areas with low literacy, distant water sources and mobile populations in Kenya, and reminders plus district outreach in Uganda.

The gain

Ensemble models combining TabNet with tree-based classifiers identified likely measles vaccine defaulters better than individual models, and SHAP pinpointed country-specific drivers that could guide targeted outreach.

The rundown

The pipeline used feature selection from literature and expert input, vaccination eligibility filtering, missing value handling and phi-k correlation-based feature elimination. Class imbalance was addressed with SMOTE for nominal continuous features after RobustScaler, and models were tuned using stratified k-fold cross-validation and grid search.

In Kenya the most influential features were time to water source, respondents' education, previous district and current district. In Uganda they were polio inactive, Pentavalent3, current district and respondents' education.

The authors recommend country-tailored strategies: outreach in Kenya to low-literacy, distant-water and mobile areas, and in Uganda reminders to avoid missing preceding vaccines and outreach to high-default districts.

Sources

  1. Peer-reviewedJMIR AI2026-10-13

The debate