BDAN: Mitigating Temporal Difference Across Electrodes in Cross-Subject Motor Imagery Classification via Generative Bridging Domain

Zhige Chen,Rui Yang,Mengjie Huang,Chengxuan Qin,Zidong Wang
2024-04-16
Abstract:Because of "the non-repeatability of the experiment settings and conditions" and "the variability of brain patterns among subjects", the data distributions across sessions and electrodes are different in cross-subject motor imagery (MI) studies, eventually reducing the performance of the classification model. Systematically summarised based on the existing studies, a novel temporal-electrode data distribution problem is investigated under both intra-subject and inter-subject scenarios in this paper. Based on the presented issue, a novel bridging domain adaptation network (BDAN) is proposed, aiming to minimise the data distribution difference across sessions in the aspect of the electrode, thus improving and enhancing model performance. In the proposed BDAN, deep features of all the EEG data are extracted via a specially designed spatial feature extractor. With the obtained spatio-temporal features, a special generative bridging domain is established, bridging the data from all the subjects across sessions. The difference across sessions and electrodes is then minimized using the customized bridging loss functions, and the known knowledge is automatically transferred through the constructed bridging domain. To show the effectiveness of the proposed BDAN, comparison experiments and ablation studies are conducted on a public EEG dataset. The overall comparison results demonstrate the superior performance of the proposed BDAN compared with the other advanced deep learning and domain adaptation methods.
Human-Computer Interaction,Machine Learning
What problem does this paper attempt to address?
### Problems Addressed by the Paper The paper primarily explores the issue of electrode time differences in cross-subject Motor Imagery (MI) classification and proposes a new method to mitigate this problem. #### Main Issues In cross-subject Motor Imagery classification research, the irreproducibility of experimental settings and conditions, as well as variations in brain patterns among different subjects, lead to differences in data distribution across different sessions and electrodes, ultimately affecting the performance of classification models. Specifically, the paper identifies the following two main reasons: 1. **Irreproducibility of Experimental Settings and Conditions**: Inconsistencies in manual electrode placement, different habits of operators, and changes in the experimental environment (such as conductive media, electrode-skin contact conditions, etc.) can cause variations in electrode impedance, thereby affecting the quality of EEG data and causing differences in data distribution over time and across electrodes. 2. **Variability in Subjects' Brain Patterns**: Differences in brain structure and function among different subjects, including morphological differences, neural conduction speed, and the way neurons are connected. Additionally, long-term data collection can affect subjects' attention, emotional state, and fatigue level, thereby influencing brain activity patterns and leading to changes in data recording. These issues result in significant data distribution differences in Single-Source to Single-Target (STS) tasks, reducing the generalization ability of the model. #### Solution To address the above issues, the authors propose a novel Bridging Domain Adaptation Network (BDAN), which aims to minimize data distribution differences across different sessions and electrodes by generating a bridging domain, thereby improving model performance. - **Specially Designed Spatial Feature Extractor**: Used to extract deep spatial features from all EEG data. - **Generating Bridging Domain**: Establishes a bridging domain based on features extracted from the source and target domains, automatically transferring known spatial-temporal knowledge. - **Custom Bridging Loss Function**: Evaluates data distribution differences through a customized bridging loss function and optimizes the classification model to minimize these differences. The effectiveness and superiority of the proposed method are validated through comparative experiments and ablation studies on public EEG datasets.