在隐私保护下,用少量数据实现癫痫发作的个性化检测。
Federated Few-Shot Learning for Epileptic Seizure Detection Under Privacy Constraints
- 分阶段联邦少样本学习,避免数据集中化。
- 仅用5个标注片段,患者级模型准确率达77%。
- 适合医疗数据分散、隐私严格的场景使用。
许多深度学习方法已用于基于脑电图(EEG)的癫痫发作检测,但大多依赖大规模集中标注数据集。临床实践中,EEG数据稀少,患者数据分散于各机构,且受严格隐私法规限制,禁止数据汇聚。因此,在真实医疗环境中构建可用的AI癫痫检测模型仍具挑战。为此,我们提出一种两阶段联邦少样本学习(FFSL)框架,实现个性化EEG癫痫发作检测。方法在包含六个癫痫事件类别的TUH Event Corpus上训练与评估。第一阶段:在非独立同分布的模拟医院站点间,通过联邦学习微调预训练生物信号变换器(BIOT),实现无数据集中化的共享表征学习。第二阶段:联邦少样本个性化将分类器适配至每位患者,仅需五个标注的EEG片段,保留发作特异性信息并共享跨站点知识。联邦微调阶段平衡准确率为0.43(集中式为0.52),Cohen's kappa为0.42(0.49),加权F1为0.69(0.74)。在FFSL阶段,四个站点的客户端模型平均平衡准确率达0.77,Cohen's kappa为0.62,加权F1为0.73,覆盖异质事件分布。结果表明,FFSL可在数据有限与隐私约束下有效支持患者自适应的癫痫发作检测。
原文摘要 · Abstract (English)
Many deep learning approaches have been developed for EEG-based seizure detection; however, most rely on access to large centralized annotated datasets. In clinical practice, EEG data are often scarce, patient-specific distributed across institutions, and governed by strict privacy regulations that prohibit data pooling. As a result, creating usable AI-based seizure detection models remains challenging in real-world medical settings. To address these constraints, we propose a two-stage federated few-shot learning (FFSL) framework for personalized EEG-based seizure detection. The method is trained and evaluated on the TUH Event Corpus, which includes six EEG event classes. In Stage 1, a pretrained biosignal transformer (BIOT) is fine-tuned across non-IID simulated hospital sites using federated learning, enabling shared representation learning without centralizing EEG recordings. In Stage 2, federated few-shot personalization adapts the classifier to each patient using only five labeled EEG segments, retaining seizure-specific information while still benefiting from cross-site knowledge. Federated fine-tuning achieved a balanced accuracy of 0.43 (centralized: 0.52), Cohen's kappa of 0.42 (0.49), and weighted F1 of 0.69 (0.74). In the FFSL stage, client-specific models reached an average balanced accuracy of 0.77, Cohen's kappa of 0.62, and weighted F1 of 0.73 across four sites with heterogeneous event distributions. These results suggest that FFSL can support effective patient-adaptive seizure detection under realistic data-availability and privacy constraints.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。