用频域几何与注意力机制,提升跨人脑电压力检测准确率
I\textsuperscript{2}RiMA: Spectral Riemannian Representation with Temporal Attention for Mental Stress Detection based on EEG Signals

- 在频点级构建协方差矩阵,保留通道几何与频率特征
- 通过频段聚类减少冗余,提升82.78%跨被试检测准确率
- 适合脑电情绪识别、医疗可穿戴设备研发人员参考
跨被试脑电(EEG)压力检测仍具挑战,因压力相关模式具有个体差异性和频率特异性。传统黎曼方法主要在时域建模空间协方差,忽略对高阶认知状态解码至关重要的神经振荡;而标准时间分块常破坏片段间的时间连贯性。为此,我们提出I²RiMA,一种基于频域黎曼流形与时空注意力的脑电压力检测网络。该方法在每个频率点独立构建空间协方差矩阵,并映射至对称正定(SPD)流形切空间,保持通道级几何结构与频率特异性判别线索。进一步引入频率聚类聚合机制,自适应形成与脑电节律对齐的紧凑频段簇,降低冗余。最后,设计跨内-跨片注意力模块,动态融合局部频谱动态与全局时间上下文。在三个数据集上的实验表明,I²RiMA显著优于五种先进基线,最高达82.78%平衡准确率,参数仅160万,计算量3195万次浮点运算。
原文摘要 · Abstract (English)
Cross-subject EEG stress detection remains challenging because discriminative stress-related patterns are both subject-dependent and frequency-specific. Conventional Riemannian methods model spatial covariance mainly in the time domain, overlooking neural oscillations that are critical for high-level cognitive state decoding, while standard temporal tokenization often fragments inter-slice temporal coherence. To address these limitations, we propose \method{}, an Intra-Inter Riemannian Manifold Attention Network for EEG-based stress detection. \method{} constructs spatial covariance matrices independently at each frequency point and maps them to the SPD tangent space, preserving channel-wise geometry together with frequency-specific discriminative cues. It further introduces frequency cluster aggregation to select informative spectral components and reduce redundancy by forming compact, data-driven frequency clusters aligned with EEG rhythms. Finally, an intra-inter slice attention module adaptively integrates local slice-level spectral dynamics and global temporal context across EEG sequences. Experiments on three datasets show that \method{} consistently outperforms five state-of-the-art baselines, achieving up to 82.78\% balanced accuracy while remaining efficient with only 1.60M parameters and 31.95M FLOPs.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。