arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2402.09438eess.SPcs.LG

基于EEG的运动想象分类的跨被试深度架构

Subject-Independent Deep Architecture for EEG-based Motor Imagery Classification

University of Rochester(罗切斯特大学);Goergen Institute for Data Science, University of Rochester(罗切斯特大学戈尔根数据科学学院)

Shadi Sartipi, Mujdat Cetin

AI总结:

提出一种跨被试半监督深度架构SSDA,结合柱状时空自编码器提取特征与中心损失优化分类,在有限标注下显著提升基于EEG的运动想象分类性能。

摘要:

基于脑电信号(EEG)的运动想象(MI)分类是非侵入式脑机接口(BCI)系统中广泛使用的技术。由于EEG记录存在被试间的异质性和标注数据不足,设计一种能在标注样本有限的情况下独立于被试进行MI分类的分类器是理想的。为克服这些限制,我们提出了一种新颖的跨被试半监督深度架构(SSDA)。该SSDA包含两部分:无监督部分和有监督部分。训练集包含来自多个被试的标注和未标注数据样本。首先,无监督部分即柱状时空自编码器(CST-AE),通过最大化原始数据与重建数据之间的相似性,从所有训练样本中提取潜在特征。采用降维方法在保持表征判别能力的同时降低其维度。其次,有监督部分利用无监督部分获取的潜在特征,基于标注训练样本学习分类器。此外,我们在有监督部分采用中心损失,以最小化每个类别中各点到其中心的嵌入空间距离。模型以端到端方式优化网络的两个部分。SSDA的性能在训练阶段未被模型见过的测试被试上进行评估。为评估性能,我们使用了两个基准的基于EEG的MI任务数据集。结果表明,SSDA优于现有最先进的方法,且少量标注训练样本即可实现强大的分类性能。

英文摘要:

Motor imagery (MI) classification based on electroencephalogram (EEG) is a widely-used technique in non-invasive brain-computer interface (BCI) systems. Since EEG recordings suffer from heterogeneity across subjects and labeled data insufficiency, designing a classifier that performs the MI independently from the subject with limited labeled samples would be desirable. To overcome these limitations, we propose a novel subject-independent semi-supervised deep architecture (SSDA). The proposed SSDA consists of two parts: an unsupervised and a supervised element. The training set contains both labeled and unlabeled data samples from multiple subjects. First, the unsupervised part, known as the columnar spatiotemporal auto-encoder (CST-AE), extracts latent features from all the training samples by maximizing the similarity between the original and reconstructed data. A dimensional scaling approach is employed to reduce the dimensionality of the representations while preserving their discriminability. Second, a supervised part learns a classifier based on the labeled training samples using the latent features acquired in the unsupervised part. Moreover, we employ center loss in the supervised part to minimize the embedding space distance of each point in a class to its center. The model optimizes both parts of the network in an end-to-end fashion. The performance of the proposed SSDA is evaluated on test subjects who were not seen by the model during the training phase. To assess the performance, we use two benchmark EEG-based MI task datasets. The results demonstrate that SSDA outperforms state-of-the-art methods and that a small number of labeled training samples can be sufficient for strong classification performance.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑