Multiple graph fusion based on Riemannian geometry for motor imagery classification

Session-independent subject-adaptive mental imagery BCI using selective filter-bank adaptive Riemannian features

The brain-computer interfaces (BCIs) facilitate the users to exploit information encoded in neural signals, specifically electroencephalogram (EEG), to control devices and for neural rehabilitation. Mental imagery (MI)-driven BCI predicts the user's pre-meditated mental objectives, which could be deployed as command signals. This paper presents a novel learning-based framework for classifying MI tasks using EEG-based BCI. In particular, our work focuses on the variation in inter-session data and the extraction of multi-spectral user-tailored features for robust performance. Thus, the goal is to create a calibration-free subject-adaptive learning framework for various mental imagery tasks not restricted to motor imagery alone. In this regard, critical spectral bands and the best temporal window are first selected from the EEG training trials of the subject based on the Riemannian user learning distance metric (Dscore) that checks for distinct and stable patterns. The filtered covariance matrices of the EEG trials in each spectral band are then transformed towards a reference covariance matrix using the Riemannian transfer learning, enabling the different sessions to be comparable. The evaluation of our proposed Selective Time-window and Multi-scale Filter-Bank with Adaptive Riemannian (STFB-AR) features on four public datasets, including disabled subjects, showed around 15% and 8% improvement in mean accuracy over baseline and fixed filter-bank models, respectively.

Riemannian transfer learning based on log-Euclidean metric for EEG classification

Introduction

Brain computer interfaces (BCI), which establish a direct interaction between the brain and the external device bypassing peripheral nerves, is one of the hot research areas. How to effectively convert brain intentions into instructions for controlling external devices in real-time remains a key issue that needs to be addressed in brain computer interfaces. The Riemannian geometry-based methods have achieved competitive results in decoding EEG signals. However, current Riemannian classifiers tend to overlook changes in data distribution, resulting in degenerated classification performance in cross-session and/or cross subject scenarios.

Methods

This paper proposes a brain signal decoding method based on Riemannian transfer learning, fully considering the drift of the data distribution. Two Riemannian transfer learning methods based log-Euclidean metric are developed, such that historical data (source domain) can be used to aid the training of the Riemannian decoder for the current task, or data from other subjects can be used to boost the training of the decoder for the target subject.

Results

The proposed methods were verified on BCI competition III, IIIa, and IV 2a datasets. Compared with the baseline that without transfer learning, the proposed algorithm demonstrates superior classification performance. In contrast to the Riemann transfer learning method based on the affine invariant Riemannian metric, the proposed method obtained comparable classification performance, but is much more computationally efficient.

Discussion

With the help of proposed transfer learning method, the Riemannian classifier obtained competitive performance to existing methods in the literature. More importantly, the transfer learning process is unsupervised and time-efficient, possessing potential for online learning scenarios.