arXiv:2607.01279cs.LG2026-07

用频域几何与注意力机制,提升跨人脑电压力检测准确率

I\textsuperscript{2}RiMA: Spectral Riemannian Representation with Temporal Attention for Mental Stress Detection based on EEG Signals

论文配图:I\textsuperscript{2}RiMA: Spectral Riemannian Representation with Temporal Attention for Mental Stress Detection based on EEG Signals
图 1 · 摘自论文原文
  • 在频点级构建协方差矩阵,保留通道几何与频率特征
  • 通过频段聚类减少冗余,提升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 官方产品;中文卡片由大模型生成,请以原文为准。