Accuracy of neural networks in brain wave diagnosis of schizophrenia

Abstract

This research explores the application of a modified deep learning model for electroencephalography (EEG) signal classification in the context of schizophrenia diagnosis. This study aims to utilize the temporal and spatial characteristics of EEG data to improve classification accuracy. Four popular convolutional neural network (CNN) architectures, namely LeNet-5, AlexNet, VGG16, and ResNet-18, are adapted to handle 1D EEG signals. In addition, a hybrid architecture of CNN-gated recurrent unit (GRU) and CNN-long short-term memory (LSTM) is proposed to capture spatial and temporal dynamics. The model was evaluated on a dataset consisting of EEG recordings from 14 patients with paranoid schizophrenia and 14 healthy controls. The results show high accuracy and F1 scores for all modified models, with CNN-LSTM and CNN-GRU achieving the highest performance with scores of 0.96 and 0.97, respectively. Receiver operating characteristic (ROC) curves demonstrate the model's ability to distinguish between healthy controls and schizophrenia patients. The proposed model offers a promising approach for automated schizophrenia diagnosis based on EEG signals, potentially assisting clinicians in early detection and intervention. Future work will focus on larger data sets and explore transfer learning techniques to improve the generalization ability of the model.