Dual-Modal Information Bottleneck Network for Seizure Detection.

Journal: International journal of neural systems
Published Date:

Abstract

In recent years, deep learning has shown very competitive performance in seizure detection. However, most of the currently used methods either convert electroencephalogram (EEG) signals into spectral images and employ 2D-CNNs, or split the one-dimensional (1D) features of EEG signals into many segments and employ 1D-CNNs. Moreover, these investigations are further constrained by the absence of consideration for temporal links between time series segments or spectrogram images. Therefore, we propose a Dual-Modal Information Bottleneck (Dual-modal IB) network for EEG seizure detection. The network extracts EEG features from both time series and spectrogram dimensions, allowing information from different modalities to pass through the Dual-modal IB, requiring the model to gather and condense the most pertinent information in each modality and only share what is necessary. Specifically, we make full use of the information shared between the two modality representations to obtain key information for seizure detection and to remove irrelevant feature between the two modalities. In addition, to explore the intrinsic temporal dependencies, we further introduce a bidirectional long-short-term memory (BiLSTM) for Dual-modal IB model, which is used to model the temporal relationships between the information after each modality is extracted by convolutional neural network (CNN). For CHB-MIT dataset, the proposed framework can achieve an average segment-based sensitivity of 97.42%, specificity of 99.32%, accuracy of 98.29%, and an average event-based sensitivity of 96.02%, false detection rate (FDR) of 0.70/h. We release our code at https://github.com/LLLL1021/Dual-modal-IB.

Authors

  • Jiale Wang
    School of Information Science and Engineering, Shandong Normal University, Jinan 250358, P. R. China.
  • Xinting Ge
    School of Information Science and Engineering, Shandong Normal University, Jinan 250358, P. R. China.
  • Yunfeng Shi
    School of Information Science and Engineering, Shandong Normal University, Jinan 250358, P. R. China.
  • Mengxue Sun
    School of Information Science and Engineering, Shandong Normal University, Jinan 250358, P. R. China.
  • Qingtao Gong
    Ulsan Ship and Ocean College, Ludong University, Yantai 264025, P. R. China.
  • Haipeng Wang
    Institute of Information Fusion, Naval, Aviation University, Yantai 264001, P. R. China.
  • Wenhui Huang
    Department of Radiology, Guangdong General Hospital/Guangdong Academy of Medical Sciences, Guangzhou, Guangdong, PR China; School of Medicine, South China University of Technology, Guangzhou, Guangdong, PR China.