提出AGTCNet模型,用图结构+时序建模提升脑电运动想象分类准确率
AGTCNet: A Graph-Temporal Approach for Principled Motor Imagery EEG Classification
- 构建图-时序网络,利用电极拓扑和注意力机制捕捉脑电信号时空特征
- 在多个数据集上实现最高66.82%的跨被试分类准确率,细调后达82.88%
- 模型更小更快,推理速度提升64.65%,适合实际脑机接口部署
脑机接口(BCI)技术利用脑电图(EEG)为运动障碍者提供与环境交互的新方式。尽管前景广阔,但实现跨被试与跨会话的鲁棒系统仍面临挑战,源于个体间神经活动的复杂性与变异性,以及脑电硬件限制。现有方法难以有效建模多通道脑电信号中的复杂时空依赖关系。本文提出一种新的图-时序卷积网络AGTCNet,利用电极拓扑作为先验信息,结合图卷积注意力网络(GCAT),联合学习表达性强的时空脑电表示。该模型显著优于现有方法,在BCI Competition IV Dataset 2a上实现66.82%的跨被试平均准确率,细调后提升至82.88%;在EEG Motor Movement/Imagery Dataset上,4类与2类跨被试分类准确率分别为64.14%和85.22%,细调后分别达到72.13%和90.54%。模型体积减少49.87%,推理速度提升64.65%,具备高实用价值。
原文摘要 · Abstract (English)
Brain-computer interface (BCI) technology utilizing electroencephalography (EEG) marks a transformative innovation, empowering motor-impaired individuals to engage with their environment on equal footing. Despite its promising potential, developing subject-invariant and session-invariant BCI systems remains a significant challenge due to the inherent complexity and variability of neural activity across individuals and over time, compounded by EEG hardware constraints. While prior studies have sought to develop robust BCI systems, existing approaches remain ineffective in capturing the intricate spatiotemporal dependencies within multichannel EEG signals. This study addresses this gap by introducing the attentive graph-temporal convolutional network (AGTCNet), a novel graph-temporal model for motor imagery EEG (MI-EEG) classification. Specifically, AGTCNet leverages the topographic configuration of EEG electrodes as an inductive bias and integrates graph convolutional attention network (GCAT) to jointly learn expressive spatiotemporal EEG representations. The proposed model significantly outperformed existing MI-EEG classifiers, achieving state-of-the-art performance while utilizing a compact architecture, underscoring its effectiveness and practicality for BCI deployment. With a 49.87% reduction in model size, 64.65% faster inference time, and shorter input EEG signal, AGTCNet achieved a moving average accuracy of 66.82% for subject-independent classification on the BCI Competition IV Dataset 2a, which further improved to 82.88% when fine-tuned for subject-specific classification. On the EEG Motor Movement/Imagery Dataset, AGTCNet achieved moving average accuracies of 64.14% and 85.22% for 4-class and 2-class subject-independent classifications, respectively, with further improvements to 72.13% and 90.54% for subject-specific classifications.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。