Hybrid Attention Network for Epileptic EEG Classification

被引:18
作者
Zhao, Yanna [1 ]
He, Jiatong [1 ]
Zhu, Fenglin [1 ]
Xiao, Tiantian [1 ]
Zhang, Yongfeng [1 ]
Wang, Ziwei [1 ]
Xu, Fangzhou [2 ,3 ]
Niu, Yi [1 ]
机构
[1] Shandong Normal Univ, Sch Informat Sci & Engn, Jinan 250358, Peoples R China
[2] Qilu Univ Technol, Shandong Acad Sci, Int Sch Optoelect Engn, Jinan 250353, Peoples R China
[3] Jinan Engn Lab Human Machine Intelligent Cooperat, Jinan 250353, Peoples R China
基金
中国国家自然科学基金;
关键词
Seizure detection; EEG; graph attention network; transformer; focal loss; SEIZURE DETECTION;
D O I
10.1142/S0129065723500314
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Automatic seizure detection from electroencephalography (EEG) based on deep learning has been significantly improved. However, existing works have not adequately excavate the spatial-temporal information between EEG channels. Besides, most works mainly focus on patient-specific scenarios while cross-patient seizure detection is more challenging and meaningful. Regarding the above problems, we propose a hybrid attention network (HAN) for automatic seizure detection. Specifically, the graph attention network (GAT) extracts spatial features at the front end, and Transformer gets time features as the back end. HAN leverages the attention mechanism and fully extracts the spatial-temporal correlation of EEG signals. The focal loss function is introduced to HAN to deal with the imbalance of the dataset accompanied by seizure detection based on EEG. Both patient-specific and patient-independent experiments are carried out on the public CHB-MIT database. Experimental results demonstrate the efficacy of HAN in both experimental settings.
引用
收藏
页数:14
相关论文
共 50 条
  • [21] Domain adaptation for epileptic EEG classification using adversarial learning and Riemannian manifold
    Peng, Peizhen
    Xie, Liping
    Zhang, Kanjian
    Zhang, Jinxia
    Yang, Lu
    Wei, Haikun
    BIOMEDICAL SIGNAL PROCESSING AND CONTROL, 2022, 75
  • [22] TESANet: Self-attention network for olfactory EEG classification
    Tong, Chengxuan
    Ding, Yi
    Liang, Kevin Lim Jun
    Zhang, Zhuo
    Zhang, Haihong
    Guan, Cuntai
    2022 INTERNATIONAL JOINT CONFERENCE ON NEURAL NETWORKS (IJCNN), 2022,
  • [23] A New Wavelet-Based Neural Network for Classification of Epileptic-Related States using EEG
    Juarez-Guerra, E.
    Alarcon-Aquino, V.
    Gomez-Gil, P.
    Ramirez-Cortes, J. M.
    Garcia-Trevino, E. S.
    JOURNAL OF SIGNAL PROCESSING SYSTEMS FOR SIGNAL IMAGE AND VIDEO TECHNOLOGY, 2020, 92 (02): : 187 - 211
  • [24] An Attention-based Architecture for EEG Classification
    Zoppis, Italo
    Zanga, Alessio
    Manzoni, Sara
    Cisotto, Giulia
    Morreale, Angela
    Stella, Fabio
    Mauri, Giancarlo
    PROCEEDINGS OF THE 13TH INTERNATIONAL JOINT CONFERENCE ON BIOMEDICAL ENGINEERING SYSTEMS AND TECHNOLOGIES, VOL 4: BIOSIGNALS, 2020, : 214 - 219
  • [25] Machine learning-based EEG signals classification model for epileptic seizure detection
    Aayesha
    Qureshi, Muhammad Bilal
    Afzaal, Muhammad
    Qureshi, Muhammad Shuaib
    Fayaz, Muhammad
    MULTIMEDIA TOOLS AND APPLICATIONS, 2021, 80 (12) : 17849 - 17877
  • [26] Machine learning-based EEG signals classification model for epileptic seizure detection
    Muhammad Bilal Aayesha
    Muhammad Qureshi
    Muhammad Shuaib Afzaal
    Muhammad Qureshi
    Multimedia Tools and Applications, 2021, 80 : 17849 - 17877
  • [27] A stable feature extraction method in classification epileptic EEG signals
    Kaya, Yilmaz
    Ertugrul, Omer Faruk
    AUSTRALASIAN PHYSICAL & ENGINEERING SCIENCES IN MEDICINE, 2018, 41 (03) : 721 - 730
  • [28] NEURAL NETWORK CLASSIFICATION OF EEG SIGNAL FOR THE DETECTION OF SEIZURE
    Yasmeen, Shaguftha
    Karki, Maya V.
    2017 2ND IEEE INTERNATIONAL CONFERENCE ON RECENT TRENDS IN ELECTRONICS, INFORMATION & COMMUNICATION TECHNOLOGY (RTEICT), 2017, : 553 - 558
  • [29] Imbalanced classification for epileptic EEG signals based on deep learning
    Fei H.
    Yuan Q.
    Zheng Y.
    Yi Qi Yi Biao Xue Bao/Chinese Journal of Scientific Instrument, 2021, 42 (03): : 231 - 240
  • [30] A COMPARISON OF DYNAMIC MODELING APPROACHES FOR EPILEPTIC EEG DETECTION AND CLASSIFICATION
    Song, Xiaomu
    Aguilar, Luis
    Yoon, Suk-chung
    42ND ANNUAL INTERNATIONAL CONFERENCES OF THE IEEE ENGINEERING IN MEDICINE AND BIOLOGY SOCIETY: ENABLING INNOVATIVE TECHNOLOGIES FOR GLOBAL HEALTHCARE EMBC'20, 2020, : 523 - 527