| Literature DB >> 35205506 |
Egils Avots1, Klāvs Jermakovs1, Maie Bachmann2, Laura Päeske2, Cagri Ozcinar1, Gholamreza Anbarjafari1,3,4.
Abstract
Depression is a public health issue that severely affects one's well being and can cause negative social and economic effects to society. To raise awareness of these problems, this research aims at determining whether the long-lasting effects of depression can be determined from electroencephalographic (EEG) signals. The article contains an accuracy comparison for SVM, LDA, NB, kNN, and D3 binary classifiers, which were trained using linear (relative band power, alpha power variability, spectral asymmetry index) and nonlinear (Higuchi fractal dimension, Lempel-Ziv complexity, detrended fluctuation analysis) EEG features. The age- and gender-matched dataset consisted of 10 healthy subjects and 10 subjects diagnosed with depression at some point in their lifetime. Most of the proposed feature selection and classifier combinations achieved accuracy in the range of 80% to 95%, and all the models were evaluated using a 10-fold cross-validation. The results showed that the motioned EEG features used in classifying ongoing depression also work for classifying the long-lasting effects of depression.Entities:
Keywords: depression; electroencephalogram (EEG); ensemble learning; feature extraction and selection; machine learning
Year: 2022 PMID: 35205506 PMCID: PMC8871180 DOI: 10.3390/e24020211
Source DB: PubMed Journal: Entropy (Basel) ISSN: 1099-4300 Impact factor: 2.524
Figure 1International 10–20 system for EEG recording.
Information about subjects in the dataset.
| EST-Q | |||||||
|---|---|---|---|---|---|---|---|
| Sex | Age | HAM-D | DEP | AUR | PAF | SAR | AST |
| M | 24/25 | 8/7 | 13/5 | 0/7 | 0/0 | 2/0 | 5/7 |
| F | 34/33 | 9/1 | 5/1 | 12/3 | 0/0 | 0/0 | 5/2 |
| F | 35/35 | 21/6 | 16/7 | 11/8 | 0/2 | 2/1 | 10/4 |
| M | 35/36 | 4/1 | 4/4 | 3/8 | 1/0 | 0/1 | 0/3 |
| F | 37/35 | 13/3 | 8/4 | 11/6 | 0/0 | 0/1 | 13/2 |
| F | 38/39 | 9/0 | 4/3 | 17/7 | 2/1 | 0/0 | 11/3 |
| M | 43/40 | 6/1 | 8/3 | 9/2 | 0/0 | 1/0 | 7/2 |
| F | 46/46 | 12/5 | 8/6 | 2/3 | 0/0 | 1/0 | 3/1 |
| F | 48/48 | 8/0 | 4/2 | 7/8 | 0/1 | 2/0 | 6/5 |
| M | 53/60 | 3/7 | 11/5 | 8/8 | 0/0 | 0/1 | 8/9 |
Subject pairs: depressed/healthy.
Figure 2Weight W calculation for an ensemble classifier using training data. Each block contains one healthy and depressed subject. TP—true positive, TN—true negative, FP—false positive, and FN—false negative.
Baseline classifier accuracy for all feature types.
| Classifier Accuracy (%) | |||||
|---|---|---|---|---|---|
| Feature | RBF | LDA | Naive | kNN | D3 |
| Trbp | 54.40 | 65.00 | 73.30 | 44.45 | 38.65 |
| Arbp | 50.00 | 64.15 | 70.95 | 58.70 | 66.55 |
| Brbp | 70.00 | 52.90 | 65.05 | 62.20 | 79.85 |
| Grbp | 38.45 | 52.95 | 59.40 | 50.90 | 54.85 |
| APV | 35.40 | 27.05 | 31.85 | 37.70 | 64.65 |
| SASI | 54.55 | 55.00 | 54.60 | 59.10 | 53.15 |
| HFD | 55.40 | 41.55 | 51.80 | 71.55 | 82.55 |
| LZC | 80.70 | 57.10 | 58.50 | 75.95 | 63.50 |
| DFA | 68.15 | 75.25 | 63.15 | 70.80 | 74.55 |
Figure 3Electrode importance ranking for Brbp according to ReliefF. Feature set {O2,O1} was chosen as the best option for the Brbp selected (ranked) feature set, as the RMS value was highest at electrode O1.
Selected electrodes based on the F-test and ReliefF. Electrodes are ordered according to their importance score.
| Selected Features | ||
|---|---|---|
| Feature | Univariate Feature | ReliefF |
| Trbp | F4 F8 | O1 PZ P4 |
| Arbp | F7 C3 T5 PZ | O1 O2 |
| Brbp | O1 O2 F4 | O2 O1 |
| Grbp | F7 F3 FZ F4 | FP1 |
| APV | T3 C3 F3 | F3 |
| SASI | FP1 FP2 F7 F3 | FP1 F3 |
| HFD | FP1 FP2 FZ F8 C3 T5 | FP1 O1 |
| LZC | F3 F4 T4 FP1 FP2 | FP1 FZ |
| DFA | O1 O2 FP2 F7 | FP1 FP2 O1 |
Classifier accuracy for features selected by univariate feature ranking using F-tests.
| Classifier Accuracy (%) | |||||
|---|---|---|---|---|---|
| Feature | RBF | LDA | Naive | kNN | D3 |
| Trbp (2) | 55.20 | 65.60 | 69.85 | 55.75 | 45.70 |
| Arbp (8) | 62.35 | 73.85 | 70.95 | 72.65 | 67.15 |
| Brbp (3) | 90.00 | 79.90 | 82.05 | 91.50 | 85.85 |
| Grbp (4) | 70.20 | 52.20 | 71.10 | 54.55 | 60.00 |
| APV (3) | 61.00 | 50.25 | 47.75 | 55.55 | 62.95 |
| SASI (4) | 60.55 | 65.40 | 61.45 | 52.65 | 66.95 |
| HFD (11) | 73.40 | 42.20 | 53.20 | 75.70 | 82.55 |
| LZC (16) | 74.90 | 58.40 | 59.45 | 75.00 | 68.15 |
| DFA (7) | 73.05 | 50.35 | 55.45 | 75.85 | 68.85 |
(*) Number of features used (see Table 3).
Classifier accuracy for features selected by the ReliefF algorithm.
| Classifier Accuracy (%) | |||||
|---|---|---|---|---|---|
| Feature | RBF | LDA | Naive | kNN | D3 |
| Trbp (5) | 66.15 | 79.60 | 80.00 | 72.25 | 55.85 |
| Arbp (2) | 81.20 | 78.70 | 75.95 | 90.00 | 85.30 |
| Brbp (2) | 90.00 | 80.85 | 79.90 | 90.00 | 90.00 |
| Grbp (1) | 75.00 | 69.00 | 75.00 | 70.00 | 63.25 |
| APV (1) | 62.85 | 37.50 | 62.80 | 71.20 | 66.40 |
| SASI (2) | 63.35 | 72.35 | 70.70 | 55.00 | 72.55 |
| HFD (2) | 77.45 | 53.65 | 66.20 | 81.00 | 85.70 |
| LZC (2) | 81.25 | 78.25 | 72.75 | 81.95 | 69.55 |
| DFA (3) | 78.80 | 56.40 | 72.55 | 86.00 | 72.90 |
(*) Number of features used (see Table 3).
Ensemble classifier accuracy.
| Classifier Accuracy (%) | |||||
|---|---|---|---|---|---|
| Features and | RBF | LDA | Naive | kNN | D3 |
| All + Maj. | 70.55 | 48.85 | 61.85 | 68.65 | 79.05 |
| F-test + Maj. | 80.80 | 65.60 | 79.55 | 77.85 | 75.70 |
| ReliefF + Maj. | 88.30 | 80.85 | 93.30 | 88.25 | 88.25 |
| All + Weig. | 65.50 | 51.20 | 63.85 | 69.00 | 73.80 |
| F-test + Weig. | 79.90 | 69.45 | 77.25 | 78.85 | 73.80 |
| ReliefF + Weig. | 84.70 | 75.65 | 83.15 | 88.95 | 85.55 |
| All + Ada. | 70.80 | 53.10 | 71.40 | 63.75 | 69.15 |
| F-test + Ada. | 79.20 | 62.95 | 81.85 | 84.70 | 70.50 |
| ReliefF + Ada. | 81.05 | 72.20 | 78.50 | 86.70 | 79.85 |
Note: F-tests and ReliefF features are from Table 3.
Classifier accuracy for concatenated features.
| Classifier Accuracy (%) | |||||
|---|---|---|---|---|---|
| Features | RBF | LDA | Naive | kNN | D3 |
| All features | 53.25 | 52.35 | 65.20 | 55.50 | 56.70 |
| F-test | 56.70 | 62.10 | 80.75 | 72.55 | 72.90 |
| ReliefF | 70.00 | 71.75 | 76.90 | 80.00 | 65.05 |
| F-test: top | 64.60 | 69.10 | 77.00 | 81.25 | 74.40 |
| ReliefF ** | 89.25 | 80.00 | 81.80 | 95.00 | 95.00 |
** Top ranked: Arbp.O1, Brbp.O2, Arbp.O2.