Skip to content

This code repository collects the source code of the representative deep learning-based MI-EEG models and runs a leaderboard to fairly compare these models.

License

Notifications You must be signed in to change notification settings

Henrywang621/DL-based-MI-EEG-models

Repository files navigation

A Code Repository for Deep Learning (DL)-Based Motor Imagery (MI) EEG Classification

This code repository collects the available source code of the representative DL-based models for classifying MI-EEG signals and runs a leaderboard table to fairly compare these models. The provided shell scripts can conveniently evaluate representative models on public MI-EEG datasets. This repository aims to help researchers learn about current state-of-the-art models and evaluate their proposed models quickly. More summarization and discussions about the representative models can be found in our survey paper (Link). This repository is updated regularly to contain the latest available MI-EEG decoding models.

Table of contents

Installation

git clone https://github.com/Henrywang621/DL-based-MI-EEG-models.git
cd DL-based-MI-EEG-models
conda env create -f tf-gpu.yml
conda env create -f torch37b6.yml
conda env create -f torch37c.yml
wget https://repo.anaconda.com/archive/Anaconda3-2021.05-Linux-x86_64.sh
bash Anaconda3-2021.05-Linux-x86_64.sh
conda init
source ~/.bashrc

Usage

If you want to evaluate the collected models on the BCI IV 2a dataset, please use the commands below.

cd BCIIV2a_CrossSubjs
chmod +x ./train.sh
sh train.sh

Representative Models and Their Corresponding Source Code

methods title author year source code
Mixed LSTM/1DConv Learning Representations from EEG with Deep Recurrent-Convolutional Neural Networks. [Paper] Bashivan et al. 2016 Code
Shallow ConvNet Deep learning with convolutional neural networks for EEG decoding and visualization. [Paper] Tibor et al. 2017 Code
Deep ConvNet Deep learning with convolutional neural networks for EEG decoding and visualization. [Paper] Tibor et al. 2017 Code
EEGNet EEGNet: a compact convolutional neural network for EEG-based brain–computer interfaces. [Paper] Lawhern et al. 2018 Code
An end-to-end model An end-to-end deep learning approach to MI-EEG signal classification for BCIs. [Paper] Dose et al. 2018 Code
Cascade model Cascade and Parallel Convolutional Recurrent Neural Networks on EEG-based Intention Recognition for Brain Computer Interface. [Paper] Zhang et al. 2018 Code
Parallel model Cascade and Parallel Convolutional Recurrent Neural Networks on EEG-based Intention Recognition for Brain Computer Interface. [Paper] Zhang et al. 2018 Code
A LSTM model Validating Deep Neural Networks for Online Decoding of Motor Imagery Movements from EEG Signals. [Paper] Tayeb et al. 2018 Code
pCNN Validating Deep Neural Networks for Online Decoding of Motor Imagery Movements from EEG Signals. [Paper] Tayeb et al. 2019 Code
EEGNet fusion Fusion Convolutional Neural Network for Cross-Subject EEG Motor Imagery Classification. [Paper] Roots et al. 2020 Code
C-LSTM Data augmentation for self-paced motor imagery classification with C-LSTM. [Paper]] Freer et al. 2020 Code
GCRAM Motor Imagery Classification via Temporal Attention Cues of Graph Embedded EEG Signals. [Paper] Zhang et al. 2020 Code
TS-SEFFNet A Temporal-Spectral-Based Squeeze-and-Excitation Feature Fusion Network for Motor Imagery EEG Decoding. [Paper] Li et al. 2021 Code
MIN2Net MIN2Net: End-to-End Multi-Task Learning for Subject-Independent Motor Imagery EEG Classification. [Paper] Phairot et al. 2022 Code
EEG-Transformer Transformer based Spatial-Temporal Feature Learning for EEG Decoding. [Paper] Song et al. 2022 Code

Citations

@article{WANG2024102738,
title = {An in-depth survey on Deep Learning-based Motor Imagery Electroencephalogram (EEG) classification},
journal = {Artificial Intelligence in Medicine},
volume = {147},
pages = {102738},
year = {2024},
issn = {0933-3657},
doi = {https://doi.org/10.1016/j.artmed.2023.102738},
url = {https://www.sciencedirect.com/science/article/pii/S093336572300252X},
author = {Xianheng Wang and Veronica Liesaputra and Zhaobin Liu and Yi Wang and Zhiyi Huang},

About

This code repository collects the source code of the representative deep learning-based MI-EEG models and runs a leaderboard to fairly compare these models.

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published