Towards an effective cross-task mental workload recognition model using electroencephalography based on feature selection and support vector machine regression.
Electroencephalographic (EEG) has been believed to be a potential psychophysiological measure of mental workload. There however remain a number of challenges in building a generalized mental workload recognition model, one of which includes the inability of an EEG-based workload classifier trained on a specific task to handle other tasks. The primary goal of the present study was to examine the possibility of addressing this challenge using feature selection and regression model. Support vector machine classifier and regression models were examined under within-task conditions (trained and tested on the same task) and cross-task conditions (trained on one task and tested on another task) for well-trained verbal and spatial n-back tasks. A specifically designed cross-task recursive feature elimination (RFE) based feature selection was used to handle the possible causes responsible for the deterioration of the performance of cross-task regression model. The within-task classification and regression performed fairly well. Cross-task classification and regression performance, however, deteriorated to unacceptable levels (around chance level). Trained and tested with the most robust feature subset selected by cross-task RFE, the performance of cross-task regression was significantly improved, and there were no significant changes in the performance of within-task regression. It can be inferred that workload-related features can be picked out from those which have been contaminated using RFE, and regression models rather than classifiers may be a wiser choice for cross-task conditions. These encouraging results suggest that the cross-task workload recognition model built in this study is much more generalizable across task when compared to the model built in traditional way.
Department of Biomedical Engineering, College of Precision Instrument and Optoelectronics Engineering, Tianjin University, No. 92 Weijin Road, Nankai District, Tianjin 300072, PR China; Tianjin Key Laboratory of Biomedical Detecting Techniques and Instruments, Tianjin, PR China.
Yufeng
Y
Ke
Department of Biomedical Engineering, College of Precision Instrument and Optoelectronics Engineering, Tianjin University, No. 92 Weijin Road, Nankai District, Tianjin 300072, PR China; Tianjin Key Laboratory of Biomedical Detecting Techniques and Instruments, Tianjin, PR China. Electronic address: qhz@tju.edu.cn.
Hongzhi
H
Qi
Department of Biomedical Engineering, College of Precision Instrument and Optoelectronics Engineering, Tianjin University, No. 92 Weijin Road, Nankai District, Tianjin 300072, PR China; Tianjin Key Laboratory of Biomedical Detecting Techniques and Instruments, Tianjin, PR China.
Lixin
L
Zhang
National Laboratory of Human Factors Engineering, China Astronaut Research and Training Centre, Beijing, PR China.
Shanguang
S
Chen
National Laboratory of Human Factors Engineering, China Astronaut Research and Training Centre, Beijing, PR China.
Xuejun
X
Jiao
Department of Biomedical Engineering, College of Precision Instrument and Optoelectronics Engineering, Tianjin University, No. 92 Weijin Road, Nankai District, Tianjin 300072, PR China; Tianjin Key Laboratory of Biomedical Detecting Techniques and Instruments, Tianjin, PR China.
Peng
P
Zhou
Department of Biomedical Engineering, College of Precision Instrument and Optoelectronics Engineering, Tianjin University, No. 92 Weijin Road, Nankai District, Tianjin 300072, PR China; Tianjin Key Laboratory of Biomedical Detecting Techniques and Instruments, Tianjin, PR China.
Xin
X
Zhao
Department of Biomedical Engineering, College of Precision Instrument and Optoelectronics Engineering, Tianjin University, No. 92 Weijin Road, Nankai District, Tianjin 300072, PR China; Tianjin Key Laboratory of Biomedical Detecting Techniques and Instruments, Tianjin, PR China.
Baikun
B
Wan
Department of Biomedical Engineering, College of Precision Instrument and Optoelectronics Engineering, Tianjin University, No. 92 Weijin Road, Nankai District, Tianjin 300072, PR China; Tianjin Key Laboratory of Biomedical Detecting Techniques and Instruments, Tianjin, PR China. Electronic address: richardming@tju.edu.cn.
Dong
D
Ming
D000465:Algorithms; D001921:Brain / Q000502:physiology*; D004569:Electroencephalography / Q000379:methods*; D005260:Female; D006801:Humans; D008297:Male; D008962:Models, Theoretical; D011930:Reaction Time / Q000502:physiology; D060388:Support Vector Machine; D016526:Workload; D055815:Young Adult
Cross-task; Electroencephalography; Mental workload; Recursive feature elimination; Support vector machine