Search papers, labs, and topics across Lattice.
This paper introduces a Riemannian self-attention network called GBWAtt, which leverages the Bures-Wasserstein Metric (BWM) to enhance EEG decoding by effectively capturing local relationships between signals. By addressing the limitations of existing methods that rely on the Affine-Invariant Metric (AIM), particularly its quadratic dependency and inability to manage ill-conditioned SPD matrices, the authors demonstrate that their approach significantly improves performance on EEG tasks. Experimental validation across three benchmarking datasets confirms the robustness and effectiveness of the GBWAtt model in practical applications for brain-computer interfaces.
Riemannian self-attention can revolutionize EEG decoding by overcoming the limitations of traditional metrics, leading to more accurate brain-computer interfaces.
Brain-Computer Interface (BCI) based on electroencephalography (EEG) enables direct interaction between the brain and external environments and has significant applications in assistive technologies, medical rehabilitation, and entertainment. Recently, EEG decoding methods based on Symmetric Positive Definite (SPD) learning have demonstrated superior performance. However, these methods typically employ basic network architectures and do not explicitly capture local relationships between EEG signals. This limitation is problematic for EEG signals due to their inherently low Signal-to-Noise Ratio (SNR). Moreover, most existing Riemannian manifold-based methods are restricted to specific metrics. The most widely used is the Affine-Invariant Metric (AIM). However, it has a quadratic dependency on the SPD matrices and cannot handle ill-conditioned SPD matrices, which hinders the effectiveness of networks. In contrast, the Bures-Wasserstein Metric (BWM) exhibits linear dependence on SPD matrices and demonstrates superior performance for ill conditioning. To overcome these challenges, we propose a Riemannian self-attention network based on the BWM. Additionally, the recently introduced power-deformed generalized Bures-Wasserstein metric reveals a nonlinear relationship between SPD matrices and matrix power deformation. This metric provides a more nuanced representation of the geometric structure of the SPD manifold. Consequently, we extend our model to a learnable version. For simplicity, we refer to it as GBWAtt. Experimental results on three EEG benchmarking datasets validate the robustness and effectiveness of our proposed method. The code is available at https://github.com/jissc/GBWAtt.
