Mohammed Saidul Islam, Iqram Hussain, Md Mezbaur Rahman, Se Jin Park, Md Azam Hossain
9 min
This study explores the application of machine learning (ML) models to automate the diagnosis of ischemic stroke using Electroencephalography (EEG) data. Recognizing that standard clinical neurological tests can be difficult to administer immediately post-stroke due to patient impairment, the authors assess whether EEG signals collected during active states—such as walking, working, or reading—can provide a reliable and non-invasive alternative. The research team evaluated three ensemble ML algorithms—Adaptive Gradient Boosting, XGBoost, and LightGBM—and integrated Explainable AI (XAI) tools, specifically Eli5 and LIME, to decipher how these models reach their diagnostic conclusions.
The results demonstrate that Adaptive Gradient Boosting is the most effective model among those tested, achieving an accuracy, AUC, and F1 score of approximately 80%. When the researchers applied XAI tools to understand the 'black-box' logic of these models, both Eli5 and LIME consistently highlighted specific spectral features as key contributors to stroke identification. Specifically, power spectral density features in the delta and theta frequency bands, particularly in frontal (Fz) and central (C1) cortical locations, were identified as the most influential factors for distinguishing stroke survivors from healthy control subjects.
By improving the interpretability of automated stroke diagnostics, this work addresses a critical barrier to the clinical adoption of AI: the lack of transparency in model decision-making. By identifying delta and theta waves as localized, clinically relevant biomarkers for stroke, the study provides a foundation for more accessible, real-time physiological monitoring. This capability could assist healthcare professionals in making more explainable diagnostic decisions and support more targeted, effective post-stroke rehabilitation and recovery programs.
State-of-the-art healthcare technologies are incorporating advanced Artificial Intelligence (AI) models, allowing for rapid and easy disease diagnosis. However, most AI models are considered “black boxes,” because there is no explanation for the decisions made by these models. Users may find it challenging to comprehend and interpret the results. Explainable AI (XAI) can explain the machine learning (ML) outputs and contribution of features in disease prediction models. Electroencephalography (EEG) is a potential predictive tool for understanding cortical impairment caused by an ischemic stroke and can be utilized for acute stroke prediction, neurologic prognosis, and post-stroke treatment. This study aims to utilize ML models to classify the ischemic stroke group and the healthy control group for acute stroke prediction in active states. Moreover, XAI tools (Eli5 and LIME) were utilized to explain the behavior of the model and determine the significant features that contribute to stroke prediction models. In this work, we studied 48 patients admitted to a hospital with acute ischemic stroke and 75 healthy adults who had no history of identified other neurological illnesses. EEG was obtained within three months following the onset of ischemic stroke symptoms using frontal, central, temporal, and occipital cortical electrodes (Fz, C1, T7, Oz). EEG data were collected in an active state (walking, working, and reading tasks). In the results of the ML approach, the Adaptive Gradient Boosting models showed around 80% accuracy for the classification of the control group and the stroke group. Eli5 and LIME were utilized to explain the behavior of the stroke prediction model and interpret the model locally around the prediction. The Eli5 and LIME interpretable models emphasized the spectral delta and theta features as local contributors to stroke prediction. From the findings of this explainable AI research, it is expected that the stroke-prediction XAI model will help with post-stroke treatment and recovery, as well as help healthcare professionals, make their diagnostic decisions more explainable.
Sam: Were the recordings taken before a stroke happened, or after the diagnosis? The authors call this stroke prediction, but those are different tasks.
Alex: After diagnosis, within three months. The stroke group included people across a range of severity. So the demonstrated task was distinguishing known stroke survivors from healthy adults, not predicting who would have a future stroke.
Sam: That also means the comparison isn’t against other neurological illnesses that might resemble stroke. What did the model actually receive from each recording?
Alex: The researchers used four scalp recording locations, covering frontal, central, temporal, and occipital regions. They divided the cleaned recordings into short time segments. For each segment, they measured power in different frequency bands: how much electrical activity fell within each range.
Sam: So instead of feeding in the entire waveform, they summarized its rhythms. Which rhythms become important later?
Alex: Delta and theta are slow-frequency bands of brain activity. The dataset also included faster bands, called alpha, beta, and gamma. Each band was measured at each recording location, giving the models a compact set of regional features.
Sam: And how did they turn those summaries into a decision? Was this a new classifier, or an application of existing methods?
Alex: They compared existing boosting methods, which combine learners that progressively address prediction errors. The methods were AdaBoost, XGBoost, and LightGBM. The researchers scaled the features, searched for suitable model settings, and reserved a portion of the feature data for testing.
Sam: Short segments raise a validation question. Were all recordings from a given person kept together, so the test set contained genuinely different participants?
Alex: The paper describes a training-and-test split of EEG feature data, but doesn’t specify participant-level separation. We therefore can’t establish from this account that the test participants were entirely unseen during training. That limits how confidently I’d interpret performance as generalization to new people.
Sam: With that qualification, what did the comparison show? Was there a clear winner or mostly similar performance?
Alex: AdaBoost led across the reported classification measures, while the other methods were close behind. Its stroke recall was 78 percent: the proportion of stroke examples it correctly identified. So even in this test, a meaningful share of stroke examples went unrecognized.
Sam: That makes the explanations useful for inspecting errors, not just celebrating correct predictions. How did the authors open up the models?
Alex: They used two explanation tools, Eli5 and LIME. Eli5 assigns contributions to features involved in a model’s decision. LIME means Local Interpretable Model-Agnostic Explanations: it approximates the model near one example with something simpler to interpret.
Sam: The introduction talks about explaining the whole model as well as individual predictions. Did the results actually show both levels?
Alex: The stated ambition includes whole-model explanations, but the presented examples emphasize individual test instances. Eli5 was applied to XGBoost and LightGBM, not the best-performing AdaBoost model, because of a library limitation. LIME was used with all the compared methods.
Sam: Give us one concrete prediction. What does the explanation add beyond a label saying “stroke”?
Alex: In the AdaBoost example, the model assigned an 86 percent probability to the stroke class. LIME highlighted frontal delta power as the strongest contributor toward that classification. Central delta and temporal theta also contributed, but less strongly.
Sam: That probability belongs to one example, then, not to the study’s overall reliability. And the explanation says what influenced the model, not what caused the person’s stroke.
Alex: Those are the boundaries. The explanation describes the classifier’s behavior around that example. It doesn’t establish a biological cause, and the paper doesn’t demonstrate that the probability is a calibrated estimate of clinical risk.
Sam: Did the different explanation tools converge on the same signals? Agreement would make the pattern more compelling.
Alex: Frontal delta was a recurring contributor, and the authors emphasize delta and theta overall. But the feature rankings weren’t identical. Some examples also gave substantial weight to gamma, so I’d describe the finding as a recurring pattern rather than one universal explanation.
Sam: How does that pattern fit the earlier EEG literature discussed in the paper?
Alex: The authors describe abnormal delta activity as associated with brain injury and cite earlier reports of increased delta power after stroke. Their discussion is more cautious about theta. Previous studies disagreed about its usefulness, although some found it informative for impairment or recovery.
Sam: So the explanations have some physiological plausibility, but that doesn’t eliminate the need to test them elsewhere. What limitation do the authors themselves emphasize?
Alex: Sparse scalp coverage. With only a few recording locations, they acknowledge the possibility of missing the damaged brain region. They propose extending the work with multiple kinds of biological signals, but that extension isn’t a result of this study.
Sam: Given those limits, who should read the full document, and where should they start?
Alex: Researchers developing EEG classifiers or explainable clinical machine learning should read it. Start with preprocessing and feature extraction, then the classification and model-explanation results. Check the test-split description and compare the individual feature rankings, rather than relying only on the headline accuracy.
Sam: And what should everyone else remember when an interpretable medical model is presented as a step toward diagnosis?
Alex: This study supports inspecting a stroke classifier, not yet deploying it as a clinical diagnostic tool. An explanation tells you why a model answered; validation tells you whether to trust that answer.
Sam: Keep those two questions separate.