Chen Hao, Jin Ming, Li Zhunan, Fan Cunhang, Li Jinpeng, He Huiguang
HwaMei Hospital, University of Chinese Academy, Ningbo, China.
Center for Pattern Recognition and Intelligent Medicine, Ningbo Institute of Life and Health Industry, University of Chinese Academy of Sciences, Ningbo, China.
Front Neurosci. 2021 Dec 7;15:778488. doi: 10.3389/fnins.2021.778488. eCollection 2021.
As an essential element for the diagnosis and rehabilitation of psychiatric disorders, the electroencephalogram (EEG) based emotion recognition has achieved significant progress due to its high precision and reliability. However, one obstacle to practicality lies in the variability between subjects and sessions. Although several studies have adopted domain adaptation (DA) approaches to tackle this problem, most of them treat multiple EEG data from different subjects and sessions together as a single source domain for transfer, which either fails to satisfy the assumption of domain adaptation that the source has a certain marginal distribution, or increases the difficulty of adaptation. We therefore propose the multi-source marginal distribution adaptation (MS-MDA) for EEG emotion recognition, which takes both domain-invariant and domain-specific features into consideration. First, we assume that different EEG data share the same low-level features, then we construct independent branches for multiple EEG data source domains to adopt one-to-one domain adaptation and extract domain-specific features. Finally, the inference is made by multiple branches. We evaluate our method on SEED and SEED-IV for recognizing three and four emotions, respectively. Experimental results show that the MS-MDA outperforms the comparison methods and state-of-the-art models in cross-session and cross-subject transfer scenarios in our settings. Codes at https://github.com/VoiceBeer/MS-MDA.
作为精神疾病诊断和康复的关键要素,基于脑电图(EEG)的情绪识别因其高精度和可靠性已取得显著进展。然而,实际应用中的一个障碍在于不同受试者和不同测试环节之间的差异。尽管已有多项研究采用域适应(DA)方法来解决这一问题,但大多数研究将来自不同受试者和测试环节的多个EEG数据作为一个单一源域进行转移处理,这要么无法满足域适应中源域具有一定边际分布的假设,要么增加了适应的难度。因此,我们提出了用于EEG情绪识别的多源边际分布适应(MS-MDA)方法,该方法同时考虑了域不变特征和域特定特征。首先,我们假设不同的EEG数据共享相同的低级特征,然后为多个EEG数据源域构建独立分支,以采用一对一的域适应并提取域特定特征。最后,由多个分支进行推理。我们分别在SEED和SEED-IV数据集上评估我们的方法,以识别三种和四种情绪。实验结果表明,在我们的设置下,MS-MDA在跨测试环节和跨受试者转移场景中优于比较方法和当前最先进的模型。代码见https://github.com/VoiceBeer/MS-MDA 。