1
|
Ru Y, An G, Wei Z, Chen H. Epilepsy detection based on multi-head self-attention mechanism. PLoS One 2024; 19:e0305166. [PMID: 38861543 PMCID: PMC11166279 DOI: 10.1371/journal.pone.0305166] [Citation(s) in RCA: 0] [Impact Index Per Article: 0] [Reference Citation Analysis] [Abstract] [MESH Headings] [Grants] [Track Full Text] [Journal Information] [Subscribe] [Scholar Register] [Received: 02/21/2024] [Accepted: 05/24/2024] [Indexed: 06/13/2024] Open
Abstract
CNN has demonstrated remarkable performance in EEG signal detection, yet it still faces limitations in terms of global perception. Additionally, due to individual differences in EEG signals, the generalization ability of epilepsy detection models is week. To address this issue, this paper presents a cross-patient epilepsy detection method utilizing a multi-head self-attention mechanism. This method first utilizes Short-Time Fourier Transform (STFT) to transform the original EEG signals into time-frequency features, then models local information using Convolutional Neural Network (CNN), subsequently captures global dependency relationships between features using the multi-head self-attention mechanism of Transformer, and finally performs epilepsy detection using these features. Meanwhile, this model employs a light multi-head attention mechanism module with an alternating structure, which can comprehensively extract multi-scale features while significantly reducing computational costs. Experimental results on the CHB-MIT dataset show that the proposed model achieves accuracy, sensitivity, specificity, F1 score, and AUC of 92.89%, 96.17%, 92.99%, 94.41%, and 96.77%, respectively. Compared to the existing methods, the method proposed in this paper obtains better performance along with better generalization.
Collapse
Affiliation(s)
- Yandong Ru
- Key Laboratory of Oceanographic Big Data Mining & Application of Zhejiang Province, Zhoushan, China
- School of Information Engineering, Zhejiang Ocean University, Zhoushan, China
| | - Gaoyang An
- Heilongjiang University of Science and Technology, Harbin, China
| | - Zheng Wei
- Heilongjiang University of Science and Technology, Harbin, China
| | - Hongming Chen
- Key Laboratory of Oceanographic Big Data Mining & Application of Zhejiang Province, Zhoushan, China
- School of Information Engineering, Zhejiang Ocean University, Zhoushan, China
| |
Collapse
|
2
|
Ali E, Angelova M, Karmakar C. Epileptic seizure detection using CHB-MIT dataset: The overlooked perspectives. ROYAL SOCIETY OPEN SCIENCE 2024; 11:230601. [PMID: 39076791 PMCID: PMC11286169 DOI: 10.1098/rsos.230601] [Citation(s) in RCA: 0] [Impact Index Per Article: 0] [Reference Citation Analysis] [Abstract] [Key Words] [Track Full Text] [Figures] [Subscribe] [Scholar Register] [Received: 05/19/2023] [Revised: 11/23/2023] [Accepted: 03/28/2024] [Indexed: 07/31/2024]
Abstract
Epilepsy is a life-threatening neurological condition. Manual detection of epileptic seizures (ES) is laborious and burdensome. Machine learning techniques applied to electroencephalography (EEG) signals are widely used for automatic seizure detection. Some key factors are worth considering for the real-world applicability of such systems: (i) continuous EEG data typically has a higher class imbalance; (ii) higher variability across subjects is present in physiological signals such as EEG; and (iii) seizure event detection is more practical than random segment detection. Most prior studies failed to address these crucial factors altogether for seizure detection. In this study, we intend to investigate a generalized cross-subject seizure event detection system using the continuous EEG signals from the CHB-MIT dataset that considers all these overlooked aspects. A 5-second non-overlapping window is used to extract 92 features from 22 EEG channels; however, the most significant 32 features from each channel are used in experimentation. Seizure classification is done using a Random Forest (RF) classifier for segment detection, followed by a post-processing method used for event detection. Adopting all the above-mentioned essential aspects, the proposed event detection system achieved 72.63% and 75.34% sensitivity for subject-wise 5-fold and leave-one-out analyses, respectively. This study presents the real-world scenario for ES event detectors and furthers the understanding of such detection systems.
Collapse
Affiliation(s)
- Emran Ali
- School of Information Technology, Deakin University, Melbourne Burwood Campus, Melbourne, Victoria3125, Australia
| | - Maia Angelova
- School of Information Technology, Deakin University, Melbourne Burwood Campus, Melbourne, Victoria3125, Australia
- Aston Digital Futures Institute, EPS, Aston University, Birmingham, UK
- Institute of Biophysics and Biomedical Engineering, Bulgarian Academy of Sciences, Sofia, Bulgaria
| | - Chandan Karmakar
- School of Information Technology, Deakin University, Melbourne Burwood Campus, Melbourne, Victoria3125, Australia
| |
Collapse
|
3
|
Zhang Z, Xiao M, Ji T, Jiang Y, Lin T, Zhou X, Lin Z. Efficient and generalizable cross-patient epileptic seizure detection through a spiking neural network. Front Neurosci 2024; 17:1303564. [PMID: 38268711 PMCID: PMC10805904 DOI: 10.3389/fnins.2023.1303564] [Citation(s) in RCA: 0] [Impact Index Per Article: 0] [Reference Citation Analysis] [Abstract] [Key Words] [Track Full Text] [Figures] [Journal Information] [Subscribe] [Scholar Register] [Received: 09/28/2023] [Accepted: 12/18/2023] [Indexed: 01/26/2024] Open
Abstract
Introduction Epilepsy is a global chronic disease that brings pain and inconvenience to patients, and an electroencephalogram (EEG) is the main analytical tool. For clinical aid that can be applied to any patient, an automatic cross-patient epilepsy seizure detection algorithm is of great significance. Spiking neural networks (SNNs) are modeled on biological neurons and are energy-efficient on neuromorphic hardware, which can be expected to better handle brain signals and benefit real-world, low-power applications. However, automatic epilepsy seizure detection rarely considers SNNs. Methods In this article, we have explored SNNs for cross-patient seizure detection and discovered that SNNs can achieve comparable state-of-the-art performance or a performance that is even better than artificial neural networks (ANNs). We propose an EEG-based spiking neural network (EESNN) with a recurrent spiking convolution structure, which may better take advantage of temporal and biological characteristics in EEG signals. Results We extensively evaluate the performance of different SNN structures, training methods, and time settings, which builds a solid basis for understanding and evaluation of SNNs in seizure detection. Moreover, we show that our EESNN model can achieve energy reduction by several orders of magnitude compared with ANNs according to the theoretical estimation. Discussion These results show the potential for building high-performance, low-power neuromorphic systems for seizure detection and also broaden real-world application scenarios of SNNs.
Collapse
Affiliation(s)
- Zongpeng Zhang
- Department of Biostatistics, School of Public Health, Peking University, Beijing, China
| | - Mingqing Xiao
- National Key Lab of General AI, School of Intelligence Science and Technology, Peking University, Beijing, China
| | - Taoyun Ji
- Department of Pediatrics, Peking University First Hospital, Beijing, China
| | - Yuwu Jiang
- Department of Pediatrics, Peking University First Hospital, Beijing, China
| | - Tong Lin
- National Key Lab of General AI, School of Intelligence Science and Technology, Peking University, Beijing, China
| | - Xiaohua Zhou
- Department of Biostatistics, School of Public Health, Peking University, Beijing, China
- Beijing International Center for Mathematical Research, Peking University, Beijing, China
- Peking University Chongqing Institute for Big Data, Chongqing, China
| | - Zhouchen Lin
- National Key Lab of General AI, School of Intelligence Science and Technology, Peking University, Beijing, China
- Institute for Artificial Intelligence, Peking University, Beijing, China
| |
Collapse
|
4
|
Si X, Yang Z, Zhang X, Sun Y, Jin W, Wang L, Yin S, Ming D. Patient-independent seizure detection based on long-term iEEG and a novel lightweight CNN. J Neural Eng 2023; 20. [PMID: 36626831 DOI: 10.1088/1741-2552/acb1d9] [Citation(s) in RCA: 0] [Impact Index Per Article: 0] [Reference Citation Analysis] [Abstract] [Key Words] [Track Full Text] [Journal Information] [Subscribe] [Scholar Register] [Received: 05/04/2022] [Accepted: 01/10/2023] [Indexed: 01/11/2023]
Abstract
Objective.Patient-dependent seizure detection based on intracranial electroencephalography (iEEG) has made significant progress. However, due to the difference in the locations and number of iEEG electrodes used for each patient, patient-independent seizure detection based on iEEG has not been carried out. Additionally, current seizure detection algorithms based on deep learning have outperformed traditional machine learning algorithms in many performance metrics. However, they still have shortcomings of large memory footprints and slow inference speed.Approach.To solve the above problems of the current study, we propose a novel lightweight convolutional neural network model combining the Convolutional Block Attention Module (CBAM). Its performance for patient-independent seizure detection is evaluated on two long-term continuous iEEG datasets: SWEC-ETHZ and TJU-HH. Finally, we reproduce four other patient-independent methods to compare with our method and calculate the memory footprints and inference speed for all methods.Main results.Our method achieves 83.81% sensitivity (SEN) and 85.4% specificity (SPE) on the SWEC-ETHZ dataset and 86.63% SEN and 92.21% SPE on the TJU-HH dataset. In particular, it takes only 11 ms to infer 10 min iEEG (128 channels), and its memory footprint is only 22 kB. Compared to baseline methods, our method not only achieves better patient-independent seizure detection performance but also has a smaller memory footprint and faster inference speed.Significance.To our knowledge, this is the first iEEG-based patient-independent seizure detection study. This facilitates the application of seizure detection algorithms to the future clinic.
Collapse
Affiliation(s)
- Xiaopeng Si
- Academy of Medical Engineering and Translational Medicine, Tianjin University, Tianjin 300072, People's Republic of China.,Tianjin Key Laboratory of Brain Science and Neural Engineering, Tianjin University, Tianjin 300072, People's Republic of China
| | - Zhuobin Yang
- Academy of Medical Engineering and Translational Medicine, Tianjin University, Tianjin 300072, People's Republic of China.,Tianjin Key Laboratory of Brain Science and Neural Engineering, Tianjin University, Tianjin 300072, People's Republic of China
| | - Xingjian Zhang
- Academy of Medical Engineering and Translational Medicine, Tianjin University, Tianjin 300072, People's Republic of China.,Tianjin Key Laboratory of Brain Science and Neural Engineering, Tianjin University, Tianjin 300072, People's Republic of China
| | - Yulin Sun
- Academy of Medical Engineering and Translational Medicine, Tianjin University, Tianjin 300072, People's Republic of China.,Tianjin Key Laboratory of Brain Science and Neural Engineering, Tianjin University, Tianjin 300072, People's Republic of China
| | - Weipeng Jin
- Department of Neurosurgery, Huanhu Hospital, Tianjin University, Tianjin 300072, People's Republic of China
| | - Le Wang
- Department of Neurosurgery, Huanhu Hospital, Tianjin University, Tianjin 300072, People's Republic of China
| | - Shaoya Yin
- Department of Neurosurgery, Huanhu Hospital, Tianjin University, Tianjin 300072, People's Republic of China
| | - Dong Ming
- Academy of Medical Engineering and Translational Medicine, Tianjin University, Tianjin 300072, People's Republic of China.,Tianjin Key Laboratory of Brain Science and Neural Engineering, Tianjin University, Tianjin 300072, People's Republic of China
| |
Collapse
|