Hilbert-Huang Transform Embedded Self-Attention Neural Network for EEG-Based Major Depressive Disorder vs. Healthy Controls Classification

Junxian Chen^1*Kaikun Tian^2*YU YE³Jiaming Liu²

• ¹Suzhou Industrial Park Institute of Services Outsourcing, Suzhou, China
• ²Hubei Normal University, Huangshi, China
• ³The Central Hospital of Huangshi City, Huangshi, China

This paper proposes a novel approach for distinguishing Major Depressive Disorder (MDD) patients from healthy controls (HC), namely depression screening, using EEG signals, where the Hilbert-Huang Transform (HHT) is integrated into a Self-Attention neural network (HHT-SANN). The incorporation of the HHT enhances the model's time-frequency analysis capabilities and allows for more effective nonlinear processing of the EEG data. By embedding the HHT within the self-attention module, the model captures intricate temporal and spectral patterns that are critical for accurate depression classification. We evaluated our method on a clinical EEG dataset comprising 34 MDD patients and 30 healthy controls from the Hospital of Universiti Sains Malaysia. Experimental results indicate that the proposed method achieves an accuracy of 98.78%, sensitivity of 99.23%, and specificity of 98.27%, outperforming traditional models and offering a more robust solution for depression detection. This work contributes to advancing the field of neuroinformatics by providing a more interpretable and effective model for mental health diagnostics based on EEG data.