arXiv:2504.07987eess.SPcs.LG2025-04被引 4

提出mixEEG框架,提升跨被试脑电联邦学习的泛化能力。

mixEEG: Enhancing EEG Federated Learning for Cross-subject EEG Classification with Tailored mixup

  • 针对脑电特性定制mixup,用平均数据生成伪标签
  • 在癫痫检测与情绪识别任务上显著提升模型迁移性能
  • 适合关注脑电隐私保护与跨被试建模的研究者

由于不同被试间认知过程和生理结构差异,跨被试脑电(EEG)分类面临巨大挑战。现代EEG模型依赖神经网络,需大量数据以实现高性能与泛化能力。然而,脑电数据涉及隐私,限制了医院与机构间的共享,导致多数EEG任务缺乏大规模数据集。联邦学习(FL)允许多个分布式客户端在不直接传输原始数据的情况下协同训练全局模型,从而保护隐私。本文首次在联邦学习框架下研究跨被试EEG分类问题。我们提出一种简单而有效的框架mixEEG:针对脑电模态特性定制原始mixup方法,共享未见被试的无标签平均数据,而非直接共享原始数据,在域适应设置下更好地保护隐私,并提供平均标签作为伪标签。在癫痫检测与情绪识别数据集上进行大量实验,结果表明mixEEG在不同数据集和模型架构下均能一致增强全局模型的跨被试迁移能力。代码已公开于:https://github.com/XuanhaoLiu/mixEEG。

原文摘要 · Abstract (English)

The cross-subject electroencephalography (EEG) classification exhibits great challenges due to the diversity of cognitive processes and physiological structures between different subjects. Modern EEG models are based on neural networks, demanding a large amount of data to achieve high performance and generalizability. However, privacy concerns associated with EEG pose significant limitations to data sharing between different hospitals and institutions, resulting in the lack of large dataset for most EEG tasks. Federated learning (FL) enables multiple decentralized clients to collaboratively train a global model without direct communication of raw data, thus preserving privacy. For the first time, we investigate the cross-subject EEG classification in the FL setting. In this paper, we propose a simple yet effective framework termed mixEEG. Specifically, we tailor the vanilla mixup considering the unique properties of the EEG modality. mixEEG shares the unlabeled averaged data of the unseen subject rather than simply sharing raw data under the domain adaptation setting, thus better preserving privacy and offering an averaged label as pseudo-label. Extensive experiments are conducted on an epilepsy detection and an emotion recognition dataset. The experimental result demonstrates that our mixEEG enhances the transferability of global model for cross-subject EEG classification consistently across different datasets and model architectures. Code is published at: https://github.com/XuanhaoLiu/mixEEG.

脑电分析联邦学习隐私保护迁移学习

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。