An interpretable and generalizable deep learning model for iEEG-based seizure prediction using prototype learning and contrastive learning

被引:1
作者
Gao, Yikai [1 ,2 ]
Liu, Aiping [2 ]
Cui, Heng [2 ]
Qian, Ruobing [2 ]
Chen, Xun [1 ,2 ]
机构
[1] Department of Neurosurgery, The First Affiliated Hospital of USTC, Division of Life Sciences and Medicine, University of Science and Technology of China, Anhui, Hefei
[2] Department of Electronic Engineering and Information Science, University of Science and Technology of China, Hefei
基金
中国国家自然科学基金;
关键词
Deep learning; Generalizability; Interpretability; Intracranial electroencephalography; Seizure prediction; Signal processing;
D O I
10.1016/j.compbiomed.2024.109257
中图分类号
学科分类号
摘要
Epileptic seizure prediction plays a crucial role in enhancing the quality of life for individuals with epilepsy. Over recent years, a multitude of deep learning-based approaches have emerged to tackle this challenging task, leading to significant advancements. However, the ‘black-box’ nature of deep learning models and the considerable interpatient variability significantly impede their interpretability and generalization, thereby severely hampering their efficacy in real-world clinical applications. To address these issues, our study aims to establish an interpretable and generalizable seizure prediction model that meets the demands of clinical diagnosis. Our method extends self-interpretable prototype learning networks into a novel domain adaptation framework designed specifically for cross-patient seizure prediction. The proposed framework enables patient-level interpretability by tracing the origins of significant prototypes. For instance, it could provide information about the seizure type of the patient to which the prototype belongs. This surpasses the existing sample-level interpretability, which is limited to individual patient samples. To further improve the model's generalization capability, we introduce a contrastive semantic alignment loss constraint to the embedding space, enhancing the robustness of the learned prototypes. We evaluate our proposed model using the Freiburg intracranial electroencephalography (iEEG) dataset, which consists of 20 patients and a total of 82 seizures. The experimental results demonstrated a high sensitivity of 79.0%, a low false prediction rate of 0.183, and a high area under the receiver operating characteristic curve (AUC) value of 0.804, achieving state-of-the-art performance with self-interpretable evidence in contrast to the current cross-patient seizure prediction methods. Our study represents a significant step forward in developing an interpretable and generalizable model for seizure prediction, thereby facilitating the application of deep learning models in clinical diagnosis. © 2024 Elsevier Ltd
引用
收藏
相关论文
共 50 条
  • [41] SeizFt: Interpretable Machine Learning for Seizure Detection Using Wearables
    Al-Hussaini, Irfan
    Mitchell, Cassie S.
    BIOENGINEERING-BASEL, 2023, 10 (08):
  • [42] Patient-Specific Seizure Prediction via Adder Network and Supervised Contrastive Learning
    Zhao, Yuchang
    Li, Chang
    Liu, Xiang
    Qian, Ruobing
    Song, Rencheng
    Chen, Xun
    IEEE TRANSACTIONS ON NEURAL SYSTEMS AND REHABILITATION ENGINEERING, 2022, 30 : 1536 - 1547
  • [43] Prediction of shear strength of RC deep beams based on interpretable machine learning
    Ma, Cailong
    Wang, Sixuan
    Zhao, Jianping
    Xiao, Xufeng
    Xie, Chenxi
    Feng, Xinlong
    CONSTRUCTION AND BUILDING MATERIALS, 2023, 387
  • [44] Deep learning-based dispersion prediction model for hazardous chemical leaks using transfer learning
    Han, Xiaoyi
    Zhu, Jiaxing
    Li, Haosen
    Xu, Wei
    Feng, Junjie
    Hao, Lin
    Wei, Hongyuan
    PROCESS SAFETY AND ENVIRONMENTAL PROTECTION, 2024, 188 : 363 - 373
  • [45] Epileptic Seizures Detection Using iEEG Signals and Deep Learning Models
    Abderrahim, Nourane
    Echtioui, Amira
    Khemakhem, Rafik
    Zouch, Wassim
    Ghorbel, Mohamed
    Ben Hamida, Ahmed
    CIRCUITS SYSTEMS AND SIGNAL PROCESSING, 2023, 43 (3) : 1597 - 1626
  • [46] Contrastive Learning and Deep Fusion Recommendation Model Based on ID Features
    Li, Bing
    Wang, Xile
    Dong, Jiangtao
    Hou, Yuqi
    Yang, Biao
    IEEE ACCESS, 2024, 12 : 163001 - 163015
  • [47] An Efficient Deep Learning System for Epileptic Seizure Prediction
    Abdelhameed, Ahmed M.
    Bayoumi, Magdy
    2021 IEEE INTERNATIONAL SYMPOSIUM ON CIRCUITS AND SYSTEMS (ISCAS), 2021,
  • [48] A novel deep learning prediction model for concrete dam displacements using interpretable mixed attention mechanism
    Ren, Qiubing
    Li, Mingchao
    Li, Heng
    Shen, Yang
    ADVANCED ENGINEERING INFORMATICS, 2021, 50
  • [49] Epileptic Seizures Detection Using iEEG Signals and Deep Learning Models
    Nourane Abderrahim
    Amira Echtioui
    Rafik Khemakhem
    Wassim Zouch
    Mohamed Ghorbel
    Ahmed Ben Hamida
    Circuits, Systems, and Signal Processing, 2024, 43 : 1597 - 1626
  • [50] End-to-end model for automatic seizure detection using supervised contrastive learning
    Li, Haotian
    Dong, Xingchen
    Zhong, Xiangwen
    Li, Chuanyu
    Cui, Haozhou
    Zhou, Weidong
    ENGINEERING APPLICATIONS OF ARTIFICIAL INTELLIGENCE, 2024, 133