MixNet融合传统与现代方法,提升脑机接口中运动想象的分类精度。
MixNet: Joining Force of Classical and Modern Approaches Toward the Comprehensive Pipeline in Motor Imagery EEG Classification
- 结合频谱-空间特征与多任务学习框架,增强跨被试泛化能力
- 在6个基准数据集上均优于现有最优算法,低密度脑电亦表现优异
- 适合轻量级可穿戴设备,为物联网场景下的便携式脑机接口提供支持
深度学习的进展显著推动了基于运动想象(MI)的脑机接口(BCI)系统发展,提升了对脑电图(EEG)信号的解码能力。然而,多数研究难以在不同被试间识别出具有判别性的模式,限制了分类性能。本文提出MixNet,一种新型分类框架,通过利用MI数据的频谱-空间信号,并采用名为MIN2Net的多任务学习架构实现分类。其中,频谱-空间信号通过滤波组共空间模式(FBCSPs)方法生成。由于多任务学习在各任务间可能存在不同的泛化率和过拟合风险,我们引入自适应梯度混合机制,动态调节多个损失权重,并根据各任务的泛化/过拟合趋势调整学习速度。在六个不同规模的基准数据集上的实验表明,MixNet在被试依赖与被试独立设置下均持续优于所有最先进的算法。此外,在低密度EEG的运动想象分类中,MixNet同样全面超越现有方法,为基于低密度电极布局的轻量化、便携式脑电可穿戴设备等物联网应用提供了重要启示。
原文摘要 · Abstract (English)
Recent advances in deep learning (DL) have significantly impacted motor imagery (MI)-based brain-computer interface (BCI) systems, enhancing the decoding of electroencephalography (EEG) signals. However, most studies struggle to identify discriminative patterns across subjects during MI tasks, limiting MI classification performance. In this article, we propose MixNet, a novel classification framework designed to overcome this limitation by utilizing spectral-spatial signals from MI data, along with a multitask learning architecture named MIN2Net, for classification. Here, the spectral-spatial signals are generated using the filter-bank common spatial patterns (FBCSPs) method on MI data. Since the multitask learning architecture is used for the classification task, the learning in each task may exhibit different generalization rates and potential overfitting across tasks. To address this issue, we implement adaptive gradient blending, simultaneously regulating multiple loss weights and adjusting the learning pace for each task based on its generalization/overfitting tendencies. Experimental results on six benchmark data sets of different data sizes demonstrate that MixNet consistently outperforms all state-of-the-art algorithms in subject-dependent and -independent settings. Finally, the low-density EEG MI classification results show that MixNet outperforms all state-of-the-art algorithms, offering promising implications for Internet of Thing (IoT) applications, such as lightweight and portable EEG wearable devices based on low-density montages.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。