Wang Xuepu, Li Bowen, Lin Yanfei, Gao Xiaorong
School of Integrated Circuits and Electronics, Beijing Institute of Technology, Beijing 100081, People's Republic of China.
School of Medicine, Tsinghua University, Beijing 100084, People's Republic of China.
J Neural Eng. 2024 Feb 16;21(1). doi: 10.1088/1741-2552/ad2710.
Many subject-dependent methods were proposed for electroencephalogram (EEG) classification in rapid serial visual presentation (RSVP) task, which required a large amount of data from new subject and were time-consuming to calibrate system. Cross-subject classification can realize calibration reduction or zero calibration. However, cross-subject classification in RSVP task is still a challenge.This study proposed a multi-source domain adaptation based tempo-spatial convolution (MDA-TSC) network for cross-subject RSVP classification. The proposed network consisted of three modules. First, the common feature extraction with multi-scale tempo-spatial convolution was constructed to extract domain-invariant features across all subjects, which could improve generalization of the network. Second, the multi-branch domain-specific feature extraction and alignment was conducted to extract and align domain-specific feature distributions of source and target domains in pairs, which could consider feature distribution differences among source domains. Third, the domain-specific classifier was exploited to optimize the network through loss functions and obtain prediction for the target domain.The proposed network was evaluated on the benchmark RSVP dataset, and the cross-subject classification results showed that the proposed MDA-TSC network outperformed the reference methods. Moreover, the effectiveness of the MDA-TSC network was verified through both ablation studies and visualization.The proposed network could effectively improve cross-subject classification performance in RSVP task, and was helpful to reduce system calibration time.
许多针对快速序列视觉呈现(RSVP)任务中的脑电图(EEG)分类提出的基于受试者的方法,需要来自新受试者的大量数据,并且校准系统很耗时。跨受试者分类可以实现校准减少或零校准。然而,RSVP任务中的跨受试者分类仍然是一个挑战。本研究提出了一种基于多源域自适应的时空卷积(MDA-TSC)网络用于跨受试者RSVP分类。所提出的网络由三个模块组成。首先,构建具有多尺度时空卷积的公共特征提取模块,以提取所有受试者的域不变特征,这可以提高网络的泛化能力。其次,进行多分支域特定特征提取和对齐,以成对提取和对齐源域和目标域的域特定特征分布,这可以考虑源域之间的特征分布差异。第三,利用域特定分类器通过损失函数优化网络并获得目标域的预测。所提出的网络在基准RSVP数据集上进行了评估,跨受试者分类结果表明所提出的MDA-TSC网络优于参考方法。此外,通过消融研究和可视化验证了MDA-TSC网络的有效性。所提出的网络可以有效提高RSVP任务中的跨受试者分类性能,并有助于减少系统校准时间。