Ensemble Multi-task Learning Approach for Explainable EEG-Based Stroke Prediction
摘要
Stroke is a major cause of long-term disability. Early and accurate diagnosis of stroke severity can improve patient outcomes. EEG is a non-invasive way to analyze brain activity changes during stroke, but interpreting complex EEG data remains challenging. This study develops an explainable multi-task learning approach for EEG-based stroke analysis. Feedforward neural networks extract hierarchical spatial-temporal EEG features. An ensemble of models with different architectures and electrode combinations is trained for robustness. A multi-task framework jointly models stroke diagnosis, type, location, and severity prediction with high accuracies of 98.11%, 100%, 100%, and 100% respectively. Explainability methods like LIME and SHAP are used to visualize influential EEG channels and time windows for each prediction. The models are evaluated on a public stroke EEG dataset and achieve state-of-the-art performance on multi-label classification and severity regression. Explanation methods provide clinically interpretable insights into key EEG patterns underlying decision-making. This presents an effective and transparent framework for multi-faceted EEG-based stroke analysis to support clinical decisions.