Epileptic EEG Classification via Graph Transformer Network
Jian Lian, Fangzhou Xu
Abstract
Deep learning-based epileptic seizure recognition via electroencephalogram signals has shown considerable potential for clinical practice. Although deep learning algorithms can enhance epilepsy identification accuracy compared with classical machine learning techniques, classifying epileptic activities based on the association between multichannel signals in electroencephalogram recordings is still challenging in automated seizure classification from electroencephalogram signals. Furthermore, the performance of generalization is hardly maintained by the fact that existing deep learning models were constructed using just one architecture. This study focuses on addressing this challenge using a hybrid framework. Alternatively put, a hybrid deep learning model, which is based on the ground-breaking graph neural network and transformer architectures, was proposed. The proposed deep architecture consists of a graph model to discover the inner relationship between multichannel signals and a transformer to reveal the heterogeneous associations between the channels. To evaluate the performance of the proposed approach, the comparison experiments were conducted on a publicly available dataset between the state-of-the-art algorithms and ours. Experimental results demonstrate that the proposed method is a potentially valuable instrument for epoch-based epileptic EEG classification.