STEADYNet: Spatiotemporal EEG analysis for dementia detection using convolutional neural network

Cogn Neurodyn. 2024 Oct;18(5):3195-3208. doi: 10.1007/s11571-024-10153-6. Epub 2024 Jul 19.

Abstract

Dementia is a neuro-degenerative disorder with a high death rate, mainly due to high human error, time, and cost of the current clinical diagnostic techniques. The existing dementia detection methods using hand-crafted electroencephalogram (EEG) signal features are unreliable. A convolution neural network using spatiotemporal EEG signals (STEADYNet) is presented to improve the dementia detection. The STEADYNet uses a multichannel temporal EEG signal as input. The network is grouped into feature extraction and classification components. The feature extraction comprises two convolution layers to generate complex features, a max-pooling layer to reduce the EEG signal's spatiotemporal redundancy, and a dropout layer to improve the network's generalization. The classification processes the feature extraction output nonlinearly using two fully-connected layers to generate salient features and a softmax layer to generate disease probabilities. Two publicly available multiclass datasets of dementia are used for evaluation. The STEADYNet outperforms existing automatic dementia detection methods with accuracies of 99.29 % , 99.65 % , and 92.25 % for Alzheimer's disease, mild cognitive impairment, and frontotemporal dementia, respectively. The STEADYNet has a low inference time and floating point operations, suitable for real-time applications. It may aid neurologists in efficient detection and treatment. A Python implementation of the STEADYNet is available at https://github.com/SandeepSangle12/STEADYNet.git.

Keywords: Alzheimer’s disease; Convolution neural network; Electroencephalogram; Frontotemporal dementia.