用图神经网络解码脑电波步态,精度超现有方法。
EEG2GAIT: A Hierarchical Graph Convolutional Network for EEG-based Gait Decoding
- 构建分层图卷积网络,捕捉脑电通道多尺度空间关系。
- 引入时频联合奖励损失,使预测相关系数达0.959。
- 适合脑机接口、下肢康复等应用,数据来自50人实验。
从脑电信号中解码步态动态面临诸多挑战:运动过程的空间依赖性复杂,需精准提取时序与频域特征,且高质量步态脑电数据集稀缺。为此,本文提出EEG2GAIT,一种基于分层图卷积网络(GCN Pyramid)的新型模型,用于捕获脑电通道的多层级空间嵌入。为进一步提升解码性能,引入混合时频奖励损失(HTSR),融合时域、频域及奖励机制损失。同时,构建了新的同步脑电-下肢关节角度数据集GED,涵盖50名参与者在两次实验室访问中的数据。大量实验证明,使用HTSR的EEG2GAIT在GED数据集上取得优异表现:皮尔逊相关系数(r)为0.959,决定系数为0.914,平均绝对误差(MAE)为0.193;在MoBI数据集上同样优于已有方法,对应指标分别为0.779、0.597和4.384。统计分析证实其提升显著。消融实验验证了分层GCN模块与HTSR损失的有效性,显著性分析揭示运动相关脑区在解码中的作用。结果表明,EEG2GAIT在脑机接口,特别是下肢康复与辅助技术方面具有重要应用潜力。
原文摘要 · Abstract (English)
Decoding gait dynamics from EEG signals presents significant challenges due to the complex spatial dependencies of motor processes, the need for accurate temporal and spectral feature extraction, and the scarcity of high-quality gait EEG datasets. To address these issues, we propose EEG2GAIT, a novel hierarchical graph-based model that captures multi-level spatial embeddings of EEG channels using a Hierarchical Graph Convolutional Network (GCN) Pyramid. To further improve decoding performance, we introduce a Hybrid Temporal-Spectral Reward (HTSR) loss function, which integrates time-domain, frequency-domain, and reward-based loss components. In addition, we contribute a new Gait-EEG Dataset (GED), consisting of synchronized EEG and lower-limb joint angle data collected from 50 participants across two laboratory visits. Extensive experiments demonstrate that EEG2GAIT with HTSR achieves superior performance on the GED dataset, reaching a Pearson correlation coefficient (r) of 0.959, a coefficient of determination of 0.914, and a Mean Absolute Error (MAE) of 0.193. On the MoBI dataset, EEG2GAIT likewise consistently outperforms existing methods, achieving an r of 0.779, a coefficient of determination of 0.597, and an MAE of 4.384. Statistical analyses confirm that these improvements are significant compared to all prior models. Ablation studies further validate the contributions of the hierarchical GCN modules and the proposed HTSR loss, while saliency analysis highlights the involvement of motor-related brain regions in decoding tasks. Collectively, these findings underscore EEG2GAIT's potential for advancing brain-computer interface applications, particularly in lower-limb rehabilitation and assistive technologies.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。