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 条
  • [41] Feature extraction based on sparse representation with application to epileptic EEG classification
    Wang, Jing
    Gao, X. Z.
    Guo, Ping
    INTERNATIONAL JOURNAL OF IMAGING SYSTEMS AND TECHNOLOGY, 2013, 23 (02) : 104 - 113
  • [42] EEG Epileptic Data Classification Using the Schrodinger Operator's Spectrum
    Sadoun, Maria Sara Nour
    Rahman, Muhammad Mahboob Ur
    Al-Naffouri, Tareq
    Laleg-Kirati, Taous-Meriem
    2023 45TH ANNUAL INTERNATIONAL CONFERENCE OF THE IEEE ENGINEERING IN MEDICINE & BIOLOGY SOCIETY, EMBC, 2023,
  • [43] Classification of Epileptic EEG Signals Using Synchrosqueezing Transform and Machine Learning
    Cura, Ozlem Karabiber
    Akan, Aydin
    INTERNATIONAL JOURNAL OF NEURAL SYSTEMS, 2021, 31 (05)
  • [44] Positional multi-length and mutual-attention network for epileptic seizure classification
    Zhang, Guokai
    Zhang, Aiming
    Liu, Huan
    Luo, Jihao
    Chen, Jianqing
    FRONTIERS IN COMPUTATIONAL NEUROSCIENCE, 2024, 18
  • [45] EPILEPTIC EEG CLASSIFICATION USING NONLINEAR PARAMETERS ON DIFFERENT FREQUENCY BANDS
    Martis, Roshan Joy
    Tan, Jen Hong
    Chua, Chua Kuang
    Loon, Too Cheah
    Jie, Sharon Yeo Wan
    Tong, Louis
    JOURNAL OF MECHANICS IN MEDICINE AND BIOLOGY, 2015, 15 (03)
  • [46] Multi-kernel-based random vector functional link network with decomposed features for epileptic EEG signal classification
    Parija, Sebamai
    Dash, Pradipta Kishore
    Bisoi, Ranjeeta
    IET SIGNAL PROCESSING, 2020, 14 (03) : 162 - 174
  • [47] Epileptic EEG classification based on extreme learning machine and nonlinear features
    Yuan, Qi
    Zhou, Weidong
    Li, Shufang
    Cai, Dongmei
    EPILEPSY RESEARCH, 2011, 96 (1-2) : 29 - 38
  • [48] Epileptic EEG Signal Classification with ANFIS based on Harmony Search Method
    Wang, Jing
    Gao, X. Z.
    Tanskanen, Jarno M. A.
    Guo, Ping
    PROCEEDINGS OF THE 2012 EIGHTH INTERNATIONAL CONFERENCE ON COMPUTATIONAL INTELLIGENCE AND SECURITY (CIS 2012), 2012, : 690 - 694
  • [49] EEG-based Classification of Drivers Attention using Convolutional Neural Network
    Atilla, Fred
    Alimardani, Maryam
    PROCEEDINGS OF THE 2021 IEEE INTERNATIONAL CONFERENCE ON HUMAN-MACHINE SYSTEMS (ICHMS), 2021, : 59 - 62
  • [50] An adaptive driver fatigue classification framework using EEG and attention-based hybrid neural network with individual feature subsets
    Wang, Yuchen
    Fang, Zijie
    Sun, Xiaoyong
    Lin, Xuefen
    Niu, Lei
    Ma, Weifeng
    BIOMEDICAL SIGNAL PROCESSING AND CONTROL, 2023, 85