A Feature Selection Method for Multilevel Mental Fatigue EEG Classification

Two feature selection approaches for multilevel mental fatigue electroencephalogram (EEG) classification are presented in this paper, in which random forest (RF) is combined with the heuristic initial feature ranking scheme (INIT) or with the recursive feature elimination scheme (RFE). In a "leave-one-proband-out" evaluation strategy, both feature selection approaches are evaluated on the recorded mental fatigue EEG time series data from 12 subjects (each for a 25-h duration) after initial feature extractions. The latter of the two approaches performs better both in classification performance and more importantly in feature reduction. RF with RFE achieved its lowest test error rate of 12.3% using 24 top-ranked features, whereas RF with INIT reached its lowest test error rate of 15.1% using 64 top-ranked features, compared to a test error rate of 22.1% using all 304 features. The results also show that 17 key features (out of 24 top-ranked features) are consistent between the subjects using RF with RFE, which is superior to the set of 64 features as determined by RF with INIT.