CALSczNet: Convolution Neural Network with Attention and LSTM for the Detection of Schizophrenia Using EEG Signals

Almaghrabi,Hussain,Alotaibi
DOI: https://doi.org/10.3390/math12131989
IF: 2.4
2024-06-28
Mathematics
Abstract:Schizophrenia (SZ) is a serious psychological disorder that affects nearly 1% of the global population. The progression of SZ disorder causes severe brain damage; its early diagnosis is essential to limit adverse effects. Electroencephalography (EEG) is commonly used for SZ detection, but its manual screening is laborious, time-consuming, and subjective. Automatic methods based on machine learning have been introduced to overcome these issues, but their performance is not satisfactory due to the non-stationary nature of EEG signals. To enhance the detection performance, a novel deep learning-based method is introduced, namely, CALSczNet. It uses temporal and spatial convolutions to learn temporal and spatial patterns from EEG trials, uses Temporal Attention (TA) and Local Attention (LA) to adaptively and dynamically attend to salient features to tackle the non-stationarity of EEG signals, and finally, it employs Long Short-Term Memory (LSTM) to work out the long-range dependencies of temporal features to learn the discriminative features. The method was evaluated on the benchmark public-domain Kaggle dataset of the basic sensory tasks using 10-fold cross-validation. It outperforms the state-of-the-art methods on all conditions with 98.6% accuracy, 98.65% sensitivity, 98.72% specificity, 98.72% precision, and an F1-score of 98.65%. Furthermore, this study suggested that the EEG signal of the subject performing either simultaneous motor and auditory tasks or only auditory tasks provides higher discriminative features to detect SZ in patients. Finally, it is a robust, effective, and reliable method that will assist psychiatrists in detecting SZ at an early stage and provide suitable and timely treatment.
mathematics
What problem does this paper attempt to address?
### What problem does this paper attempt to solve? This paper aims to address the problem of automatic detection of Schizophrenia (SZ), particularly using Electroencephalogram (EEG) signals. Specifically, the study proposes a new deep learning model named CALSczNet, which improves existing detection methods through the following aspects: 1. **Automatic Feature Extraction**: - Utilizing Convolutional Neural Networks (CNN) to extract spatiotemporal features from EEG signals. - Using Long Short-Term Memory (LSTM) networks to capture long-term dependencies in time series. 2. **Attention Mechanism**: - Introducing two types of attention mechanisms (Temporal Attention TA and Local Attention LA) to dynamically focus on important features in EEG signals and ignore unimportant parts. 3. **Lightweight Design**: - Adopting depth-wise separable convolution layers to reduce the number of parameters and prevent overfitting. 4. **Data Augmentation**: - Proposing a data augmentation technique to increase the number of training samples, further improving the model's robustness and generalization ability. ### Main Contributions 1. **Proposing the CALSczNet Model**: This model effectively extracts spatiotemporal features and reduces the number of parameters through depth-wise separable convolution layers. 2. **Introducing Two Attention Mechanisms**: Temporal Attention (TA) to optimize low-level temporal features and Local Attention (LA) to optimize intermediate-level spatial features. 3. **Using LSTM to Capture Long-Term Dependencies**: Accelerates the classification process. 4. **Data Augmentation Technique**: Increases the number of training samples to avoid overfitting. 5. **Analyzing Performance Under Different Conditions**: The study finds that EEG signals under specific conditions (such as performing both motor and auditory tasks or only auditory tasks) can provide more distinguishing features. 6. **Improving Detection Performance**: Enhances the accuracy of SZ detection using a small number of left frontal lobe channels. ### Experimental Results - Evaluated on the benchmark public Kaggle dataset, achieving 98.6% accuracy, 98.65% sensitivity, 98.72% specificity, 98.72% precision, and 98.65% F1 score. - The method outperforms existing methods under all conditions. In summary, this paper aims to develop an efficient and reliable deep learning model to assist psychiatrists in the early diagnosis of schizophrenia and provide timely treatment plans.