This work investigates the efficacy of different deep learning and Riemannian geometry-based classification models in the context of motor imagery (MI) based brain-computer interface (BCI) using electroencephalography (EEG) data. The authors propose an optimal transport theory-based approach using Earth Mover's Distance (EMD) to quantify the comparison of the feature relevance maps generated by these models with the domain knowledge of neuroscience.
The authors implemented three state-of-the-art models: 1) Riemannian geometry-based classifier, 2) EEGNet, and 3) EEG Conformer. They observed that the models with diverse architectures perform significantly better when trained on channels relevant to motor imagery than data-driven channel selection.
The authors used Explainable AI (XAI) techniques, specifically Gradient-weighted Class Activation Mapping (Grad-CAM), to generate feature relevance maps for the EEGNet and EEG Conformer models. They then compared these feature relevance maps with the domain knowledge of motor cortical regions using the proposed EMD-based approach.
The results show that the feature relevance maps from the Riemannian geometry-based classifier are closest to the domain knowledge, followed by EEGConformer and EEGNet. This highlights the necessity for interpretability and incorporating metrics beyond accuracy, underscoring the value of combining domain knowledge and quantifying model interpretations with data-driven approaches in creating reliable and robust Brain-Computer Interfaces (BCIs).
На другой язык
из исходного контента
arxiv.org
Дополнительные вопросы