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

• Published in: IEEE transactions on instrumentation and measurement Vol. 72; p. 1
• Main Authors: Liang, Deng, Liu, Aiping, Gao, Yikai, Li, Chang, Qian, Ruobing, Chen, Xun
• Format: Journal Article
• Language: English
• Published: New York IEEE 01.01.2023; The Institute of Electrical and Electronics Engineers, Inc. (IEEE)
• Subjects: Adaptation; Adaptation models; Alignment; Brain modeling; Consistency; consistency regularization; Convulsions & seizures; Data models; Domains; Electroencephalogram (EEG); Electroencephalography; feature alignment; Feature extraction; Knowledge management; Modules; Prediction models; Regularization; seizure prediction; semi-supervised domain adaptation; Training
• DOI: 10.1109/TIM.2023.3261919
• ISSN: 0018-9456

The inter-patient variability still poses a great challenge for the real-world application of 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, 0.182/h false prediction rate (FPR) and 0.849 AUC on the CHB-MIT database and 75.7% sensitivity, 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.