Eeg Feature Selection Based on Weighted-Normalized Mutual Information for Mental Fatigue Classification

Pengbo Zhang,Xue Wang,Xuanping Li,Peng Dai
DOI: https://doi.org/10.1109/i2mtc.2016.7520423
2016-01-01
Abstract:Analyzing EEG (electroencephalograph) signal is one of the most important methods of mental fatigue classification. Previous studies mainly describe the EEG features they use, however, the process of selecting features has not been synthetically researched. This study proposes the weighted-normalized mutual information feature selection (W-NMIFS) method to select features. W-NMIFS criterion optimizes the estimation of redundancy information in mutual information algorithm. Considering general features which are used in EEG analysis, several important ones are selected by W-NMIFS criterion. In order to guarantee the rigor of the experiment, the source datasets are collected from 10 volunteers using 10-20 EEG system. 3 different mental tasks are designed to introduce mental fatigue in low, medium and high levels. LDA (Linear discriminant analysis) classifier is used to assess mental fatigue and compare different feature selection algorithms. The results of mental fatigue classification show that the chosen features using W-NMIFS criterion are suitable for LDA algorithm. The classification precision is improved compared with mRMR (minimal-redundancy-maximal-relevance), NMIFS (normalized mutual information feature selection) and N-NMIFS (novel normalized mutual information feature selection) algorithm.
What problem does this paper attempt to address?