Research Repository

EEG-Based Multi-Class Workload Identification Using Feature Fusion and Selection

Pei, Zian and Wang, Hongtao and Bezerianos, Anastasios and Li, Junhua (2021) 'EEG-Based Multi-Class Workload Identification Using Feature Fusion and Selection.' IEEE Transactions on Instrumentation and Measurement, 70. pp. 1-8. ISSN 0018-9456

09178781.pdf - Published Version
Available under License Creative Commons Attribution.

Download (1MB) | Preview


The effectiveness of workload identification is one of the critical aspects in a monitoring instrument of mental state. In this field, the workload is usually recognised as binary classes. There are scarce studies towards multi-class workload identification because the challenge of the success of workload identification is much tough, even though one more workload class is added. Besides, most of the existing studies only utilized spectral power features from individual channels but ignoring abundant inter-channel features that represent the interactions between brain regions. In this study, we utilized features representing intra-channel information and inter-channel information to classify multiple classes of workload based on EEG. We comprehensively compared each category of features contributing to workload identification and elucidated the roles of feature fusion and feature selection for the workload identification. The results demonstrated that feature combination (83.12% in terms of accuracy) enhanced the classification performance compared to individual feature categories (i.e., band power features, 75.90%; connection features, 81.72%, in terms of accuracy). With the F-score feature selection, the classification accuracy was further increased to 83.47%. When the features of graph metric were fused, the accuracy was reached to 84.34%. Our study provided comprehensive performance comparisons between methods and feature categories for the multi-class workload identification and demonstrated that feature selection and fusion played an important role in the enhancement of workload identification. These results could facilitate further studies of multi-class workload identification and practical application of workload identification.

Item Type: Article
Uncontrolled Keywords: Brain connectivity; electroencephalogram (EEG); feature fusion; feature selection; graph metric; mental workload identification; power spectral density
Divisions: Faculty of Science and Health
Faculty of Science and Health > Computer Science and Electronic Engineering, School of
SWORD Depositor: Elements
Depositing User: Elements
Date Deposited: 08 Oct 2020 15:35
Last Modified: 22 Nov 2022 22:02

Actions (login required)

View Item View Item