Genetic algorithm based deep learning model adaptation for improvising the motor imagery classification

ABSTRACT

Deep learning methods have proved a promising performance for electroencephalography-based brain-computer interfaces (EEG-BCI). It is particularly encouraging that a subject-independent model can be trained using a large amount of other subjects’ data. Transfer learning methods such as adaptation or fine-tuning can be used on the pre-trained model to improve the performance. This study examined the influence of fine-tuning on the subject-independent model for EEG-based motor imagery (MI) classification using a genetic algorithm (GA). The proposed method is evaluated on the binary class MI dataset from the Korea University EEG dataset. Results show that the proposed GA-based fine-tuning approach statistically improved the average classification accuracy of the baseline model from 84.46% to 87.29%. More interestingly, our approach shows significant improvement in cases where the performance of the baseline model is poor after fine-tuning using other approaches. Further, layer-wise relevance propagation (LRP) is used to analyze the adapted models to gain a deeper understanding of the neurophysiological explanations underlying the model’s decision.

KEYWORDS:

• Brain-computer interface (BCI)
• fine-tuning
• electroencephalography (EEG)
• motor-imagery (MI)
• convolutional neural network (CNN)

Disclosure statement

No potential conflict of interest was reported by the author(s).

Author contributions

All authors have participated in (a) conception and design, or analysis and interpretation of the data; (b) drafting the article or revising it critically for important intellectual content; and (c) approval of the final version.

Additional information

Funding

The authors declare that no funds, grants, or other support were received during the preparation of this manuscript.