EEG-Based Depression Recognition Using Convolutional Neural Network with FFT and EMD

Jing Zhu,Xiaowei Li,Pengfei Hou,Bin Hu,Xin Zhang
DOI: https://doi.org/10.1109/bibm58861.2023.10385385
2023-01-01
Abstract:Deep learning methods have been widely adopted in the field of computer-aided EEG diagnosis, one of the major topics to be investigated is the input format of EEG data. Related researches have reported the application of Fast Fourier Transform (FFT) to topology-preserving multi-spectral images generation on three frequency bands (i.e. theta, alpha and beta) jointed to preserve the spatial information. In our work, we proposed a new approach of using Empirical Mode Decomposition (EMD) instead of FFT algorithm to generate topology-preserving multi-spectral images on intrinsic mode functions (IMFs) jointed and only on the single IMF. Meanwhile, images generated on three frequency bands jointed and the single frequency band were also used for comparison. We then applied two convolutional neural network (CNN) structures to distinguish depression patients and normal subjects from the topology-preserving multi-spectral images. As a result, both convolutional networks obtain better accuracies on three IMFs jointed (about 5% better, ≈ 75% vs. ≈ 70%) than three frequency bands jointed. Analysis based on the single frequency band indicates that alpha band performs the best, with an accuracy of more than 78%, and among all classification results, the best classification accuracy obtained is 80.23% on IMF2. The results are encouraging, despite the limited size of our cohort, the use of EMD and our findings cast a new light on application of deep learning method to EEG-based depression recognition.
What problem does this paper attempt to address?