An Effective and Interpretable EEG-Based Depression Recognition Method Using Hybrid Feature Selection
Abstract
1. Introduction
- A novel two-stage feature selection strategy combining RankSearch with Genetic Algorithm (GA), which dynamically adjusts feature weights and GA’s crossover/mutation probabilities to optimize feature subsets and enhance model performance.
- Focused prefrontal electrode optimization analysis employing multi-domain feature extraction, with comprehensive evaluation of their effectiveness in depression identification.
- Systematic comparison of various machine learning classifiers with different feature selection methods, providing new perspectives and methodologies for early depression diagnosis.
2. Materials and Methods
2.1. Dataset
2.2. Preprocessing
2.3. Feature Extraction
2.4. Feature Selection
2.5. Classification Model
2.6. Evaluation Methods
3. Results
3.1. Classification Results
3.2. Feature Analysis
3.3. Comparison with Other Feature Selection Methods
4. Discussion
4.1. Neurophysiological Basis of Key Features
4.2. Comparison with Alternative Methods
4.3. Limitations of the Current Study
5. Conclusions
Author Contributions
Funding
Institutional Review Board Statement
Informed Consent Statement
Data Availability Statement
Acknowledgments
Conflicts of Interest
References
- Karunarathne, A.; Gunnell, D.; Konradsen, F.; Eddleston, M. How many premature deaths from pesticide suicide have occurred since the agricultural Green Revolution? Clin. Toxicol. 2020, 58, 227–232. [Google Scholar]
- Li, G.; Fife, D.; Wang, G.; Sheehan, J.J.; Bodén, R.; Brandt, L.; Brenner, P.; Reutfors, J.; DiBernardo, A. All-cause mortality in patients with treatment-resistant depression: A cohort study in the US population. Ann. Gen. Psychiatry 2019, 18, 23. [Google Scholar] [CrossRef] [PubMed]
- Liu, Y.; Pu, C.; Xia, S.; Deng, D.; Wang, X.; Li, M. Machine learning approaches for diagnosing depression using EEG: A review. Transl. Neurosci. 2022, 13, 224–235. [Google Scholar] [CrossRef]
- Akbari, H.; Sadiq, M.T.; Rehman, A.U.; Ghazvini, M.; Naqvi, R.A.; Payan, M.; Bagheri, H.; Bagheri, H. Depression recognition based on the reconstruction of phase space of EEG signals and geometrical features. Appl. Acoust. 2021, 179, 108078. [Google Scholar] [CrossRef]
- Sun, S.; Liu, L.; Shao, X.; Yan, C.; Li, X.; Hu, B. Abnormal brain topological structure of mild depression during visual search processing based on EEG signals. IEEE Trans. Neural Syst. Rehabil. Eng. 2022, 30, 1705–1715. [Google Scholar] [PubMed]
- Pizzagalli, D.A.; Sherwood, R.J.; Henriques, J.B.; Davidson, R.J. Frontal brain asymmetry and reward responsiveness: A source-localization study. Psychol. Sci. 2005, 16, 805–813. [Google Scholar]
- Liu, S.; Liu, X.; Yan, D.; Chen, S.; Liu, Y.; Hao, X.; Ou, W.; Huang, Z.; Su, F.; He, F.; et al. Alterations in patients with first-episode depression in the eyes-open and eyes-closed conditions: A resting-state EEG study. IEEE Trans. Neural Syst. Rehabil. Eng. 2022, 30, 1019–1029. [Google Scholar]
- Ying, M.; Zhu, J.; Li, X.; Hu, B. A functional connectivity-based model with a lightweight attention mechanism for depression recognition using EEG signals. IEEE Trans. Neural Syst. Rehabil. Eng. 2024. early access. [Google Scholar] [CrossRef]
- Sam, A.; Boostani, R.; Hashempour, S.; Taghavi, M.; Sanei, S. Depression identification using EEG signals via a hybrid of LSTM and spiking neural networks. IEEE Trans. Neural Syst. Rehabil. Eng. 2023, 31, 4725–4737. [Google Scholar]
- Shen, J.; Zhang, X.; Huang, X.; Wu, M.; Gao, J.; Lu, D.; Ding, Z.; Hu, B. An optimal channel selection for EEG-based depression detection via kernel-target alignment. IEEE J. Biomed. Health Inform. 2020, 25, 2545–2556. [Google Scholar]
- Cai, H.; Chen, Y.; Han, J.; Zhang, X.; Hu, B. Study on feature selection methods for depression detection using three-electrode EEG data. Interdiscip. Sci. Comput. Life Sci. 2018, 10, 558–565. [Google Scholar] [CrossRef]
- Zhu, L.; Wang, R.; Jin, X.; Li, Y.; Tian, F.; Cai, R.; Qian, K.; Hu, X.; Hu, B.; Yamamoto, Y.; et al. Explainable depression classification based on EEG feature selection from audio stimuli. IEEE Trans. Neural Syst. Rehabil. Eng. 2025. early access. [Google Scholar] [CrossRef]
- Erguzel, T.T.; Tas, C.; Cebi, M. A wrapper-based approach for feature selection and classification of major depressive disorder-bipolar disorders. Comput. Biol. Med. 2015, 64, 127–137. [Google Scholar]
- Mumtaz, W.; Xia, L.; Mohd Yasin, M.A.M.; Ali, S.S.A.; Malik, A.S. A wavelet-based technique to predict treatment outcome for major depressive disorder. PLoS ONE 2017, 12, e0171409. [Google Scholar] [CrossRef] [PubMed]
- Jung, T.P.; Makeig, S.; Humphries, C.; Lee, T.W.; McKeown, M.J.; Iragui, V.; Sejnowski, T.J. Removing electroencephalographic artifacts by blind source separation. Psychophysiology 2000, 37, 163–178. [Google Scholar] [CrossRef] [PubMed]
- Zikov, T.; Bibian, S.; Dumont, G.A.; Huzmezan, M.; Ries, C.R. A wavelet based de-noising technique for ocular artifact correction of the electroencephalogram. In Proceedings of the Second Joint 24th Annual Conference and the Annual Fall Meeting of the Biomedical Engineering Society (EMBS/BMES), Houston, TX, USA, 23–26 October 2002; Volume 1, pp. 98–105. [Google Scholar]
- Pamungkas, Y.; Wibawa, A.D.; Rais, Y. Classification of emotions (positive-negative) based on EEG statistical features using RNN, LSTM, and Bi-LSTM algorithms. In Proceedings of the 2nd International Seminar on Machine Learning, Optimization, and Data Science (ISMODE), Jakarta, Indonesia, 27–28 January 2022; pp. 275–280. [Google Scholar]
- Bachmann, M.; Paeske, L.; Kalev, K.; Aarma, K.; Lehtmets, A.; Oopik, P.; Lass, J.; Hinrikus, H. Methods for classifying depression in single channel EEG using linear and nonlinear signal analysis. Comput. Methods Programs Biomed. 2018, 155, 11–17. [Google Scholar] [CrossRef]
- Mahato, S.; Paul, S. Detection of major depressive disorder using linear and non-linear features from EEG signals. Microsyst. Technol. 2019, 25, 1065–1076. [Google Scholar]
- Zhu, J.; Yang, C.; Xie, X.; Wei, S.; Li, Y.; Li, X.; Hu, B. Mutual information based fusion model (MIBFM): Mild depression recognition using EEG and pupil area signals. IEEE Trans. Affect. Comput. 2023, 14, 2102–2115. [Google Scholar] [CrossRef]
- Xue, B.; Zhang, M.; Browne, W.N.; Yao, X. A survey on evolutionary computation approaches to feature selection. IEEE Trans. Evol. Comput. 2016, 20, 606–626. [Google Scholar]
- Surma, R.; Wojcieszyńska, D.; Mulla, S.I.; Guzik, U. Current strategy of non-model-based bioprocess optimizations with genetic algorithms in bioscience—A systematic review. Comput. Biol. Med. 2025, 192, 110247. [Google Scholar]
- Shon, D.; Im, K.; Park, J.H.; Lim, D.S.; Jang, B.; Kim, J.M. Emotional stress state detection using genetic algorithm-based feature selection on EEG signals. Int. J. Environ. Res. Public Health 2018, 15, 2461. [Google Scholar] [CrossRef]
- Avots, E.; Jermakovs, K.; Bachmann, M.; Päeske, L.; Ozcinar, C.; Anbarjafari, G. Ensemble approach for detection of depression using EEG features. Entropy 2022, 24, 211. [Google Scholar] [CrossRef]
- Adnan, M.; Alarood, A.A.S.; Uddin, M.I.; Rehman, I.u. Utilizing grid search cross-validation with adaptive boosting for augmenting performance of machine learning models. PeerJ Comput. Sci. 2022, 8, e803. [Google Scholar] [CrossRef]
- Lebiecka, K.; Zuchowicz, U.; Wozniak-Kwasniewska, A.; Szekely, D.; Olejarczyk, E.; David, O. Complexity analysis of EEG data in persons with depression subjected to transcranial magnetic stimulation. Front. Physiol. 2018, 9, 1385. [Google Scholar] [CrossRef] [PubMed]
- Mohammadi, Y.; Moradi, M.H. Prediction of depression severity scores based on functional connectivity and complexity of the EEG signal. Clin. EEG Neurosci. 2021, 52, 1550059420965431. [Google Scholar] [CrossRef] [PubMed]
- Ji, S.; An, W.; Zhang, J.; Zhou, C.; Liu, C.; Yu, H. The different impacts of functional network centrality and connectivity on the complexity of brain signals in healthy control and first-episode drug-naïve patients with major depressive disorder. Brain Imaging Behav. 2024, 19, 111–123. [Google Scholar] [CrossRef] [PubMed]
- Xu, X.; Xu, J.; Du, R.; Xu, T. Research on depression recognition model and its temporal characteristics based on multiscale entropy of EEG signals. Entropy 2025, 27, 142. [Google Scholar] [CrossRef]
- Kang, X.; Liu, X.; Chen, S.; Zhang, W.; Liu, S.; Ming, D. Major depressive disorder recognition by quantifying EEG signal complexity using proposed APLZC and AWPLZC. J. Affect. Disord. 2024, 356, 105–114. [Google Scholar] [CrossRef]
- Chen, H.; Yang, J.; Zhang, B.; Zhang, L.; Wang, J.; Zhang, H.; Zhang, H.; Yu, C.; Zhang, J.; Xu, Z. Quantitative analysis of the resting-state EEG power spectrum in patients with epilepsy comorbid with anxiety and depression. Acta Epileptol. 2025, 7, 22. [Google Scholar] [CrossRef]
- Hassan, M.; Kaabouch, N. Impact of feature selection techniques on the performance of machine learning models for depression detection using EEG data. Appl. Sci. 2024, 14, 10532. [Google Scholar] [CrossRef]
- Bhadra, S.; Kumar, C.J. Enhancing the efficacy of depression detection system using optimal feature selection from EHR. Comput. Methods Biomech. Biomed. Eng. 2024, 27, 222–236. [Google Scholar] [CrossRef] [PubMed]
- Li, X.; Hu, B.; Sun, S.; Cai, H. EEG-based mild depressive detection using feature selection methods and classifiers. Comput. Methods Programs Biomed. 2016, 136, 151–161. [Google Scholar] [CrossRef] [PubMed]








| Function | Features |
|---|---|
| Time-domain | |
| Standard deviation, Peak-to-peak amplitude, Root mean square, Hjorth parameters (Activity, Mobility, Complexity) | 6 |
| Frequency domain | |
| Band Power (Delta, Theta, Alpha, Beta, Gamma), Mean Power, Power Spectrum, Peak Frequency, Spectral Asymmetry Index, Band Power Ratios (Alpha/Beta, Theta/Alpha, Theta/Beta) | 12 |
| Entropy and Complexity | |
| Differential Entropy, Sample Entropy, Permutation Entropy, Spectral Entropy, Wavelet Entropy, Fuzzy Entropy, Singular Value Decomposition Entropy, Lempel–Ziv Complexity, Higuchi Fractal Dimension, C0 Complexity, Correlation Dimension, Largest Lyapunov Exponent, Lyapunov Exponent Spectrum | 13 |
| Parameters | Value |
|---|---|
| Fitness Function | 5-fold CV accuracy using Random Forest |
| Elitism Strategy | Hall of Fame (keep best 1) |
| Random Seed Strategy | Dynamically set in each run |
| Crossover Probability | 0.8 |
| Gene Mutation Probability | 0.05 |
| Population Size | 50 |
| Number of Generations | 20 |
| Model | Acc | Sen | Spec | F1 | AUC | |
|---|---|---|---|---|---|---|
| DT | 0.9237 | 0.9270 | 0.9209 | 0.9300 | 0.8469 | 0.9239 |
| KNN | 0.9458 | 0.9661 | 0.9254 | 0.9457 | 0.8915 | 0.9458 |
| RF | 0.9441 | 0.9599 | 0.9304 | 0.9500 | 0.8879 | 0.9451 |
| SVM | 0.9458 | 0.9672 | 0.9272 | 0.9500 | 0.8913 | 0.9472 |
| XGBoost | 0.9508 | 0.9599 | 0.9430 | 0.9500 | 0.9014 | 0.9514 |
| Method | Average | |||||
|---|---|---|---|---|---|---|
| Acc | Sen | Spec | F1 | κ | AUC | |
| 93 features | 0.9546 | 0.9618 | 0.9483 | 0.9576 | 0.9089 | 0.9550 |
| 73 features | 0.9580 | 0.9669 | 0.9502 | 0.9579 | 0.9157 | 0.9586 |
| 30 features | 0.9420 | 0.9560 | 0.9294 | 0.9451 | 0.8838 | 0.9427 |
| 10 features | 0.8532 | 0.8505 | 0.8560 | 0.8609 | 0.7057 | 0.8533 |
| Methods | Acc | Sen | Spec | F1 | AUC | |
|---|---|---|---|---|---|---|
| ACO | 0.8665 | 0.8681 | 0.8647 | 0.8666 | 0.7329 | 0.8664 |
| LASSO | 0.5211 | 0.4491 | 0.5929 | 0.5480 | 0.0419 | 0.5210 |
| PCA | 0.8732 | 0.8720 | 0.8566 | 0.8517 | 0.7445 | 0.8533 |
| This Work | 0.9420 | 0.9560 | 0.9294 | 0.9451 | 0.8838 | 0.9427 |
| Authors | FS_Model | Classifier | Acc |
|---|---|---|---|
| Erguzel [13] | IACO | SVM | 80.19% |
| Hassan [32] | EN, MI, test, FFS-SGD, SVM-RFE, mRMR | LDA, SVM, RF, GBDT | 93.54% |
| Bhadra [33] | RFE, MI, PSO, GA, FA | SVM, LR, DT, RF, GB | 88.46% |
| Li [34] | BF, GSW, LFS, RS | BN, SVM, RF, LR, KNN | 92.00% |
| Cai [11] | WSE, CAE, PCA, GRAE | SVM, KNN, DT, LR, RF | 76.40% |
| This Work | RS+GA | DT, KNN, RF, SVM, XGBoost | 95.08% |
Disclaimer/Publisher’s Note: The statements, opinions and data contained in all publications are solely those of the individual author(s) and contributor(s) and not of MDPI and/or the editor(s). MDPI and/or the editor(s) disclaim responsibility for any injury to people or property resulting from any ideas, methods, instructions or products referred to in the content. |
© 2026 by the authors. Licensee MDPI, Basel, Switzerland. This article is an open access article distributed under the terms and conditions of the Creative Commons Attribution (CC BY) license.
Share and Cite
Xu, X.; Fan, Q.; Ju, S.; Du, R. An Effective and Interpretable EEG-Based Depression Recognition Method Using Hybrid Feature Selection. Bioengineering 2026, 13, 410. https://doi.org/10.3390/bioengineering13040410
Xu X, Fan Q, Ju S, Du R. An Effective and Interpretable EEG-Based Depression Recognition Method Using Hybrid Feature Selection. Bioengineering. 2026; 13(4):410. https://doi.org/10.3390/bioengineering13040410
Chicago/Turabian StyleXu, Xin, Qiuyun Fan, Shanjing Ju, and Ruoyu Du. 2026. "An Effective and Interpretable EEG-Based Depression Recognition Method Using Hybrid Feature Selection" Bioengineering 13, no. 4: 410. https://doi.org/10.3390/bioengineering13040410
APA StyleXu, X., Fan, Q., Ju, S., & Du, R. (2026). An Effective and Interpretable EEG-Based Depression Recognition Method Using Hybrid Feature Selection. Bioengineering, 13(4), 410. https://doi.org/10.3390/bioengineering13040410

