Optimized EEG-Based Depression Detection and Severity Staging Using GAN-Augmented Neuro-Fuzzy and Deep Learning Models

Sudhir Dhekane; Anand Khandare

doi:10.35882/jeeemi.v7i4.1107

Sudhir Dhekane Thakur College of Engineering and Technology, Mumbai, India; D.J. Sanghvi College of Engineering, Mumbai, India https://orcid.org/0000-0002-9822-1826
Anand Khandare Thakur College of Engineering and Technology, Mumbai, India https://orcid.org/0000-0002-7732-8464

DOI: https://doi.org/10.35882/jeeemi.v7i4.1107

Keywords: Depression Severity Classification; ANFIS; SHAP Interpretability; Data Augmentation

Abstract

Detecting depression and identifying its severity remain challenging tasks, especially in diverse environments where fair and reliable outcomes are expected. This study aims to address this problem with advanced machine learning models to achieve high accuracy and explainability; making the approach suitable for the real world depression screening and stage evaluation by implementing EEG-based depression detection and staging. We established the parameters of development of EEG-based depression detection in optimization of channel selection together with machine-learning models. Extreme channel selection was performed during this study with Recursive Feature Elimination (RFE) whereby major 11 channels identified, and the MLP classifier achieved 98.7% accuracy supported by AI explainability, thus outpacing the XGBoost and LGBM by 5.2 to 8.2% across multiple datasets (n=184 to 382) and greatly endorsed incredible generalization (precision=1.000, recall=0.966). This makes MLP a trustworthy BCI tool for real-world implementation of depression screening. We also examined assigning depression stages (Mild/Moderate/Severe) on EEG data with models supported or not with GAN-based augmentation (198 to 5,000 samples). CNNs did well on Moderate-stage classification, while ANFIS kept a firm accuracy of 98.34% at perfect metric consistency (precision/recall=0.98) with AI explainability. GAN augmentation improved the classifications of severe cases by 15%, indicating a good marriage of neuro-fuzzy systems and synthetic data for the precise stage determination. This is an important contribution to BCI research since it offers a data-efficient and scalable framework for EEG based depression diagnosis and severity evaluation, thus contributing to the bridge between competitive modeling and clinical applicability. This work, therefore, lays down a pathway for the design of accessible and automated depression screening aids in both high-resource and low-resource settings

Downloads

Download data is not yet available.

Author Biography

Anand Khandare, Thakur College of Engineering and Technology, Mumbai, India

DR. ANAND KHANDARE has received M.E. degree in Computer Engineering and Ph.D in CSE. He is currently working as Professor & Associate Dean (Planning & Operations-Digital Resources) at Thakur College of Engineering & Technology, Mumbai. His current research interest includes Machine Learning and Deep Learning, Database management, natural language Processing and Data Warehouse and Brain Computer Interface. He has received Got Silver category award for Infosys Campus Connect Program. He has been felicitated by Bhaktivedanta Hospital for successfully launching Application. He has earned funding for many projects and successfully completed the same. He has published number of papers in the international journal and Conferences as a part of professional development. Further he has reviewed journals and articles for various conference

References

S. A. Fields, J. Schueler, K. M. Arthur, and B. Harris, “The Role of Impulsivity in Major Depression: A Systematic Review”, doi: 10.1007/s40473-021-00231-y/Published.

Z. Li, M. Ruan, J. Chen, and Y. Fang, “Major Depressive Disorder: Advances in Neuroscience Research and Translational Applications,” Jun. 01, 2021, Springer. doi: 10.1007/s12264-021-00638-3.

K. Kroenke, R. L. Spitzer, and J. B. W. Williams, “The PHQ-9 Validity of a Brief Depression Severity Measure.”

P. Riera-Serra et al., “Clinical predictors of suicidal ideation, suicide attempts and suicide death in depressive disorder: a systematic review and meta-analysis,” Eur Arch Psychiatry Clin Neurosci, Oct. 2023, doi: 10.1007/s00406-023-01716-5.

P. W. Corrigan, B. G. Druss, and D. A. Perlick, “The impact of mental illness stigma on seeking and participating in mental health care,” Psychological Science in the Public Interest, Supplement, vol. 15, no. 2, pp. 37–70, Jan. 2014, doi: 10.1177/1529100614531398.

S. Olbrich and M. Arns, “EEG biomarkers in major depressive disorder: Discriminative power and prediction of treatment response,” International Review of Psychiatry, vol. 25, no. 5, pp. 604–618, 2013, doi: 10.3109/09540261.2013.816269.

B. Ay et al., “Automated Depression Detection Using Deep Representation and Sequence Learning with EEG Signals,” J Med Syst, vol. 43, no. 7, Jul. 2019, doi: 10.1007/s10916-019-1345-y.

G. M. Bairy, U. C. Niranjan, and S. D. Puthankattil, “AUTOMATED CLASSIFICATION OF DEPRESSION EEG SIGNALS USING WAVELET ENTROPIES AND ENERGIES,” J Mech Med Biol, vol. 16, no. 3, May 2016, doi: 10.1142/S0219519416500354.

2019 27th Iranian Conference on Electrical Engineering (ICEE). IEEE, 2019.

S. Khan et al., “A machine learning based depression screening framework using temporal domain features of the electroencephalography signals,” PLoS One, vol. 19, no. 3 MARCH, Mar. 2024, doi: 10.1371/journal.pone.0299127.

TENCON 2019 - 2019 IEEE Region 10 Conference (TENCON). IEEE, 2019.

S. Soni, A. Seal, A. Yazidi, and O. Krejcar, “Graphical representation learning-based approach for automatic classification of electroencephalogram signals in depression,” Comput Biol Med, vol. 145, p. 105420, 2022, doi: https://doi.org/10.1016/j.compbiomed.2022.105420.

A. Sam, R. Boostani, S. Hashempour, M. Taghavi, and S. Sanei, “Depression Identification Using EEG Signals via a Hybrid of LSTM and Spiking Neural Networks,” IEEE Transactions on Neural Systems and Rehabilitation Engineering, vol. 31, pp. 4725–4737, 2023, doi: 10.1109/TNSRE.2023.3336467.

A. Seal, R. Bajpai, J. Agnihotri, A. Yazidi, E. Herrera-Viedma, and O. Krejcar, “DeprNet: A Deep Convolution Neural Network Framework for Detecting Depression Using EEG,” IEEE Trans Instrum Meas, vol. 70, 2021, doi: 10.1109/TIM.2021.3053999.

Y. L. Neo, “A Review of State-of-the-Art Machine Algorithms on Classifying Depressive Disorder EEG Signals,” in 2024 6th International Conference on Artificial Intelligence and Computer Applications (ICAICA), 2024, pp. 365–371. doi: 10.1109/ICAICA63239.2024.10822999.

D. Jan, M. de Vega, J. López-Pigüi, and I. Padrón, “Applying Deep Learning on a Few EEG Electrodes during Resting State Reveals Depressive States: A Data Driven Study,” Brain Sci, vol. 12, no. 11, Nov. 2022, doi: 10.3390/brainsci12111506.

U. R. Acharya, S. L. Oh, Y. Hagiwara, J. H. Tan, H. Adeli, and D. P. Subha, “Automated EEG-based screening of depression using deep convolutional neural network,” Comput Methods Programs Biomed, vol. 161, pp. 103–113, Jul. 2018, doi: 10.1016/j.cmpb.2018.04.012.

D. Dhabliya, K. S. Bhuvaneshwari, P. Kapoor, S. Sowmiya, A. Garg, and L. Yashoda, “Hybrid ANFIS Model For Depression Recognition Consuming Three-Channel EEG Data,” in 2024 International Conference on Recent Advances in Science and Engineering Technology (ICRASET), 2024, pp. 1–5. doi: 10.1109/ICRASET63057.2024.10895792.

U. R. Acharya et al., “A novel depression diagnosis index using nonlinear features in EEG signals,” Eur Neurol, vol. 74, no. 1–2, pp. 79–83, Dec. 2015, doi: 10.1159/000438457.

R. A. Movahed, G. P. Jahromi, S. Shahyad, and G. H. Meftahi, “A major depressive disorder classification framework based on EEG signals using statistical, spectral, wavelet, functional connectivity, and nonlinear analysis,” J Neurosci Methods, vol. 358, Jul. 2021, doi: 10.1016/j.jneumeth.2021.109209.

S. Dhekane and A. Khandare, “Applied Machine Learning in EEG data Classification to Classify Major Depressive Disorder by Critical Channels,” Journal of Electronics, Electromedical Engineering, and Medical Informatics, vol. 7, no. 3, pp. 581–596, Jul. 2025, doi: 10.35882/jeeemi.v7i3.719.

J. Shen, J. Chen, Y. Ma, Z. Cao, Y. Zhang, and B. Hu, “Explainable Depression Recognition from EEG Signals via Graph Convolutional Network,” in 2023 IEEE International Conference on Bioinformatics and Biomedicine (BIBM), 2023, pp. 1406–1412. doi: 10.1109/BIBM58861.2023.10386011.

L. Kumar, K. Kaustubh, T. Rajkhowa, and S. A. Mattur, “EEG-ViLSTM: A Deep Learning Approach for Depression Detection Using EEG Signals,” in 2025 National Conference on Communications (NCC), 2025, pp. 1–6. doi: 10.1109/NCC63735.2025.10983346.

W. Yuan et al., “Discovery of Shared Latent Nonlinear Effective Connectivity for EEG-Based Depression Detection,” IEEE Trans Neural Netw Learn Syst, vol. 36, no. 6, pp. 10663–10677, 2025, doi: 10.1109/TNNLS.2024.3514182.

A. Saeedi, M. Saeedi, A. Maghsoudi, and A. Shalbaf, “Major depressive disorder diagnosis based on effective connectivity in EEG signals: a convolutional neural network and long short-term memory approach,” Cogn. Neurodyn., vol. 15, no. 2, pp. 239–252, Apr. 2021.

S. Mahato, S. Paul, N. Goyal, S. N. Mohanty, and S. Jain, “3EDANFIS: Three channel EEG-based depression detection technique with Hybrid Adaptive Neuro Fuzzy Inference System,” Recent Pat. Eng., vol. 17, no. 6, Nov. 2023.

M. Rehman, S. M. Umar Saeed, S. Khan, S. H. Noorani, and U. Rauf, “EEG-Based Depression Detection: A Temporal Domain Feature-Centric Machine Learning Approach,” in 2023 International Conference on Frontiers of Information Technology (FIT), 2023, pp. 208–213. doi: 10.1109/FIT60620.2023.00046.

S. M. Park et al., “Identification of Major Psychiatric Disorders From Resting-State Electroencephalography Using a Machine Learning Approach,” Front Psychiatry, vol. 12, Aug. 2021, doi: 10.3389/fpsyt.2021.707581.

Zhang, Z., Yang, J., Xiong, P., Hao, H., Zhang, J., Li, L., Wang, C., & Liu, X. (2025). A cross-subject MDD detection approach based on multiscale nonlinear analysis in resting state EEG. Neuroscience, 582, 1–10.

Tigga, N. P., & Garg, S. (2022). Efficacy of novel attention-based gated recurrent units transformer for depression detection using electroencephalogram signals. Health information science and systems, 11(1), 1. https://doi.org/10.1007/s13755-022-00205-8