Spatial Imputation Drives Cross-Domain Alignment for EEG Classification
2508.03437v1
cs.CV, cs.AI, 62M10, I.5.1; J.3
2025-08-06
Авторы:
Hongjun Liu, Chao Yao, Yalan Zhang, Xiaokun wang, Xiaojuan Ban
Резюме на русском
**Резюме:**
Классификация сигналов электроэнцефалограммы (EEG) сталкивается с серьезными трудностями, вызванными различиями в конфигурациях электродов, протоколах сбора данных и аппаратуре различных доменов. В этой работе представлена модель **IMAC**, которая решает эту проблему путем выравнивания кросс-доменных данных через задачу импуции (заполнения пропусков) пространственных временных рядов. IMAC выравнивает разнородные конфигурации электродов с помощью метода унификации пространственных представлений, основанного на 3D-to-2D маппинге. Отличительной особенностью является введение channel-dependent масок и реконструкции сигнала в виде задачи импуции с низкого разрешения на высокое. Этот подход эмулирует реальные различия между доменами, такие как отсутствие каналов и временные нестабильности, обеспечивая робастность модели. Дополнительно, IMAC использует дескрипторную структуру для раздельного моделирования пространственной и временной информации, что повышает эффективность и адаптивность. Результаты на 10 открытых наборах данных показывают, что IMAC достигает лучших показателей классификации в кросс-субъектном и кросс-центровом сценариях, превосходя базовые методы на до 35% по метрике целостности и показывая высокую стабильность при реальных и имитированных распределениях.
**(145 слов)**
Abstract
Electroencephalogram (EEG) signal classification faces significant challenges
due to data distribution shifts caused by heterogeneous electrode
configurations, acquisition protocols, and hardware discrepancies across
domains. This paper introduces IMAC, a novel channel-dependent mask and
imputation self-supervised framework that formulates the alignment of
cross-domain EEG data shifts as a spatial time series imputation task. To
address heterogeneous electrode configurations in cross-domain scenarios, IMAC
first standardizes different electrode layouts using a 3D-to-2D positional
unification mapping strategy, establishing unified spatial representations.
Unlike previous mask-based self-supervised representation learning methods,
IMAC introduces spatio-temporal signal alignment. This involves constructing a
channel-dependent mask and reconstruction task framed as a low-to-high
resolution EEG spatial imputation problem. Consequently, this approach
simulates cross-domain variations such as channel omissions and temporal
instabilities, thus enabling the model to leverage the proposed imputer for
robust signal alignment during inference. Furthermore, IMAC incorporates a
disentangled structure that separately models the temporal and spatial
information of the EEG signals separately, reducing computational complexity
while enhancing flexibility and adaptability. Comprehensive evaluations across
10 publicly available EEG datasets demonstrate IMAC's superior performance,
achieving state-of-the-art classification accuracy in both cross-subject and
cross-center validation scenarios. Notably, IMAC shows strong robustness under
both simulated and real-world distribution shifts, surpassing baseline methods
by up to $35$\% in integrity scores while maintaining consistent classification
accuracy.