Skip to content

End-to-End Multi-Task Learning for Subject-Independent Motor Imagery EEG Classification (IEEE Transactions on Biomedical Engineering)

License

Notifications You must be signed in to change notification settings

IoBT-VISTEC/MIN2Net

Repository files navigation

End-to-End Multi-Task Learning for Subject-Independent Motor Imagery EEG Classification

Open In Colab Pypi Downloads DOI

Python API and the novel algorithm for motor imagery EEG recognition named MIN2Net. The API benefits BCI researchers ranging from beginners to experts. We demonstrate the examples in using the API for loading benchmark datasets, preprocessing, training, and validation of SOTA models, including MIN2Net. In summary, the API allows the researchers to construct the pipeline for benchmarking the newly proposed models and very recently developed SOTA models.


Getting started

Dependencies

  • Python==3.6.9
  • tensorflow-gpu==2.2.0
  • tensorflow-addons==0.9.1
  • scikit-learn>=0.24.1
  • wget>=3.2
  1. Create conda environment with dependencies
wget https://raw.githubusercontent.com/IoBT-VISTEC/MIN2Net/main/environment.yml
conda env create -f environment.yml
conda activate min2net

Installation:

  1. Using pip
pip install min2net
  1. Using the released python wheel
wget https://github.com/IoBT-VISTEC/MIN2Net/releases/download/v1.0.1/min2net-1.0.1-py3-none-any.whl
pip install min2net-1.0.1-py3-none-any.whl

Tutorial

Open in Colab

Citation

To cited our paper

P. Autthasan et al., "MIN2Net: End-to-End Multi-Task Learning for Subject-Independent Motor Imagery EEG Classification," in IEEE Transactions on Biomedical Engineering, doi: 10.1109/TBME.2021.3137184.

@ARTICLE{9658165,
  author={Autthasan, Phairot and Chaisaen, Rattanaphon and Sudhawiyangkul, Thapanun and Rangpong, Phurin and Kiatthaveephong, Suktipol and Dilokthanakul, Nat and Bhakdisongkhram, Gun and Phan, Huy and Guan, Cuntai and Wilaiprasitporn, Theerawit},
  journal={IEEE Transactions on Biomedical Engineering}, 
  title={MIN2Net: End-to-End Multi-Task Learning for Subject-Independent Motor Imagery EEG Classification}, 
  year={2022},
  volume={69},
  number={6},
  pages={2105-2118},
  doi={10.1109/TBME.2021.3137184}}

License

Copyright © 2021-All rights reserved by INTERFACES (BRAIN lab @ IST, VISTEC, Thailand). Distributed by an Apache License 2.0.