Multi-task transformer network for subject-independent iEEG seizure detection

被引:0
|
作者
Sun, Yulin [1 ]
Cheng, Longlong [1 ,4 ]
Si, Xiaopeng [1 ]
He, Runnan [1 ]
Pereira, Tania [5 ]
Pang, Meijun [1 ]
Zhang, Kuo [1 ]
Song, Xin [1 ]
Ming, Dong [1 ,2 ,3 ]
Liu, Xiuyun [1 ,2 ,3 ]
机构
[1] Tianjin Univ, Med Sch, Tianjin 300072, Peoples R China
[2] Tianjin Univ, State Key Lab Adv Med Mat & Devices, Tianjin 300072, Peoples R China
[3] Haihe Lab Brain Comp Interact & Human Machine Inte, Tianjin 300392, Peoples R China
[4] China Elect Cloud Brain Tianjin Technol Co Ltd, Tianjin 300392, Peoples R China
[5] INESC TEC Inst Syst & Comp Engn Technol & Sci, P-4200465 Proto, Portugal
关键词
Seizure detection; Intracranial EEG; Multi-task learning; Transformer network; Subject-independent; Channel-wise mixup; EEG; EPILEPSY;
D O I
10.1016/j.eswa.2024.126282
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Subject-independent seizure detection algorithms are typically grounded in scalp electroencephalogram (EEG) databases, due to standardized channels and locations of EEG electrodes. Intracranial EEG (iEEG) has the characteristics of low noise and high temporal resolution compared with scalp EEG. However, it is still a big challenge for seizure detection using iEEG, because of the inconsistent number and locations of implanted electrodes in different patients, which results in a lack of unified algorithms. This study introduces an innovative approach for subject-independent seizure detection using iEEG, combining channel-wise mixup, transformer networks, and multi-task learning. Channel-wise mixup enhances data utilization by effectively leveraging information from different subjects, while multi-task learning improves the generalization of the model by concurrently optimizing both the seizure detection and the subject recognition tasks. 2983 files from two well-known epilepsy databases, i.e. SWEC-ETHZ and HUP were used in our study and the result showed that our approach surpasses currently existing methods. In terms of accuracy and generalization of seizure detection, our method achieved an area under the receiver operating characteristic curve (AUC) of 0.97 and 0.95 on the two databases respectively, which are significantly higher than the result of the currently existing methods. This study proposed anew method with great potential for surgery planning of epilepsy patients.
引用
收藏
页数:12
相关论文
共 50 条
  • [41] Multi-Task Learning with Personalized Transformer for Review Recommendation
    Wang, Haiming
    Liu, Wei
    Yin, Jian
    WEB INFORMATION SYSTEMS ENGINEERING - WISE 2021, PT II, 2021, 13081 : 162 - 176
  • [42] TransNuSeg: A Lightweight Multi-task Transformer for Nuclei Segmentation
    He, Zhenqi
    Unberath, Mathias
    Ke, Jing
    Shen, Yiqing
    MEDICAL IMAGE COMPUTING AND COMPUTER ASSISTED INTERVENTION, MICCAI 2023, PT IV, 2023, 14223 : 206 - 215
  • [43] SpotNet: Self-Attention Multi-Task Network for Object Detection
    Perreault, Hughes
    Bilodeau, Guillaume-Alexandre
    Saunier, Nicolas
    Heritier, Maguelonne
    2020 17TH CONFERENCE ON COMPUTER AND ROBOT VISION (CRV 2020), 2020, : 230 - 237
  • [44] Multi-task Feature Learning Based Anomaly Detection of Network Dataflow
    Ren Hui-feng
    Yan Feng
    Dong Qing-chao
    2020 CHINESE AUTOMATION CONGRESS (CAC 2020), 2020, : 4144 - 4147
  • [45] Face Occlusion Detection Based on Multi-task Convolution Neural Network
    Xia, Yizhang
    Zhang, Bailing
    Coenen, Frans
    2015 12TH INTERNATIONAL CONFERENCE ON FUZZY SYSTEMS AND KNOWLEDGE DISCOVERY (FSKD), 2015, : 375 - 379
  • [46] Centered Multi-Task Generative Adversarial Network for Small Object Detection
    Wang, Hongfeng
    Wang, Jianzhong
    Bai, Kemeng
    Sun, Yong
    SENSORS, 2021, 21 (15)
  • [47] Multi-task Facial Landmark Detection Network for Early ASD Screening
    Lin, Ruihan
    Zhang, Hanlin
    Wang, Xinming
    Ren, Weihong
    Wu, Wenhao
    Liu, Zuode
    Xu, Xiu
    Xu, Qiong
    Liu, Honghai
    INTELLIGENT ROBOTICS AND APPLICATIONS (ICIRA 2022), PT I, 2022, 13455 : 381 - 391
  • [48] A Multi-task Learning Network for Automated Detection of Oral Epithelial Dysplasia
    Aljuaid, Abeer
    Almohaya, Mai
    Aljuaid, Ashwaq
    Anwar, Mohd
    Proceedings - 2024 IEEE International Conference on Information Reuse and Integration for Data Science, IRI 2024, 2024, : 25 - 30
  • [49] Automatic Multi-task Learning System for Abnormal Network Traffic Detection
    Huang, He
    Deng, Haojiang
    Chen, Jun
    Han, Luchao
    Wang, Wei
    INTERNATIONAL JOURNAL OF EMERGING TECHNOLOGIES IN LEARNING, 2018, 13 (04): : 4 - 20
  • [50] Multi-Task Network Representation Learning
    Xie, Yu
    Jin, Peixuan
    Gong, Maoguo
    Zhang, Chen
    Yu, Bin
    FRONTIERS IN NEUROSCIENCE, 2020, 14