arXiv:2510.21841cs.CVcs.HC2025-10

用可学习的小波变换提升脑电运动想象分类的鲁棒性与可解释性。

RatioWaveNet: A Learnable RDWT Front-End for Robust and Interpretable EEG Motor-Imagery Classification

  • 设计可训练的小波变换前端,增强传感器运动节律并抑制噪声。
  • 在最难被识别的受试者上准确率提升最高达2.54个百分点。
  • 适合需要高可靠性的脑机接口场景,尤其关注个体差异大的应用。

基于运动想象的脑机接口将隐含的运动意图转化为可执行指令,但非侵入式脑电图(EEG)解码仍面临非平稳性、信噪比低和个体差异等挑战。本文提出RatioWaveNet,其在强大的时序卷积-注意力主干(TCFormer)前加入可学习的有理稀疏小波变换(RDWT)前端。该方法进行无下采样的多分辨率子带分解,保持时间长度与平移不变性,强化传感器运动节律的同时减轻抖动和轻微伪迹;子带通过轻量级分组一维卷积融合后,输入多核卷积神经网络提取局部时空特征,分组查询注意力编码器捕捉长程上下文,最后由紧凑的时间卷积网络头实现因果时序整合。目标是验证该有原则性的小波前端是否能在传统BCI最易失败的难样本上提升鲁棒性,并在跨种子的同类与异类协议下维持平均性能。在BCI-IV-2a与2b数据集上,五次随机种子实验中,相对于原生Transformer主干,RatioWaveNet在2a上对最差受试者准确率提升+0.17/+0.42百分点(子集依赖/留一法),在2b上提升+1.07/+2.54百分点,且平均表现稳定,计算开销适中。结果表明,简单可学习的小波前端可有效嵌入基于变压器的BCI系统,在不牺牲效率的前提下显著提升最差情况下的可靠性。

原文摘要 · Abstract (English)

Brain-computer interfaces (BCIs) based on motor imagery (MI) translate covert movement intentions into actionable commands, yet reliable decoding from non-invasive EEG remains challenging due to nonstationarity, low SNR, and subject variability. We present RatioWaveNet, which augments a strong temporal CNN-Transformer backbone (TCFormer) with a trainable, Rationally-Dilated Wavelet Transform (RDWT) front end. The RDWT performs an undecimated, multi-resolution subband decomposition that preserves temporal length and shift-invariance, enhancing sensorimotor rhythms while mitigating jitter and mild artifacts; subbands are fused via lightweight grouped 1-D convolutions and passed to a multi-kernel CNN for local temporal-spatial feature extraction, a grouped-query attention encoder for long-range context, and a compact TCN head for causal temporal integration. Our goal is to test whether this principled wavelet front end improves robustness precisely where BCIs typically fail - on the hardest subjects - and whether such gains persist on average across seeds under both intra- and inter-subject protocols. On BCI-IV-2a and BCI-IV-2b, across five seeds, RatioWaveNet improves worst-subject accuracy over the Transformer backbone by +0.17 / +0.42 percentage points (Sub-Dependent / LOSO) on 2a and by +1.07 / +2.54 percentage points on 2b, with consistent average-case gains and modest computational overhead. These results indicate that a simple, trainable wavelet front end is an effective plug-in to strengthen Transformer-based BCIs, improving worst-case reliability without sacrificing efficiency.

脑机接口小波变换运动想象可解释性

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