Exploring brain lobe-specific insights in an explainable framework for EEG-based schizophrenia detection

Abstract

Schizophrenia (ScZ) is a growing global health concern that affects millions of people and puts severe pressure on healthcare systems. Early detection and accurate diagnosis are crucial for adequate management. Electroencephalography (EEG) has evolved into a promising non-invasive tool for detecting ScZ in contemporary research. However, specific biomarkers, especially those related to brain lobes, cannot often be identified by current EEG-based diagnostic methods. Different brain lobes are associated with distinct cognitive functions and patterns of diseases. Also, there is a gap in the incorporation of the XAI technique, as medical diagnosis needs trustworthiness and explainability. This study strives to address these gaps by developing a framework using mel-spectrogram images with Convolutional Neural Networks (CNNs). EEG signals are converted into mel-spectrogram images using Short-Time Fourier Transform (STFT). After that, these images are analyzed using a CNN model to perform classification between ScZ and healthy control (HC). To identify the most critical brain regions, the full brain regions are divided into five different regions, and the same classification process is performed. The performance of the proposed framework is evaluated using two publicly available EEG datasets: repOD and the kaggle basic sensory task dataset, which provides a remarkable accuracy of 99.82% and 98.31% respectively. Among regions, the frontal lobe has the most significant performance with an accuracy of 97.02% and 88.03%, respectively, in these datasets, followed by the temporal lobe. Conversely, the occipital lobe shows the lowest accuracy among lobes, with only 79.30 % and 68.33% accuracy on both occasions, showing its lower significance in the diagnosis. To bring result explainability, LIME, SHAP, and the Grad-CAM methods are applied, providing valuable insights for clinicians and researchers. These findings emphasize the potential of EEG-based brain lobe analysis in enhancing ScZ detection, diagnostic accuracy, explainability, and clinical guidance.