针对脑电图标签偏移问题,提出高效几何学习方法SPDIM,无需标注数据即可提升模型泛化能力。
SPDIM: Source-Free Unsupervised Conditional and Label Shift Adaptation in EEG
- 基于信息最大化原理,在对称正定流形上优化单一参数应对分布偏移
- 在睡眠分期与脑机接口数据集上显著优于现有无监督方法
- 适用于无目标域标注的场景,特别适合实际脑电信号跨天/跨人迁移
脑电信号具有非平稳性,导致不同时间与受试者间存在分布偏移,严重制约基于脑电的神经技术泛化能力。当目标域缺乏标注校准数据时,该问题属于源域无关的无监督域适应(SFUDA)。对于标签分布恒定的场景,基于对称正定(SPD)流形的黎曼几何统计对齐方法为当前最优。然而,如脑电睡眠分期等实际场景常出现标签偏移。本文提出一种针对特定分布偏移(包括标签偏移)的几何深度学习框架,并构建一个新颖且现实的生成模型。实验表明,现有黎曼统计对齐方法可缓解边缘与条件分布偏移,但在标签偏移下损害泛化性能。为此,提出参数高效的流形优化策略SPDIM,利用信息最大化原则,为每个目标域学习一个受SPD流形约束的单一参数。仿真验证了其对偏移的有效补偿;在公开的脑机接口与睡眠分期数据集上,SPDIM性能优于已有方法。
原文摘要 · Abstract (English)
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.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。