A Lightweight Convolutional Neural Network-Reformer Model for Efficient Epileptic Seizure Detection

被引:2
|
作者
Cui, Haozhou [1 ,2 ]
Zhong, Xiangwen [1 ,2 ]
Li, Haotian [1 ,2 ]
Li, Chuanyu [1 ,2 ]
Dong, Xingchen [1 ,2 ]
Ji, Dezan [1 ,2 ]
He, Landi [1 ,2 ]
Zhou, Weidong [1 ,2 ]
机构
[1] Shandong Univ, Sch Integrated Circuits, Jinan 250100, Peoples R China
[2] Shandong Univ, Shenzhen Inst, Shenzhen 518057, Peoples R China
基金
中国国家自然科学基金;
关键词
Electroencephalogram; seizure detection; convolutional neural network; locality sensitive hashing attention; Reformer; EEG; TRANSFORM;
D O I
10.1142/S0129065724500655
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
A real-time and reliable automatic detection system for epileptic seizures holds significant value in assisting physicians with rapid diagnosis and treatment of epilepsy. Aiming to address this issue, a novel lightweight model called Convolutional Neural Network-Reformer (CNN-Reformer) is proposed for seizure detection on long-term EEG. The CNN-Reformer consists of two main parts: the Data Reshaping (DR) module and the Efficient Attention and Concentration (EAC) module. This framework reduces network parameters while retaining effective feature extraction of multi-channel EEGs, thereby improving model computational efficiency and real-time performance. Initially, the raw EEG signals undergo Discrete Wavelet Transform (DWT) for signal filtering, and then fed into the DR module for data compression and reshaping while preserving local features. Subsequently, these local features are sent to the EAC module to extract global features and perform categorization. Post-processing involving sliding window averaging, thresholding, and collar techniques is further deployed to reduce the false detection rate (FDR) and improve detection performance. On the CHB-MIT scalp EEG dataset, our method achieves an average sensitivity of 97.57%, accuracy of 98.09%, and specificity of 98.11% at segment-based level, and a sensitivity of 96.81%, along with FDR of 0.27/h, and latency of 17.81 s at the event-based level. On the SH-SDU dataset we collected, our method yielded segment-based sensitivity of 94.51%, specificity of 92.83%, and accuracy of 92.81%, along with event-based sensitivity of 94.11%. The average testing time for 1h of multi-channel EEG signals is 1.92s. The excellent results and fast computational speed of the CNN-Reformer model demonstrate its potential for efficient seizure detection.
引用
收藏
页数:18
相关论文
共 50 条
  • [31] A Lightweight Convolutional Neural Network Flame Detection Algorithm
    Li, Wenzheng
    Yu, Zongyang
    PROCEEDINGS OF 2021 IEEE 11TH INTERNATIONAL CONFERENCE ON ELECTRONICS INFORMATION AND EMERGENCY COMMUNICATION (ICEIEC 2021), 2021, : 83 - 86
  • [32] A Lightweight Convolutional Neural Network for Salient Object Detection
    Fei, Fengchang
    Liu, Wei
    Shu, Lei
    TEHNICKI VJESNIK-TECHNICAL GAZETTE, 2024, 31 (04): : 1402 - 1410
  • [33] Efficient and generalizable cross-patient epileptic seizure detection through a spiking neural network
    Zhang, Zongpeng
    Xiao, Mingqing
    Ji, Taoyun
    Jiang, Yuwu
    Lin, Tong
    Zhou, Xiaohua
    Lin, Zhouchen
    FRONTIERS IN NEUROSCIENCE, 2024, 17
  • [34] FPGA implementation of Wavelet Neural Network for epileptic seizure detection
    Kumar, Hari R.
    Balasubramani, M.
    Babu, Ganesh
    IEICE ELECTRONICS EXPRESS, 2013, 10 (24):
  • [35] Epileptic Seizure Prediction: A Multi-Scale Convolutional Neural Network Approach
    Hussein, Ramy
    Ward, Rabab
    2019 7TH IEEE GLOBAL CONFERENCE ON SIGNAL AND INFORMATION PROCESSING (IEEE GLOBALSIP), 2019,
  • [36] Epileptic Seizure Prediction: A Semi-Dilated Convolutional Neural Network Architecture
    Hussein, Ramy
    Lee, Soojin
    Ward, Rabab
    McKeown, Martin J.
    2020 25TH INTERNATIONAL CONFERENCE ON PATTERN RECOGNITION (ICPR), 2021, : 5436 - 5443
  • [37] LD-Net: An Efficient Lightweight Denoising Model Based on Convolutional Neural Network
    Le, Trung-Hieu
    Lin, Po-Hsiung
    Huang, Shih-Chia
    IEEE OPEN JOURNAL OF THE COMPUTER SOCIETY, 2020, 1 : 173 - 181
  • [38] Cross-site Epileptic Seizure Detection Using Convolutional Neural Networks
    Currey, Danielle
    Hsu, David
    Ahmed, Raheel
    Venkataraman, Archana
    Craley, Jeff
    2021 55TH ANNUAL CONFERENCE ON INFORMATION SCIENCES AND SYSTEMS (CISS), 2021,
  • [39] Lightweight and efficient octave convolutional neural network for fire recognition
    Ayala, Angel
    Lima, Estanislau
    Fernandes, Bruno
    Bezerra, Byron L. D.
    Cruz, Francisco
    2019 IEEE LATIN AMERICAN CONFERENCE ON COMPUTATIONAL INTELLIGENCE (LA-CCI), 2019, : 87 - 92
  • [40] Convolutional Transformer Networks for Epileptic Seizure Detection
    Ke, Nan
    Lin, Tong
    Lin, Zhouchen
    Zhou, Xiao-Hua
    Ji, Taoyun
    PROCEEDINGS OF THE 31ST ACM INTERNATIONAL CONFERENCE ON INFORMATION AND KNOWLEDGE MANAGEMENT, CIKM 2022, 2022, : 4109 - 4113