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 条
  • [21] Multi-task network embedding
    Linchuan Xu
    Xiaokai Wei
    Jiannong Cao
    Philip S. Yu
    International Journal of Data Science and Analytics, 2019, 8 : 183 - 198
  • [22] MTLFormer: Multi-Task Learning Guided Transformer Network for Business Process Prediction
    Wang, Jiaojiao
    Huang, Jiawei
    Ma, Xiaoyu
    Li, Zhongjin
    Wang, Yaqi
    Yu, Dingguo
    IEEE ACCESS, 2023, 11 : 76722 - 76738
  • [23] Improving Vision Transformer with Multi-Task Training
    Ahn, Woo Jin
    Yang, Geun Yeong
    Choi, Hyun Duck
    Lim, Myo Taeg
    Kang, Tae Koo
    2022 22ND INTERNATIONAL CONFERENCE ON CONTROL, AUTOMATION AND SYSTEMS (ICCAS 2022), 2022, : 1963 - 1965
  • [24] PARFormer: Transformer-Based Multi-Task Network for Pedestrian Attribute Recognition
    Fan, Xinwen
    Zhang, Yukang
    Lu, Yang
    Wang, Hanzi
    IEEE TRANSACTIONS ON CIRCUITS AND SYSTEMS FOR VIDEO TECHNOLOGY, 2024, 34 (01) : 411 - 423
  • [25] Multi-task network embedding
    Xu, Linchuan
    Wei, Xiaokai
    Cao, Jiannong
    Yu, Philip S.
    INTERNATIONAL JOURNAL OF DATA SCIENCE AND ANALYTICS, 2019, 8 (02) : 183 - 198
  • [26] Paraphrase Bidirectional Transformer with Multi-Task Learning
    Ko, Bowon
    Choi, Ho-Jin
    2020 IEEE INTERNATIONAL CONFERENCE ON BIG DATA AND SMART COMPUTING (BIGCOMP 2020), 2020, : 217 - 220
  • [27] SMNet: Symmetric Multi-Task Network for Semantic Change Detection in Remote Sensing Images Based on CNN and Transformer
    Niu, Yiting
    Guo, Haitao
    Lu, Jun
    Ding, Lei
    Yu, Donghang
    REMOTE SENSING, 2023, 15 (04)
  • [28] Weighted Multi-Task Vision Transformer for Distraction and Emotion Detection in Driving Safety
    Wang, Yixiao
    Li, Zhe
    Guan, Guowei
    Sun, Yipin
    Wang, Chunyu
    Tohidypour, Hamid Reza
    Nasiopoulos, Panos
    Leung, Victor C. M.
    2024 INTERNATIONAL CONFERENCE ON COMPUTING, NETWORKING AND COMMUNICATIONS, ICNC, 2024, : 152 - 156
  • [29] Multi-task information propagation network for camouflaged object detection
    Zhao, Fei
    Lou, Wenzhong
    Feng, Hengzhen
    Ding, Nanxi
    Li, Chenglong
    Ma, Wenlong
    DIGITAL SIGNAL PROCESSING, 2025, 162
  • [30] Object detection network pruning with multi-task information fusion
    Shengming Li
    Linsong Xue
    Lin Feng
    Yifan Wang
    Dong Wang
    World Wide Web, 2022, 25 : 1667 - 1683