SPDIM: Source-Free Unsupervised Conditional and Label Shift Adaptation in EEG

Shanglin Li, Motoaki Kawanabe, Reinmar J Kobler

Keywords: geometric deep learning, transfer learning, source-free adaptation, electroencephalography, neurology, brain-computer interfaces

Abstract:

The non-stationary nature of electroencephalography (EEG) introduces distribution shifts across domains (e.g., days and subjects), posing a significant challenge to EEG-based neurotechnology generalization.
Without labeled calibration data for target domains, the problem is a source-free unsupervised domain adaptation (SFUDA) problem.
For scenarios with constant label distribution, Riemannian geometry-aware statistical alignment frameworks on the symmetric positive definite (SPD) manifold are considered state-of-the-art.
However, many practical scenarios, including EEG-based sleep staging, exhibit label shifts.
Here, we propose a geometric deep learning framework for SFUDA problems under specific distribution shifts, including label shifts.
We introduce a novel, realistic generative model and show that prior Riemannian statistical alignment methods on the SPD manifold can compensate for specific marginal and conditional distribution shifts but hurt generalization under label shifts.
As a remedy, we propose a parameter-efficient manifold optimization strategy termed SPDIM.
SPDIM uses the information maximization principle to learn a single SPD-manifold-constrained parameter per target domain.
In simulations, we demonstrate that SPDIM can compensate for the shifts under our generative model.
Moreover, using public EEG-based brain-computer interface and sleep staging datasets, we show that SPDIM outperforms prior approaches.

Primary Area: applications to neuroscience & cognitive science

Code Of Ethics: I acknowledge that I and all co-authors of this work have read and commit to adhering to the ICLR Code of Ethics.

Submission Guidelines: I certify that this submission complies with the submission instructions as described on https://iclr.cc/Conferences/2025/AuthorGuide.

Anonymous Url: I certify that there is no URL (e.g., github page) that could be used to find authors’ identity.

No Acknowledgement Section: I certify that there is no acknowledgement section in this submission for double blind review.

Submission Number: 8425