C$^2$SP-Net: Joint Compression and Classification Network for Epilepsy Seizure Prediction

Di Wu,Yi Shi,Ziyu Wang,Jie Yang,Mohamad Sawan
DOI: https://doi.org/10.48550/arXiv.2110.13674
2022-05-27
Abstract:Recent development in brain-machine interface technology has made seizure prediction possible. However, the communication of large volume of electrophysiological signals between sensors and processing apparatus and related computation become two major bottlenecks for seizure prediction systems due to the constrained bandwidth and limited computation resource, especially for wearable and implantable medical devices. Although compressive sensing (CS) can be adopted to compress the signals to reduce communication bandwidth requirement, it needs a complex reconstruction procedure before the signal can be used for seizure prediction. In this paper, we propose C$^2$SP-Net, to jointly solve compression, prediction, and reconstruction with a single neural network. A plug-and-play in-sensor compression matrix is constructed to reduce transmission bandwidth requirement. The compressed signal can be used for seizure prediction without additional reconstruction steps. Reconstruction of the original signal can also be carried out in high fidelity. Prediction accuracy, sensitivity, false prediction rate, and reconstruction quality of the proposed framework are evaluated under various compression ratios. The experimental results illustrate that our model outperforms the competitive state-of-the-art baselines by a large margin in prediction accuracy. In particular, our proposed method produces an average loss of 0.35 % in prediction accuracy with a compression ratio ranging from 1/2 to 1/16.
Machine Learning,Artificial Intelligence
What problem does this paper attempt to address?
The main problem that this paper attempts to solve is how to efficiently compress, transmit, and predict electrophysiological signals (such as EEG signals) in existing epilepsy seizure prediction systems under the condition of limited communication bandwidth and computational resources, especially in wearable and implantable medical devices. Specifically, although traditional compressed sensing (CS) methods can reduce the communication bandwidth requirements, a complex reconstruction process is required before using the compressed signals for epilepsy prediction, which is impractical in real - time application scenarios. To solve these problems, the authors propose C2SP - Net, a joint compression and classification network, which aims to simultaneously solve the compression, prediction, and reconstruction problems through a single neural network. The following are the specific objectives of this paper: 1. **Joint Compression and Prediction**: Propose a new deep - learning framework that can be directly used for epilepsy prediction while compressing the signal without the need for an additional reconstruction step. 2. **Efficient Signal Compression**: Design a plug - and - play in - sensor compression matrix to reduce the transmission bandwidth requirements and ensure that the compressed signal can still be used for high - precision epilepsy prediction. 3. **High - Quality Signal Reconstruction**: Even at a high compression ratio, it is able to achieve high - quality reconstruction of the original signal to meet other possible application requirements, such as expert manual analysis. 4. **Performance Evaluation**: Evaluate the prediction accuracy, sensitivity, false positive rate, and reconstruction quality of the model at different compression ratios and compare them with other state - of - the - art baseline methods. ### Key Contributions of the Paper 1. **End - to - End Deep - Learning Framework**: Solve the compression, reconstruction, and epilepsy prediction tasks as an overall optimization task, ensuring that the compression matrix is not only suitable for reconstruction but also able to capture features useful for prediction. 2. **Adaptive Reconstruction Module**: A reconstruction module consisting of linear pooling and convolution operations that can adaptively adjust its structure at different compression ratios. 3. **CNN - Based Epilepsy Prediction Network**: A convolutional neural network using the ResNet architecture that can capture the features of EEG signals at different granularity levels. 4. **Extensive Experimental Verification**: Conducted a large number of experiments on multiple open - source data sets, proving that the proposed framework is superior to existing methods in both prediction and reconstruction performance and is stable at different compression ratios. ### Formula Summary - Compression Process: \[ z = C(x) \] where \(z\in\mathbb{R}^{M\times C}\) represents the compressed signal, \(x\in\mathbb{R}^{N\times C}\) represents the original EEG signal, and \(C(\cdot)\) is the compression function. - Reconstruction Process: \[ \hat{x} = R(z) \] where \(\hat{x}\) is the reconstructed signal and \(R(\cdot)\) is the reconstruction function. - Prediction Process: \[ \hat{y} = P(z) \] where \(\hat{y}\) is the prediction result and \(P(\cdot)\) is the prediction function. - Joint Optimization Objective Function: \[ L_{\text{joint}}=L_{\text{pred}}+\lambda\cdot L_{\text{recon}} \] where \(L_{\text{pred}}\) is the prediction loss function, \(L_{\text{recon}}\) is the reconstruction loss function, and \(\lambda\) is a weight parameter that balances the two losses. Through these innovations, C2SP - Net provides a more efficient and reliable epilepsy seizure prediction solution, especially suitable for resource - constrained wearable and implantable devices.