Welcome to the InfoSci Platform
Mental Task Classification Using Deep Transfer Learning with Random Forest Classifier

Mental Task Classification Using Deep Transfer Learning with Random Forest Classifier

Sapna Singh Kshatri, Deepak Singh, Mukesh Kumar Chandrakar, G. R. Sinha
Copyright: © 2022 |Volume: 11 |Issue: 1 |Pages: 17
ISSN: 2161-1610|EISSN: 2161-1629|EISBN13: 9781683182795|DOI: 10.4018/IJBCE.301215
Cite Article Cite Article

MLA

Kshatri, Sapna Singh, et al. "Mental Task Classification Using Deep Transfer Learning with Random Forest Classifier." IJBCE vol.11, no.1 2022: pp.1-17. http://doi.org/10.4018/IJBCE.301215

APA

Kshatri, S. S., Singh, D., Chandrakar, M. K., & Sinha, G. R. (2022). Mental Task Classification Using Deep Transfer Learning with Random Forest Classifier. International Journal of Biomedical and Clinical Engineering (IJBCE), 11(1), 1-17. http://doi.org/10.4018/IJBCE.301215

Chicago

Kshatri, Sapna Singh, et al. "Mental Task Classification Using Deep Transfer Learning with Random Forest Classifier," International Journal of Biomedical and Clinical Engineering (IJBCE) 11, no.1: 1-17. http://doi.org/10.4018/IJBCE.301215

Export Reference

Mendeley
Favorite Full-Issue Download

Abstract

A BCI theoretical idea is to construct an output feature or task for a user using brain signals. These signals are then transmitted to the machine where the required task is performed. In this work, we present a mental task classification model that focuses on the notion of transfer learning and addresses the issues of data scarcity, choice of model selection, and low-performance measure. To decide the optimal network for feature extraction, we used five different pre-trained networks including VGG16, VGG19, ResNet101, ResNet18, and ResNet50. For the classification, the suggested model experiments with three baseline classifiers namely support vector machine, decision tree, and random forest. The model's experimental evaluation is done on the publicly available Keirn and Aunon databases. From the experiment, it is observed that features extracted from the transfer learning models help to identify the five different mental tasks efficiently. The highest average accuracy of 81.25% is attained on ResNet50 based features with a random forest classifier.

Request Access

You do not own this content. Please login to recommend this title to your institution's librarian or purchase it from the IGI Global bookstore.