Entropy-based feature extraction technique in conjunction with wavelet packet transform for multi-mental task classification

Biomed Tech (Berl). 2019 Sep 25;64(5):529-542. doi: 10.1515/bmt-2018-0105.

Abstract

Event-related mental task information collected from electroencephalography (EEG) signals, which are functionally related to different brain areas, possesses complex and non-stationary signal features. It is essential to be able to classify mental task information through the use in brain-computer interface (BCI) applications. This paper proposes a wavelet packet transform (WPT) technique merged with a specific entropy biomarker as a feature extraction tool to classify six mental tasks. First, the data were collected from a healthy control group and the multi-signal information comprised six mental tasks which were decomposed into a number of subspaces spread over a wide frequency spectrum by projecting six different wavelet basis functions. Later, the decomposed subspaces were subjected to three entropy-type statistical measure functions to extract the feature vectors for each mental task to be fed into a backpropagation time-recurrent neural network (BPTT-RNN) model. Cross-validated classification results demonstrated that the model could classify with 85% accuracy through a discrete Meyer basis function coupled with a Renyi entropy biomarker. The classifier model was finally tested in the Simulink platform to demonstrate the Fourier series representation of periodic signals by tracking the harmonic pattern. In order to boost the model performance, ant colony optimization (ACO)-based feature selection method was employed. The overall accuracy increased to 88.98%. The results underlined that the WPT combined with an entropy uncertainty measure methodology is both effective and versatile to discriminate the features of the signal localized in a time-frequency domain.

Keywords: electroencephalography; feature selection; wavelet entropy; wavelet families; wavelet packet transform.

MeSH terms

  • Algorithms
  • Brain / physiology*
  • Brain-Computer Interfaces
  • Electroencephalography / methods*
  • Entropy
  • Humans
  • Neural Networks, Computer
  • Wavelet Analysis