C2SP-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.1109/tnsre.2023.3235390

IF: 4.9

2023-01-01

IEEE Transactions on Neural Systems and Rehabilitation Engineering

Abstract:Recent developments in brain-machine interface technology have rendered seizure prediction possible. However, the transmission of a large volume of electrophysiological signals between sensors and processing apparatuses and the related computation become two major bottlenecks for seizure prediction systems due to the constrained bandwidth and limited computational resources, especially for power-critical wearable and implantable medical devices. Although many data compression methods can be adopted to compress the signals to reduce communication bandwidth requirement, they require complex compression and reconstruction procedures before the signal can be used for seizure prediction. In this paper, we propose (CSP)-S-2-Net, a framework to jointly solve compression, prediction, and reconstruction without extra computation overhead. The framework consists of a plug-and-play in-sensor compression matrix to reduce transmission bandwidth requirements. The compressed signal can be utilized for seizure prediction without additional reconstruction steps. Reconstruction of the original signal can also be carried out in high fidelity. Compression and classification overhead from the energy consumption perspective, prediction accuracy, sensitivity, false prediction rate, and reconstruction quality of the proposed framework are evaluated using various compression ratios. The experimental results illustrate that our proposed framework is energy efficient and 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.6% in prediction accuracy with a compression ratio ranging from 1/2 to 1/16.