Semi-Supervised Domain-Adaptive Seizure Prediction Via Feature Alignment and Consistency Regularization.

Deng Liang,Aiping Liu,Yikai Gao,Chang Li,Ruobing Qian,Xun Chen
DOI: https://doi.org/10.1109/tim.2023.3261919
IF: 5.6
2023-01-01
IEEE Transactions on Instrumentation and Measurement
Abstract:The interpatient variability still poses a great challenge for the real-world application of electroencephalogram (EEG)-based seizure prediction, where most previous methods could only work under the patient-specific fashion and fail to generalize across patients. To address this issue, some latest studies applied supervised domain adaptation (SDA), utilizing limited labeled data from the target patient to calibrate the model. However, as the epileptic EEG representation even varies within one single patient, limited target data could hardly cover the data distribution, thus the model still shows poor generalization on the target patient. To this end, we introduce a novel semi-supervised domain adaptive seizure prediction model (SSDA-SPM), using limited labeled target data and extra unlabeled target data for adaptation. SSDA-SPM mainly consists of two unsupervised modules, namely feature alignment (FA) module and consistency regularization (CR) module. The FA module aims to transfer knowledge from existing patients to the target patient coarsely by globally aligning the data distribution between them. Then the CR module further enhances the discriminability on the target patient by pushing the decision boundary into the low-density area. Our proposed method achieves 88.8% sensitivity (Sens), 0.182/h false prediction rate (FPR), and 0.849 area under the receiver operating characteristic curve (AUC) on the CHB-MIT database and 75.7% Sens, 0.165/h FPR, and 0.763 AUC on the Kaggle database. Experimental results demonstrate that our method has provided a promising solution to improve the cross-patient generalization for seizure prediction.
What problem does this paper attempt to address?