Cargando…

The classification of EEG-based winking signals: a transfer learning and random forest pipeline

Brain Computer-Interface (BCI) technology plays a considerable role in the control of rehabilitation or peripheral devices for stroke patients. This is particularly due to their inability to control such devices from their inherent physical limitations after such an attack. More often than not, the...

Descripción completa

Detalles Bibliográficos
Autores principales: Mahendra Kumar, Jothi Letchumy, Rashid, Mamunur, Muazu Musa, Rabiu, Mohd Razman, Mohd Azraai, Sulaiman, Norizam, Jailani, Rozita, P.P. Abdul Majeed, Anwar
Formato: Online Artículo Texto
Lenguaje:English
Publicado: PeerJ Inc. 2021
Materias:
Acceso en línea:https://www.ncbi.nlm.nih.gov/pmc/articles/PMC8019310/
https://www.ncbi.nlm.nih.gov/pubmed/33850667
http://dx.doi.org/10.7717/peerj.11182
_version_ 1783674353091608576
author Mahendra Kumar, Jothi Letchumy
Rashid, Mamunur
Muazu Musa, Rabiu
Mohd Razman, Mohd Azraai
Sulaiman, Norizam
Jailani, Rozita
P.P. Abdul Majeed, Anwar
author_facet Mahendra Kumar, Jothi Letchumy
Rashid, Mamunur
Muazu Musa, Rabiu
Mohd Razman, Mohd Azraai
Sulaiman, Norizam
Jailani, Rozita
P.P. Abdul Majeed, Anwar
author_sort Mahendra Kumar, Jothi Letchumy
collection PubMed
description Brain Computer-Interface (BCI) technology plays a considerable role in the control of rehabilitation or peripheral devices for stroke patients. This is particularly due to their inability to control such devices from their inherent physical limitations after such an attack. More often than not, the control of such devices exploits electroencephalogram (EEG) signals. Nonetheless, it is worth noting that the extraction of the features and the classification of the signals is non-trivial for a successful BCI system. The use of Transfer Learning (TL) has been demonstrated to be a powerful tool in the extraction of essential features. However, the employment of such a method towards BCI applications, particularly in regard to EEG signals, are somewhat limited. The present study aims to evaluate the effectiveness of different TL models in extracting features for the classification of wink-based EEG signals. The extracted features are classified by means of fine-tuned Random Forest (RF) classifier. The raw EEG signals are transformed into a scalogram image via Continuous Wavelet Transform (CWT) before it was fed into the TL models, namely InceptionV3, Inception ResNetV2, Xception and MobileNet. The dataset was divided into training, validation, and test datasets, respectively, via a stratified ratio of 60:20:20. The hyperparameters of the RF models were optimised through the grid search approach, in which the five-fold cross-validation technique was adopted. The optimised RF classifier performance was compared with the conventional TL-based CNN classifier performance. It was demonstrated from the study that the best TL model identified is the Inception ResNetV2 along with an optimised RF pipeline, as it was able to yield a classification accuracy of 100% on both the training and validation dataset. Therefore, it could be established from the study that a comparable classification efficacy is attainable via the Inception ResNetV2 with an optimised RF pipeline. It is envisaged that the implementation of the proposed architecture to a BCI system would potentially facilitate post-stroke patients to lead a better life quality.
format Online
Article
Text
id pubmed-8019310
institution National Center for Biotechnology Information
language English
publishDate 2021
publisher PeerJ Inc.
record_format MEDLINE/PubMed
spelling pubmed-80193102021-04-12 The classification of EEG-based winking signals: a transfer learning and random forest pipeline Mahendra Kumar, Jothi Letchumy Rashid, Mamunur Muazu Musa, Rabiu Mohd Razman, Mohd Azraai Sulaiman, Norizam Jailani, Rozita P.P. Abdul Majeed, Anwar PeerJ Bioinformatics Brain Computer-Interface (BCI) technology plays a considerable role in the control of rehabilitation or peripheral devices for stroke patients. This is particularly due to their inability to control such devices from their inherent physical limitations after such an attack. More often than not, the control of such devices exploits electroencephalogram (EEG) signals. Nonetheless, it is worth noting that the extraction of the features and the classification of the signals is non-trivial for a successful BCI system. The use of Transfer Learning (TL) has been demonstrated to be a powerful tool in the extraction of essential features. However, the employment of such a method towards BCI applications, particularly in regard to EEG signals, are somewhat limited. The present study aims to evaluate the effectiveness of different TL models in extracting features for the classification of wink-based EEG signals. The extracted features are classified by means of fine-tuned Random Forest (RF) classifier. The raw EEG signals are transformed into a scalogram image via Continuous Wavelet Transform (CWT) before it was fed into the TL models, namely InceptionV3, Inception ResNetV2, Xception and MobileNet. The dataset was divided into training, validation, and test datasets, respectively, via a stratified ratio of 60:20:20. The hyperparameters of the RF models were optimised through the grid search approach, in which the five-fold cross-validation technique was adopted. The optimised RF classifier performance was compared with the conventional TL-based CNN classifier performance. It was demonstrated from the study that the best TL model identified is the Inception ResNetV2 along with an optimised RF pipeline, as it was able to yield a classification accuracy of 100% on both the training and validation dataset. Therefore, it could be established from the study that a comparable classification efficacy is attainable via the Inception ResNetV2 with an optimised RF pipeline. It is envisaged that the implementation of the proposed architecture to a BCI system would potentially facilitate post-stroke patients to lead a better life quality. PeerJ Inc. 2021-03-31 /pmc/articles/PMC8019310/ /pubmed/33850667 http://dx.doi.org/10.7717/peerj.11182 Text en © 2021 Mahendra Kumar et al. https://creativecommons.org/licenses/by/4.0/ This is an open access article distributed under the terms of the Creative Commons Attribution License (https://creativecommons.org/licenses/by/4.0/) , which permits unrestricted use, distribution, reproduction and adaptation in any medium and for any purpose provided that it is properly attributed. For attribution, the original author(s), title, publication source (PeerJ) and either DOI or URL of the article must be cited.
spellingShingle Bioinformatics
Mahendra Kumar, Jothi Letchumy
Rashid, Mamunur
Muazu Musa, Rabiu
Mohd Razman, Mohd Azraai
Sulaiman, Norizam
Jailani, Rozita
P.P. Abdul Majeed, Anwar
The classification of EEG-based winking signals: a transfer learning and random forest pipeline
title The classification of EEG-based winking signals: a transfer learning and random forest pipeline
title_full The classification of EEG-based winking signals: a transfer learning and random forest pipeline
title_fullStr The classification of EEG-based winking signals: a transfer learning and random forest pipeline
title_full_unstemmed The classification of EEG-based winking signals: a transfer learning and random forest pipeline
title_short The classification of EEG-based winking signals: a transfer learning and random forest pipeline
title_sort classification of eeg-based winking signals: a transfer learning and random forest pipeline
topic Bioinformatics
url https://www.ncbi.nlm.nih.gov/pmc/articles/PMC8019310/
https://www.ncbi.nlm.nih.gov/pubmed/33850667
http://dx.doi.org/10.7717/peerj.11182
work_keys_str_mv AT mahendrakumarjothiletchumy theclassificationofeegbasedwinkingsignalsatransferlearningandrandomforestpipeline
AT rashidmamunur theclassificationofeegbasedwinkingsignalsatransferlearningandrandomforestpipeline
AT muazumusarabiu theclassificationofeegbasedwinkingsignalsatransferlearningandrandomforestpipeline
AT mohdrazmanmohdazraai theclassificationofeegbasedwinkingsignalsatransferlearningandrandomforestpipeline
AT sulaimannorizam theclassificationofeegbasedwinkingsignalsatransferlearningandrandomforestpipeline
AT jailanirozita theclassificationofeegbasedwinkingsignalsatransferlearningandrandomforestpipeline
AT ppabdulmajeedanwar theclassificationofeegbasedwinkingsignalsatransferlearningandrandomforestpipeline
AT mahendrakumarjothiletchumy classificationofeegbasedwinkingsignalsatransferlearningandrandomforestpipeline
AT rashidmamunur classificationofeegbasedwinkingsignalsatransferlearningandrandomforestpipeline
AT muazumusarabiu classificationofeegbasedwinkingsignalsatransferlearningandrandomforestpipeline
AT mohdrazmanmohdazraai classificationofeegbasedwinkingsignalsatransferlearningandrandomforestpipeline
AT sulaimannorizam classificationofeegbasedwinkingsignalsatransferlearningandrandomforestpipeline
AT jailanirozita classificationofeegbasedwinkingsignalsatransferlearningandrandomforestpipeline
AT ppabdulmajeedanwar classificationofeegbasedwinkingsignalsatransferlearningandrandomforestpipeline