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 条
  • [1] Epileptic EEG Classification via Graph Transformer Network
    Lian, Jian
    Xu, Fangzhou
    INTERNATIONAL JOURNAL OF NEURAL SYSTEMS, 2023, 33 (08)
  • [2] Automatic epileptic EEG classification based on differential entropy and attention model
    Zhang, Jian
    Wei, Zuochen
    Zou, Junzhong
    Fu, Hao
    ENGINEERING APPLICATIONS OF ARTIFICIAL INTELLIGENCE, 2020, 96
  • [3] EEG Classification Using Hybrid Convolutional Neural Network with Attention Mechanism
    Ciurea, Alexe
    Manoila, Cristina-Petruta
    Ionescu, Bogdan
    ADVANCES IN DIGITAL HEALTH AND MEDICAL BIOENGINEERING, VOL 1, EHB-2023, 2024, 109 : 783 - 791
  • [4] WTRPNet: An Explainable Graph Feature Convolutional Neural Network for Epileptic EEG Classification
    Xin, Qi
    Hu, Shaohao
    Liu, Shuaiqi
    Zhao, Ling
    Wang, Shuihua
    ACM TRANSACTIONS ON MULTIMEDIA COMPUTING COMMUNICATIONS AND APPLICATIONS, 2021, 17 (03)
  • [5] SSGCNet: A Sparse Spectra Graph Convolutional Network for Epileptic EEG Signal Classification
    Wang, Jialin
    Gao, Rui
    Zheng, Haotian
    Zhu, Hao
    Shi, C. -J. Richard
    IEEE TRANSACTIONS ON NEURAL NETWORKS AND LEARNING SYSTEMS, 2024, 35 (09) : 12157 - 12171
  • [6] Classification of epileptic EEG using neural network and wavelet transform
    Petrosian, A
    Homan, R
    Prokhorov, D
    Wunsch, D
    WAVELET APPLICATIONS IN SIGNAL AND IMAGE PROCESSING IV, PTS 1 AND 2, 1996, 2825 : 834 - 843
  • [7] Classification of Epileptic EEG Signals Using Dynamic Mode Decomposition
    Cura, Ozlem Karabiber
    Pehlivan, Sude
    Akan, Aydin
    2020 28TH SIGNAL PROCESSING AND COMMUNICATIONS APPLICATIONS CONFERENCE (SIU), 2020,
  • [8] Classification of Epileptic and Non-Epileptic EEG Events by Feature Selection f-score
    Noertjahjani, Siswandari
    Hidayat, Risanuari
    Susanto, Adhi
    Wibowo, Samekto
    2017 5TH INTERNATIONAL CONFERENCE ON INSTRUMENTATION, CONTROL, AND AUTOMATION (ICA), 2017, : 182 - 187
  • [9] Impact of the Pre-processing and Balancing of EEG Data on the Performance of Graph Neural Network for Epileptic Seizure Classification
    Mazurek, Szymon
    Blanco, Rosmary
    Falco-Roget, Joan
    Argasinski, Jan K.
    Crimi, Alessandro
    ARTIFICIAL INTELLIGENCE AND SOFT COMPUTING, ICAISC 2023, PT II, 2023, 14126 : 258 - 268
  • [10] EEG Signals Classification for Epileptic Detection: A review
    Houssein, Essam H.
    Hassanien, Aboul Ella
    Ismaeel, Alaa A. K.
    PROCEEDINGS OF THE SECOND INTERNATIONAL CONFERENCE ON INTERNET OF THINGS, DATA AND CLOUD COMPUTING (ICC 2017), 2017,