arXiv:2502.13972eess.SPcs.AI2025-02被引 2

融合卷积与注意力机制,提升稳态视觉诱发电位分类准确率

IncepFormerNet: A multi-scale multi-head attention network for SSVEP classification

  • 结合Inception与Transformer,多尺度捕捉脑电信号时序特征
  • 在1秒窗口下,两数据集准确率分别达87.41%和71.97%
  • 适合需要高精度的脑机接口系统研发人员使用

近年来,深度学习模型在脑电图分类任务中表现优异,尤其在基于稳态视觉诱发电位(SSVEP)的脑机接口(BCI)系统中。本文提出一种新模型IncepFormerNet,融合Inception与Transformer架构。该模型通过并行不同尺寸的卷积核,有效提取时间序列数据的多尺度时序信息,精准捕捉SSVEP信号中的细微变化与关键特征。同时引入Transformer的多头注意力机制,增强对全局依赖关系的理解与复杂模式的表征能力。此外,利用滤波器组技术,基于SSVEP数据的频谱特性进行特征提取。为验证模型有效性,我们在两个公开数据集上进行了实验。结果表明,在1.0秒时间窗口下,IncepFormerNet在数据集1上的准确率为87.41%,在数据集2上为71.97%。与其他深度学习模型对比,本方法显著提升分类准确率。源代码已开源:https://github.com/CECNL/SSVEP-DAN。

原文摘要 · Abstract (English)

In recent years, deep learning (DL) models have shown outstanding performance in EEG classification tasks, particularly in Steady-State Visually Evoked Potential(SSVEP)-based Brain-Computer-Interfaces(BCI)systems. DL methods have been successfully applied to SSVEP-BCI. This study proposes a new model called IncepFormerNet, which is a hybrid of the Inception and Transformer architectures. IncepFormerNet adeptly extracts multi-scale temporal information from time series data using parallel convolution kernels of varying sizes, accurately capturing the subtle variations and critical features within SSVEP signals.Furthermore, the model integrates the multi-head attention mechanism from the Transformer architecture, which not only provides insights into global dependencies but also significantly enhances the understanding and representation of complex patterns.Additionally, it takes advantage of filter bank techniques to extract features based on the spectral characteristics of SSVEP data. To validate the effectiveness of the proposed model, we conducted experiments on two public datasets, . The experimental results show that IncepFormerNet achieves an accuracy of 87.41 on Dataset 1 and 71.97 on Dataset 2 using a 1.0-second time window. To further verify the superiority of the proposed model, we compared it with other deep learning models, and the results indicate that our method achieves significantly higher accuracy than the others.The source codes in this work are available at: https://github.com/CECNL/SSVEP-DAN.

脑机接口深度学习信号分类EEG

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。