将脑电空间模式与深度网络结合,提升运动想象分类精度。
CSP-Net: Common Spatial Pattern Empowered Neural Networks for EEG-Based Motor Imagery Classification
- 用脑电空间模式滤波预处理输入,增强特征区分性
- 在4个公开数据集上优于基础CNN模型,小样本下优势更明显
- 适合脑机接口中训练数据少的场景,融合传统与深度学习
基于脑电的运动想象(MI)分类是非侵入式脑机接口的重要范式。共空间模式(CSP)通过利用不同任务下头皮能量分布差异,在MI分类中广受青睐。卷积神经网络(CNN)凭借强大学习能力也取得显著成果。本文提出两种赋能于CSP的神经网络(CSP-Nets),将知识驱动的CSP滤波器与数据驱动的CNN结合,以提升MI分类性能。CSP-Net-1在CNN前直接加入CSP层,改善输入判别性;CSP-Net-2则用CSP层替换CNN中的某一层。两个模型的CSP层参数均用训练数据设计的CSP滤波器初始化,训练时可固定或微调。在4个公开的MI数据集上的实验表明,两种CSP-Nets在跨被试和同被试分类中均持续优于对应的CNN基线,尤其在训练样本极少时表现更优。本工作展示了将知识驱动的传统机器学习与数据驱动的深度学习融合在脑电脑机接口中的优势。
原文摘要 · Abstract (English)
Electroencephalogram-based motor imagery (MI) classification is an important paradigm of non-invasive brain-computer interfaces. Common spatial pattern (CSP), which exploits different energy distributions on the scalp while performing different MI tasks, is very popular in MI classification. Convolutional neural networks (CNNs) have also achieved great success, due to their powerful learning capabilities. This paper proposes two CSP-empowered neural networks (CSP-Nets), which integrate knowledge-driven CSP filters with data-driven CNNs to enhance the performance in MI classification. CSP-Net-1 directly adds a CSP layer before a CNN to improve the input discriminability. CSP-Net-2 replaces a convolutional layer in CNN with a CSP layer. The CSP layer parameters in both CSP-Nets are initialized with CSP filters designed from the training data. During training, they can either be kept fixed or optimized using gradient descent. Experiments on four public MI datasets demonstrated that the two CSP-Nets consistently improved over their CNN backbones, in both within-subject and cross-subject classifications. They are particularly useful when the number of training samples is very small. Our work demonstrates the advantage of integrating knowledge-driven traditional machine learning with data-driven deep learning in EEG-based brain-computer interfaces.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。