通过跨被试通道交换与对抗学习,提升脑电图疾病分类的泛化能力。
A swap-adversarial framework for improving domain generalization in electrocorticography-based Parkinson's disease classification
- 基于脑图谱进行功能对应通道交换,增强跨被试数据多样性。
- 在跨被试、跨会话、跨数据集测试中均超越所有基线模型。
- 适用于高变异性神经信号场景,尤其适合癫痫与帕金森研究者。
我们提出一种新型交换对抗框架,以缓解皮层脑电图(ECoG)数据中存在的高个体差异与高维低样本问题。该框架集成三部分:鲁棒预处理、跨被试平衡通道交换(ISBCS)用于跨被试数据增强,以及域对抗学习(DAL)以抑制个体特异性偏差。ISBCS方法受生物启发,依据脑图谱仅交换功能对应的通道,降低个体间分布差异;DAL策略促使模型学习任务相关的共享特征。在跨被试、跨会话和跨数据集设置下,通过大量实验验证了该框架的有效性。其在所有设置中持续优于所有基线,尤其在高变异性环境下表现最显著。同时,在公开EEG基准数据集间实现优异跨数据集性能,表明其不仅适用于ECoG,也具备良好的EEG泛化能力。此外,我们构建了首个可复现的ECoG数据集,基于6-羟基多巴胺诱导的大鼠模型长期记录,并标注电刺激前后的神经响应。
原文摘要 · Abstract (English)
We propose a novel swap-adversarial framework that mitigates high inter-subject variability and the high-dimensional low-sample-size problem in electrocorticography (ECoG) data. It achieves robust domain generalization across ECoG and electroencephalography (EEG)-based brain-computer interface datasets. Our framework integrates (1) robust preprocessing, (2) inter-subject balanced channel swap (ISBCS) for cross-subject augmentation, and (3) domain-adversarial learning (DAL) to suppress subject-specific bias. The ISBCS method is a bio-inspired channel swapping strategy that exchanges only functionally corresponding channels across subjects, guided by a brain map, to mitigate inter-subject distribution differences. The DAL strategy encourages the model to learn task-relevant shared features. We validate the effectiveness of this framework through extensive experiments under cross-subject, cross-session, and cross-dataset settings. Our framework consistently outperforms all baselines across all settings, showing the most significant improvements in highly variable environments. It also achieves superior cross-dataset performance between public EEG benchmarks, demonstrating strong generalization capability not only for ECoG but also for EEG data. In addition, we introduce a new ECoG dataset, the first reproducible benchmark, which is constructed from long-term ECoG recordings of 6-hydroxydopamine-induced rat models and annotated with neural responses measured before and after electrical stimulation.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。