A Modified Transformer Network for Seizure Detection Using EEG Signals

被引:2
作者
Hu, Wenrong [1 ]
Wang, Juan [1 ]
Li, Feng [1 ]
Ge, Daohui [1 ]
Wang, Yuxia [1 ]
Jia, Qingwei [1 ]
Yuan, Shasha [1 ]
机构
[1] Qufu Normal Univ, Sch Comp Sci, Rizhao 276826, Peoples R China
基金
中国国家自然科学基金;
关键词
EEG; seizure detection; discrete wavelet transform; Co-MixUp; transformer; NEURAL-NETWORK;
D O I
10.1142/S0129065725500030
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Seizures have a serious impact on the physical function and daily life of epileptic patients. The automated detection of seizures can assist clinicians in taking preventive measures for patients during the diagnosis process. The combination of deep learning (DL) model with convolutional neural network (CNN) and transformer network can effectively extract both local and global features, resulting in improved seizure detection performance. In this study, an enhanced transformer network named Inresformer is proposed for seizure detection, which is combined with Inception and Residual network extracting different scale features of electroencephalography (EEG) signals to enrich the feature representation. In addition, the improved transformer network replaces the existing Feedforward layers with two half-step Feedforward layers to enhance the nonlinear representation of the model. The proposed architecture utilizes discrete wavelet transform (DWT) to decompose the original EEG signals, and the three sub-bands are selected for signal reconstruction. Then, the Co-MixUp method is adopted to solve the problem of data imbalance, and the processed signals are sent to the Inresformer network for seizure information capture and recognition. Finally, discriminant fusion is performed on the results of three-scale EEG sub-signals to achieve final seizure recognition. The proposed network achieves the best accuracy of 100% on Bonn dataset and the average accuracy of 98.03%, sensitivity of 95.65%, and specificity of 98.57% on the long-term CHB-MIT dataset. Compared to the existing DL networks, the proposed method holds significant potential for clinical research and diagnosis applications with competitive performance.
引用
收藏
页数:15
相关论文
共 61 条
[1]   Automated seizure prediction [J].
Acharya, U. Rajendra ;
Hagiwara, Yuki ;
Adeli, Hojjat .
EPILEPSY & BEHAVIOR, 2018, 88 :251-261
[2]   Deep convolutional neural network for the automated detection and diagnosis of seizure using EEG signals [J].
Acharya, U. Rajendra ;
Oh, Shu Lih ;
Hagiwara, Yuki ;
Tan, Jen Hong ;
Adeli, Hojjat .
COMPUTERS IN BIOLOGY AND MEDICINE, 2018, 100 :270-278
[3]   A wavelet-chaos methodology for analysis of EEGs and EEG subbands to detect seizure and epilepsy [J].
Adeli, Hojjat ;
Ghosh-Dastidar, Samanwoy ;
Dadmehr, Nahid .
IEEE TRANSACTIONS ON BIOMEDICAL ENGINEERING, 2007, 54 (02) :205-211
[4]   Indications of nonlinear deterministic and finite-dimensional structures in time series of brain electrical activity: Dependence on recording region and brain state [J].
Andrzejak, RG ;
Lehnertz, K ;
Mormann, F ;
Rieke, C ;
David, P ;
Elger, CE .
PHYSICAL REVIEW E, 2001, 64 (06) :8-061907
[5]  
Boonyakitanont P, 2019, A Comparison of Deep Neural Networks for Seizure Detection in EEG Signals, DOI [10.1101/702654, DOI 10.1101/702654]
[6]   Compact Convolutional Neural Network with Multi-Headed Attention Mechanism for Seizure Prediction [J].
Ding, Xin ;
Nie, Weiwei ;
Liu, Xinyu ;
Wang, Xiuying ;
Yuan, Qi .
INTERNATIONAL JOURNAL OF NEURAL SYSTEMS, 2023, 33 (03)
[7]   Epileptic Seizure Detection with an End-to-End Temporal Convolutional Network and Bidirectional Long Short-Term Memory Model [J].
Dong, Xingchen ;
Wen, Yiming ;
Ji, Dezan ;
Yuan, Shasha ;
Liu, Zhen ;
Shang, Wei ;
Zhou, Weidong .
INTERNATIONAL JOURNAL OF NEURAL SYSTEMS, 2024, 34 (03)
[8]   Multimodal multitask deep learning model for Alzheimer's disease progression detection based on time series data [J].
El-Sappagh, Shaker ;
Abuhmed, Tamer ;
Islam, S. M. Riazul ;
Kwak, Kyung Sup .
NEUROCOMPUTING, 2020, 412 :197-215
[9]   EW-Fisher: A Novel Loss Function for Deep Learning-Based Image Co-Segmentation [J].
Gong, Xiaopeng ;
Liu, Xiabi ;
Duan, Xin ;
Li, Yushuo .
NEURAL PROCESSING LETTERS, 2020, 52 (03) :2399-2413
[10]  
Goodfellow I, 2016, ADAPT COMPUT MACH LE, P1