提出时空谱注意力融合模型,提升脑电运动想象分类准确率
SSTAF: Spatial-Spectral-Temporal Attention Fusion Transformer for Motor Imagery Classification
- 设计时空谱三域注意力融合结构,动态捕捉频段、电极位置与时间变化特征
- 在EEGMMIDB和BCI Competition IV-2a数据集上分别达76.83%和68.30%准确率
- 适合脑机接口、神经康复等需要跨被试鲁棒识别的场景
基于脑电(EEG)的脑机接口在神经康复与辅助技术中展现出巨大潜力,但脑电信号的非平稳性及个体间显著差异给跨被试分类模型的构建带来挑战。本文提出一种专为上肢运动想象分类设计的新型空间-谱-时序注意力融合(SSTAF)Transformer。该架构包含谱变压器、空间变压器、一个融合变压器块和分类网络,各模块均引入注意力机制,动态关注频段、电极位置与时间动态中的判别性模式。通过短时傅里叶变换提取时频域特征,增强模型对特征区分能力。在两个公开数据集EEGMMIDB与BCI Competition IV-2a上评估,SSTAF Transformer分别达到76.83%和68.30%的准确率,优于传统CNN架构及部分现有Transformer方法。
原文摘要 · Abstract (English)
Brain-computer interfaces (BCI) in electroencephalography (EEG)-based motor imagery classification offer promising solutions in neurorehabilitation and assistive technologies by enabling communication between the brain and external devices. However, the non-stationary nature of EEG signals and significant inter-subject variability cause substantial challenges for developing robust cross-subject classification models. This paper introduces a novel Spatial-Spectral-Temporal Attention Fusion (SSTAF) Transformer specifically designed for upper-limb motor imagery classification. Our architecture consists of a spectral transformer and a spatial transformer, followed by a transformer block and a classifier network. Each module is integrated with attention mechanisms that dynamically attend to the most discriminative patterns across multiple domains, such as spectral frequencies, spatial electrode locations, and temporal dynamics. The short-time Fourier transform is incorporated to extract features in the time-frequency domain to make it easier for the model to obtain a better feature distinction. We evaluated our SSTAF Transformer model on two publicly available datasets, the EEGMMIDB dataset, and BCI Competition IV-2a. SSTAF Transformer achieves an accuracy of 76.83% and 68.30% in the data sets, respectively, outperforms traditional CNN-based architectures and a few existing transformer-based approaches.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。