A Convolutional Neural Network approach for classification of dementia stages based on 2D-spectral representation of EEG recordings

Ieracitano, C.; Mammone, N.; Bramanti, A.; Hussain, A.; Morabito, F.C.

Abstract

A data-driven machine deep learning approach is proposed for differentiating subjects with Alzheimer’s Disease (AD), Mild Cognitive Impairment (MCI) and Healthy Control (HC), by only analyzing noninvasive scalp EEG recordings. The methodology here proposed consists of evaluating the power spectral density (PSD) of the 19-channels EEG traces and representing the related spectral profiles into 2-d gray scale images (PSD-images). A customized Convolutional Neural Network with one processing module of convolution, Rectified Linear Units (ReLu) and pooling layer (CNN1) is designed to extract from PSD-images some suitable features and to perform the corresponding two and three-ways classification tasks. The resulting CNN is shown to provide better classification performance when compared to more conventional learning machines; indeed, it achieves an average accuracy of 89.8% in binary classification and of 83.3% in three-ways classification. These results encourage the use of deep processing systems (here, an engineered first stage, namely the PSD-image extraction, and a second or multiple CNN stage) in challenging clinical frameworks.

Citation

Ieracitano, C., Mammone, N., Bramanti, A., Hussain, A., & Morabito, F. (2019). A Convolutional Neural Network approach for classification of dementia stages based on 2D-spectral representation of EEG recordings. Neurocomputing, 323, 96-107. https://doi.org/10.1016/j.neucom.2018.09.071

• Journal Article Type: Article
• Acceptance Date: Sep 29, 2018
• Online Publication Date: Oct 6, 2018
• Publication Date: Jan 5, 2019
• Deposit Date: Jul 26, 2019
• Journal: Neurocomputing
• Print ISSN: 0925-2312
• Publisher: Elsevier
• Peer Reviewed: Peer Reviewed
• Volume: 323
• Pages: 96-107
• DOI: https://doi.org/10.1016/j.neucom.2018.09.071
• Keywords: Deep learning, Convolutional Neural Network, Power spectral density, Alzheimer’s disease, Mild Cognitive Impairment
• Public URL: http://researchrepository.napier.ac.uk/Output/1792061