Open AccessRealising transfer learning through convolutional neural network and support vector machine for mental task classification
Author(s) -
Singh D.,
Singh S.
Publication year - 2020
Publication title -
electronics letters
Language(s) - English
Resource type - Journals
SCImago Journal Rank - 0.375
H-Index - 146
eISSN - 1350-911X
pISSN - 0013-5194
DOI - 10.1049/el.2020.2632
Subject(s) - artificial intelligence , transfer of learning , computer science , brain–computer interface , support vector machine , machine learning , convolutional neural network , task (project management) , feature selection , artificial neural network , selection (genetic algorithm) , deep learning , electroencephalography , feature extraction , model selection , feature (linguistics) , pattern recognition (psychology) , psychology , engineering , linguistics , philosophy , systems engineering , psychiatry
Electroencephalogram (EEG) measures brainwaves that have been the widely used modality for brain–computer interface (BCI) applications. EEG signal with machine learning has gained substantial success in the BCI. However, the availability of limited training data, appropriate model selection and high false‐positive rates are the challenges that need immediate attention. Therefore, in this Letter, the authors present a mental task classification model based on the notion of transfer learning that addresses the issue of data scarcity, model selection and misclassification ratio. In the framework, the proposed model uses pre‐trained network for the extraction of diverse feature and classify using support vector machine. The authors employed four pre‐trained networks to identify the optimal network for the proposed framework: Vgg16, Vgg19, Resnet18 and Resnet50. The highest classification accuracy of 86.85% (using Resnet50) was achieved using transfer learning. Comparison results showed that convolutional neural network‐based approach outperformed conventional machine learning approaches and hence it can be concluded that the EEG‐based classification of the mental task using transfer learning model could be used in developing a superior model despite the limited data availability.