提出渐进式时空注意力模型,提升快速视觉呈现任务中脑电分类精度
Spatio-Temporal Progressive Attention Model for EEG Classification in Rapid Serial Visual Presentation Task
- 分步学习脑区空间拓扑,逐层聚焦关键电极以减少干扰
- 在新构建的红外小目标刺激脑电数据集上,准确率优于所有对比方法
- 适合关注脑机接口与神经信号建模的研究者
脑电信号作为多维序列数据,其空间与时间依赖性亟待深入挖掘。本文提出一种新型时空渐进注意力模型(STPAM),用于提升快速序列视觉呈现(RSVP)任务中的脑电分类性能。STPAM首先通过三个独立的空间专家逐步学习脑区空间拓扑信息,以最小化无关脑区的干扰:前一专家筛选出相关脑区的脑电电极作为先验知识,指导后续专家逐步聚焦于重要电极信息,强化关键脑区的作用。在此基础上,引入三个时间专家,通过渐进式注意力机制捕捉关键脑电片段的时间依赖性。此外,本文首次构建了基于微弱红外图像与小目标刺激的红外快速视觉呈现脑电数据集(IRED),并在该数据集上进行了大量实验。结果表明,所提STPAM在分类性能上显著优于所有对比方法。
原文摘要 · Abstract (English)
As a type of multi-dimensional sequential data, the spatial and temporal dependencies of electroencephalogram (EEG) signals should be further investigated. Thus, in this paper, we propose a novel spatial-temporal progressive attention model (STPAM) to improve EEG classification in rapid serial visual presentation (RSVP) tasks. STPAM first adopts three distinct spatial experts to learn the spatial topological information of brain regions progressively, which is used to minimize the interference of irrelevant brain regions. Concretely, the former expert filters out EEG electrodes in the relative brain regions to be used as prior knowledge for the next expert, ensuring that the subsequent experts gradually focus their attention on information from significant EEG electrodes. This process strengthens the effect of the important brain regions. Then, based on the above-obtained feature sequence with spatial information, three temporal experts are adopted to capture the temporal dependence by progressively assigning attention to the crucial EEG slices. Except for the above EEG classification method, in this paper, we build a novel Infrared RSVP EEG Dataset (IRED) which is based on dim infrared images with small targets for the first time, and conduct extensive experiments on it. The results show that our STPAM can achieve better performance than all the compared methods.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。