| Literature DB >> 35027888 |
Hengjin Ke1, Cang Cai2, Fengqin Wang3, Fang Hu4, Jiawei Tang1, Yuxin Shi1.
Abstract
Online end-to-end electroencephalogram (EEG) classification with high performance can assess the brain status of patients with Major Depression Disabled (MDD) and track their development status in time with minimizing the risk of falling into danger and suicide. However, it remains a grand research challenge due to (1) the embedded intensive noises and the intrinsic non-stationarity determined by the evolution of brain states, (2) the lack of effective decoupling of the complex relationship between neural network and brain state during the attack of brain diseases. This study designs a Frequency Channel-based convolutional neural network (CNN), namely FCCNN, to accurately and quickly identify depression, which fuses the brain rhythm to the attention mechanism of the classifier with aiming at focusing the most important parts of data and improving the classification performance. Furthermore, to understand the complexity of the classifier, this study proposes a calculation method of information entropy based on the affinity propagation (AP) clustering partition to measure the complexity of the classifier acting on each channel or brain region. We perform experiments on depression evaluation to identify healthy and MDD. Results report that the proposed solution can identify MDD with an accuracy of 99±0.08%, the sensitivity of 99.07±0.05%, and specificity of 98.90±0.14%. Furthermore, the experiments on the quantitative interpretation of FCCNN illustrate significant differences between the frontal, left, and right temporal lobes of depression patients and the healthy control group.Entities:
Keywords: EEG classification; attention; convolutional neural network (CNN); depression; interpretation
Year: 2021 PMID: 35027888 PMCID: PMC8750060 DOI: 10.3389/fncom.2021.773147
Source DB: PubMed Journal: Front Comput Neurosci ISSN: 1662-5188 Impact factor: 2.380
Figure 1Neural Architecture of FCCNN. “FC” denotes the fully connected layer, “AF” denotes the activation function. The hyper-parameter of convolutional layer is denoted as “filters @ [receptive map size]”.
Brain Region based on 10-20 international electroencephalogram (EEG) system.
|
|
|
|
|---|---|---|
| 1 | Frontal lobe | Fp1, Fp2, F3, F4 |
| 2 | Left temporal | F7, T3, T5 |
| 3 | Central | C3, C4, Fz, Cz, Pz |
| 4 | Right temporal | F8, T4, T6 |
| 5 | Occipital lobe | P3, P4, O1, O2 |
The details of training set and test set. HG denotes the health control group and MG denotes the MDD's group.
|
|
|
|
|
|---|---|---|---|
| HG:24 MG:27 | HG:6898 MG: 7816 | HG:6 MG:7 | HG: 1755 MG: 1973 |
Comparison of different classifiers. The value in brackets represents the SD.
|
|
|
|
|
|
|---|---|---|---|---|
|
|
|
|
| |
| MLRW (Mumtaz et al., | 87.50 | 95 | 80 | - |
| LeNet (Lecun et al., | 93.31 (6.24) | 91.93 (4.27) | 94.85 (1.81) | 2.8 |
| Resnet-16 (He et al., | 82.26 (7.59) | 88.90 (2.14) | 74.79 (3.83) | 80 |
| GoogLeNet (Szegedy et al., | 93.74 (3.65) | 96.48 (1.23) | 90.62 (4.62) | 42 |
| Ours-withoutAttention | 96.04 (3.02) | 97.75 (2.09) | 94.12 (3.58) | 3 |
| Ours | 99 (0.08) | 99.07 (0.05) | 98.90 (0.14) | 3.5 |
Figure 2Accuracy and loss rates in the training and validating processes upon MPHC.
Figure 3ROC Curve on identifying depression state on MPHC.
Figure 4p-value matrix on performance index of accuracy.
Figure 6p-value matrix on performance index of Specificity.
Figure 73D scalp topographies map visualizations from FCCNN with channel (A) and brain region (B).
Figure 8The average frequency-power representations of a different class label (Healthy & MDD) of a typical channel (Fz).
Figure 9The mean entropy values with and without the attention module.
Figure 10Entropy calculation between AP-based clustering partition (A) and traditional methods (B).
Figure 11Comparison of candidate optimization methods.