Affiliation:
1. Graduate School of Science and Engineering, Hosei University, Tokyo 184-8584, Japan
2. Department of Mechanical Engineering, Hosei University, Tokyo 184-8584, Japan
Abstract
In brain–machine interface (BMI) systems, the performance of trained Convolutional Neural Networks (CNNs) is significantly influenced by the quality of the training data. Another issue is the training time of CNNs. This paper introduces a novel approach by combining transfer learning and a Genetic Algorithm (GA) to optimize the training data of CNNs. Transfer learning is implemented across different subjects, and the data chosen by GA aim to improve CNN performance. In addition, the GA-selected data shed light on the similarity in brain activity between subjects. Two datasets are used: (1) the publicly available BCI Competition IV, in which the subjects performed motor imagery (MI) tasks, and (2) the dataset created by healthy subjects of our laboratory performing motor movement (MO) tasks. The experimental results indicate that the brain data selected by the GA improve the recognition accuracy of the target CNN (TCNN) using pre-trained base CNN (BCNN). The improvement in accuracy is 11% and 4% for the BCI Competition IV and our laboratory datasets, respectively. In addition, the GA-selected training data reduce the CNN training time. The performance of the trained CNN, utilizing transfer learning, is tested for real-time control of a robot manipulator.
Subject
Artificial Intelligence,Control and Optimization,Mechanical Engineering
Reference23 articles.
1. Brain–computer interface robotics for hand rehabilitation after stroke: A systematic review;Baniqued;J. Neuroeng. Rehabil.,2021
2. Effectiveness of robotic assisted rehabilitation for mobility and functional ability in adult stroke patients: A systematic review;Lo;JBI Evid. Synth.,2017
3. Ferrero, L., Quiles, V., Ortiz, M., Iáñez, E., and Azorín, J.M. (2021). A BMI based on motor imagery and attention for commanding a lower-limb robotic exoskeleton: A case study. Appl. Sci., 11.
4. Decoding multiclass motor imagery EEG from the same upper limb by combining Riemannian geometry features and partial least squares regression;Chu;J. Neural Eng.,2020
5. BMI control of a third arm for multitasking;Penaloza;Sci. Robot.,2018