Application of graph frequency attention convolutional neural networks in depression treatment response

Abstract

Depression, a prevalent global mental health disorder, necessitates precise treatment response prediction for the improvement of personalized care and patient prognosis. The Graph Convolutional Neural Networks (GCNs) have emerged as a promising technique for handling intricate signals and classification tasks owing to their end-to-end neural architecture and nonlinear processing capabilities. In this context, this article proposes a model named the Graph Frequency Attention Convolutional Neural Network (GFACNN). Primarily, the model transforms the EEG signals into graphs to depict the connections between electrodes and brain regions, while integrating a frequency attention module to accentuate brain rhythm information. The proposed approach delves into the application of graph neural networks in the classification of EEG data, aiming to evaluate the response to antidepressant treatment and discern between treatment-resistant and treatment-responsive cases. Experimental results obtained from an EEG dataset at Peking University People's Hospital demonstrate the notable performance of GFACNN in distinguishing treatment responses among depression patients, surpassing deep learning methodologies including CapsuleNet and GoogLeNet. This highlights the efficacy of graph neural networks in leveraging the connections within EEG signal data. Overall, GFACNN exhibits potential for the classification of depression EEG signals, thereby potentially aiding clinical diagnosis and treatment.

Keywords: EEG; classification; depression treatment response; frequency attention; graph convolutional neural networks.