Efficient Seizure Detection by Complementary Integration of Convolutional Neural Network and Vision Transformer.

Journal: International journal of neural systems
Published Date:

Abstract

Epilepsy, as a prevalent neurological disorder, is characterized by its high incidence, sudden onset, and recurrent nature. The development of an accurate and real-time automatic seizure detection system is crucial for assisting clinicians in making precise diagnoses and providing timely treatment for epilepsy. However, conventional automatic seizure detection methods often face limitations in simultaneously capturing both local features and long-range correlations inherent in EEG signals, which constrains the accuracy of these existing detection systems. To address this challenge, we propose a novel end-to-end seizure detection framework, named CNN-ViT, which complementarily integrates a Convolutional Neural Network (CNN) for capturing local inductive bias of EEG and Vision Transformer (ViT) for further mining their long-range dependency. Initially, raw electroencephalogram (EEG) signals are filtered and segmented and then sent into the CNN-ViT model to learn their local and global feature representations and identify the seizure patterns. Meanwhile, we adopt a global max-pooling strategy to reduce the scale of the CNN-ViT model and make it focus on the most discriminative features. Given the occurrence of diverse artifacts in long-term EEG recordings, we further employ post-processing techniques to improve the seizure detection performance. The proposed CNN-ViT model, when evaluated using the publicly accessible CHB-MIT EEG dataset, reveals its outstanding performance with a sensitivity of 99.34% at a segment-based level and 99.70% at an event-based level. On the SH-SDU dataset we collected, our method yielded a segment-based sensitivity of 99.86%, specificity of 94.33%, and accuracy of 94.40%, along with an event-based sensitivity of 100%. The total processing time for 1[Formula: see text]h EEG data was only 3.07[Formula: see text]s. These exceptional results demonstrate the potential of our method as a reference for clinical real-time seizure detection applications.

Authors

  • Jiaqi Wang
  • Haotian Li
    Key Laboratory for Biomedical Engineering of Ministry of Education, Department of Biomedical Engineering, College of Biomedical Engineering & Instrument Science, Zhejiang University, Hangzhou, 310027, China.
  • Chuanyu Li
    School of Integrated Circuits, Shandong University, Jinan 250100, P. R. China.
  • Weisen Lu
    School of Integrated Circuits, Shandong University, Jinan 250100, P. R. China.
  • Haozhou Cui
    School of Integrated Circuits, Shandong University, Jinan 250100, P. R. China.
  • Xiangwen Zhong
    School of Integrated Circuits, Shandong University, Jinan 250100, P. R. China.
  • Shuhao Ren
    School of Integrated Circuits, Shandong University, Jinan 250100, P. R. China.
  • Zhida Shang
    Shenzhen Institute of Shandong University, Shenzhen 518057, P. R. China.
  • Weidong Zhou