Application of Convolutional Neural Networks to Four-Class Motor Imagery Classification Problem

  • Tomas Uktveris Kaunas University of Technology
  • Vacius Jusas Kaunas University of Technology
Keywords: convolutional neural network, motor imagery, feature map, image classification, FFT energy maps

Abstract

In this paper the use of a novel feature extraction method oriented to convolutional neural networks (CNN) is discussed in order to solve four-class motor imagery classification problem. Analysis of viable CNN architectures and their influence on the obtained accuracy for the given task is argued. Furthermore, selection of optimal feature map image dimension, filter sizes and other CNN parameters used for network training is investigated. Methods for generating 2D feature maps from 1D feature vectors are presented for commonly used feature types. Initial results show that CNN can achieve high classification accuracy of 68% for the four-class motor imagery problem with less complex feature extraction techniques. It is shown that optimal accuracy highly depends on feature map dimensions, filter sizes, epoch count and other tunable factors, therefore various fine-tuning techniques must be employed. Experiments show that simple FFT energy map generation techniques are enough to reach the state of the art classification accuracy for common CNN feature map sizes. This work also confirms that CNNs are able to learn a descriptive set of information needed for optimal electroencephalogram (EEG) signal classification.

DOI: http://dx.doi.org/10.5755/j01.itc.46.2.17528

Published
2017-06-06
Section
Articles