Non-Invasive Bio-Signal Data Classification Of Psychiatric Mood Disorders Using Modified CNN and VGG16
Abstract
In this study, the aim is to develop an ensemble machine learning (ML) based deep learning (DL) model classifiers to detect and compare one type of major psychiatric disorders of mood disorders (Depressive and Bipolar disorders) using Electroencephalography (EEG). The diverse and multiple non-invasive biosignals were collected retrospectively according to the granted ethical permission. The experimental part is consisted from three main parts. First part is the data collection&development, the second part is data transformation and augmentation via Spectrogram image conversion process and online Keras data augmentation part, respectively. The third and final part is to fed these image dataset into modified Convolutional Neural Network (CNN) and VGG16 models for training and testing parts to detect, compare and discriminate mood disorders types in detail with a specific healthy group. As the performance evaluation background of the mood disorder classification models, confusion matrices and receiver operating characteristics (ROC) curves were used and finally, the accuracy achieved by CNN model was 88% and VGG16 model was %90, which is an improvement of 10% compared to the previous studies in literature. Therefore, our system can help clinicians and researchers to manage, diagnose and prognosis of the mental health of people.
Keywords
Psychiatric disorders, Mood disorder, Depressive disorder, Bipolar Disorder, Deep Learning, Pretrained model classification
References
- Acharya, UR, Oh, SL, Hagiwara, Y, Tan, JH, Adeli, H, & Subha, DP, (2018), Automated EEG- based screening of depression using deep convolutional neural network, Computer Methods and Programs in Biomedicine, 161, 103–113. https://doi. org/10.1016/j.cmpb.2018.04.012
- Aristizabal, A, Fernando, D, Denman, T, Robinson, S, Sridharan, JE, Johnston, S, Fookes, C., (2021), Identification of children at risk of schizophrenia via deep learning and EEG responses, IEEE Journal of Biomedical and Health Informatics, 25(1), 69–76. https://doi.org/10.1109/JBHI.2020.2984238
- B˘alan, O, Moise, G, Moldoveanu, A, Leordeanu, M, & Moldoveanu, F, (2019), Fear level classification based on emotional dimensions and machine learning techniques, Sensors (Basel, Switzerland), 19(7). https://doi.org/10.3390/s19071738
- B˘alan, O, Moise, G, Moldoveanu, A, Leordeanu, M, & Moldoveanu, F, (2020), An investigation of various machine and deep learning techniques applied in automatic fear level detection and acrophobia virtual therapy, Sensors (Basel, Switzerland), 20 (2). https://doi.org/10.3390/s20020496
- Biship, CM, (2007), Pattern Recognition and Machine Learning (Information Science and Statistics) (Springer-Verlag, Berlin).
- Boudouh, SS, and Bouakkaz, M, (2022), Breast Cancer: Using Deep Transfer Learning Techniques AlexNet Convolutional Neural Network For Breast Tumor Detection in Mammography Images, 2022 7th International Conference on Image and Signal Processing and their Applications (ISPA), pp. 1-7, doi: 10.1109/ISPA54004.2022.9786351.
- Dubreuil-Vall, L, Ruffini, G, & Camprodon, JA, (2020), Deep learning convolutional neural networks discriminate adult ADHD from healthy individuals on the basis of event-related spectral EEG, Frontiers in Neuroscience, 14, 251. https://doi.org/ 10.3389/fnins.2020.00251
- Garcia, CI, Grasso, F, Luchetta, A, Piccirilli, MC, Paolucci, L, and Talluri, G, (2020), A comparison of power quality disturbance detection and classification methods using CNN, LSTM and CNN-LSTM, Applied Sciences, vol. 10, no. 19, pp. 6755–6757.
- Giannakakis, G, Grigoriadis, D, Giannakaki, K, Simantiraki, O, Roniotis, A, and Tsiknakis, M, (2019), Review on psychological stress detection using biosignals, IEEE Transactions on Affective Computing, vol. 2019, Article ID 2927337, 1 page.
- Gisele, H, Barboni, M and Joaquim, CF (2015), Computer-aided diagnosis system based on fuzzy logic for breast cancer categorization, Computers in biology and medicine, 64:334–346.