A swap-adversarial framework for improving domain generalization in electrocorticography-based Parkinson's disease classification
이 뉴스, 어떠셨어요?
한 번의 탭으로 반응을 남겨요 · 로그인 불필요
Abstract
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.