Adaptive transfer learning for EEG motor imagery classification with deep convolutional neural network
In recent years, deep learning has emerged as a powerful tool for developing Brain-Computer Interface (BCI) systems. However, for deep learning models trained entirely on the data from a specific individual, the performance increase has only been marginal owing to the limited availability of subject...
Saved in:
Main Authors: | , , , |
---|---|
Other Authors: | |
Format: | Article |
Language: | English |
Published: |
2022
|
Subjects: | |
Online Access: | https://hdl.handle.net/10356/160766 |
Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
Institution: | Nanyang Technological University |
Language: | English |
id |
sg-ntu-dr.10356-160766 |
---|---|
record_format |
dspace |
spelling |
sg-ntu-dr.10356-1607662022-08-02T07:22:37Z Adaptive transfer learning for EEG motor imagery classification with deep convolutional neural network Zhang, Kaishuo Robinson, Neethu Lee, Seong-Whan Guan, Cuntai School of Computer Science and Engineering Engineering::Computer science and engineering Brain-Computer Interface Electroencephalography In recent years, deep learning has emerged as a powerful tool for developing Brain-Computer Interface (BCI) systems. However, for deep learning models trained entirely on the data from a specific individual, the performance increase has only been marginal owing to the limited availability of subject-specific data. To overcome this, many transfer-based approaches have been proposed, in which deep networks are trained using pre-existing data from other subjects and evaluated on new target subjects. This mode of transfer learning however faces the challenge of substantial inter-subject variability in brain data. Addressing this, in this paper, we propose 5 schemes for adaptation of a deep convolutional neural network (CNN) based electroencephalography (EEG)-BCI system for decoding hand motor imagery (MI). Each scheme fine-tunes an extensively trained, pre-trained model and adapt it to enhance the evaluation performance on a target subject. We report the highest subject-independent performance with an average (N=54) accuracy of 84.19% (±9.98%) for two-class motor imagery, while the best accuracy on this dataset is 74.15% (±15.83%) in the literature. Further, we obtain a statistically significant improvement (p=0.005) in classification using the proposed adaptation schemes compared to the baseline subject-independent model. This work was partially supported by the RIE2020 Advanced Manufacturing and Engineering (AME) Programmatic Fund, Singapore (No. A20G8b0102). 2022-08-02T07:22:37Z 2022-08-02T07:22:37Z 2021 Journal Article Zhang, K., Robinson, N., Lee, S. & Guan, C. (2021). Adaptive transfer learning for EEG motor imagery classification with deep convolutional neural network. Neural Networks, 136, 1-10. https://dx.doi.org/10.1016/j.neunet.2020.12.013 0893-6080 https://hdl.handle.net/10356/160766 10.1016/j.neunet.2020.12.013 33401114 2-s2.0-85099248925 136 1 10 en No. A20G8b0102 Neural Networks © 2020 Elsevier Ltd. All rights reserved |
institution |
Nanyang Technological University |
building |
NTU Library |
continent |
Asia |
country |
Singapore Singapore |
content_provider |
NTU Library |
collection |
DR-NTU |
language |
English |
topic |
Engineering::Computer science and engineering Brain-Computer Interface Electroencephalography |
spellingShingle |
Engineering::Computer science and engineering Brain-Computer Interface Electroencephalography Zhang, Kaishuo Robinson, Neethu Lee, Seong-Whan Guan, Cuntai Adaptive transfer learning for EEG motor imagery classification with deep convolutional neural network |
description |
In recent years, deep learning has emerged as a powerful tool for developing Brain-Computer Interface (BCI) systems. However, for deep learning models trained entirely on the data from a specific individual, the performance increase has only been marginal owing to the limited availability of subject-specific data. To overcome this, many transfer-based approaches have been proposed, in which deep networks are trained using pre-existing data from other subjects and evaluated on new target subjects. This mode of transfer learning however faces the challenge of substantial inter-subject variability in brain data. Addressing this, in this paper, we propose 5 schemes for adaptation of a deep convolutional neural network (CNN) based electroencephalography (EEG)-BCI system for decoding hand motor imagery (MI). Each scheme fine-tunes an extensively trained, pre-trained model and adapt it to enhance the evaluation performance on a target subject. We report the highest subject-independent performance with an average (N=54) accuracy of 84.19% (±9.98%) for two-class motor imagery, while the best accuracy on this dataset is 74.15% (±15.83%) in the literature. Further, we obtain a statistically significant improvement (p=0.005) in classification using the proposed adaptation schemes compared to the baseline subject-independent model. |
author2 |
School of Computer Science and Engineering |
author_facet |
School of Computer Science and Engineering Zhang, Kaishuo Robinson, Neethu Lee, Seong-Whan Guan, Cuntai |
format |
Article |
author |
Zhang, Kaishuo Robinson, Neethu Lee, Seong-Whan Guan, Cuntai |
author_sort |
Zhang, Kaishuo |
title |
Adaptive transfer learning for EEG motor imagery classification with deep convolutional neural network |
title_short |
Adaptive transfer learning for EEG motor imagery classification with deep convolutional neural network |
title_full |
Adaptive transfer learning for EEG motor imagery classification with deep convolutional neural network |
title_fullStr |
Adaptive transfer learning for EEG motor imagery classification with deep convolutional neural network |
title_full_unstemmed |
Adaptive transfer learning for EEG motor imagery classification with deep convolutional neural network |
title_sort |
adaptive transfer learning for eeg motor imagery classification with deep convolutional neural network |
publishDate |
2022 |
url |
https://hdl.handle.net/10356/160766 |
_version_ |
1743119588157030400 |