IoBT-VISTEC / MIN2Net

End-to-End Multi-Task Learning for Subject-Independent Motor Imagery EEG Classification (IEEE Transactions on Biomedical Engineering)
https://min2net.github.io
Apache License 2.0
76 stars 25 forks source link
ae autoencoder bci brain-computer-interface deep-learning deep-metric-learning dml eeg eeg-classification machine-learning multi-task-learning

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

  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
  2. 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.