FedL2T: Personalized Federated Learning with Two-Teacher Distillation for Seizure Prediction
2510.08984v1
cs.LG, cs.NE
2025-10-14
Авторы:
Jionghao Lou, Jian Zhang, Zhongmei Li, Lanlan Chen, Enbo Feng
Abstract
The training of deep learning models in seizure prediction requires large
amounts of Electroencephalogram (EEG) data. However, acquiring sufficient
labeled EEG data is difficult due to annotation costs and privacy constraints.
Federated Learning (FL) enables privacy-preserving collaborative training by
sharing model updates instead of raw data. However, due to the inherent
inter-patient variability in real-world scenarios, existing FL-based seizure
prediction methods struggle to achieve robust performance under heterogeneous
client settings. To address this challenge, we propose FedL2T, a personalized
federated learning framework that leverages a novel two-teacher knowledge
distillation strategy to generate superior personalized models for each client.
Specifically, each client simultaneously learns from a globally aggregated
model and a dynamically assigned peer model, promoting more direct and enriched
knowledge exchange. To ensure reliable knowledge transfer, FedL2T employs an
adaptive multi-level distillation strategy that aligns both prediction outputs
and intermediate feature representations based on task confidence. In addition,
a proximal regularization term is introduced to constrain personalized model
updates, thereby enhancing training stability. Extensive experiments on two EEG
datasets demonstrate that FedL2T consistently outperforms state-of-the-art FL
methods, particularly under low-label conditions. Moreover, FedL2T exhibits
rapid and stable convergence toward optimal performance, thereby reducing the
number of communication rounds and associated overhead. These results
underscore the potential of FedL2T as a reliable and personalized solution for
seizure prediction in privacy-sensitive healthcare scenarios.
Ссылки и действия
Дополнительные ресурсы: