Title: Assessment of mental workload using XGBoost classifier from optimised EEG features
Authors: R.K. Kapila Vani; Jayashree Padmanabhan
Addresses: Sri Venkateswara College of Engineering, Sriperumbudur 602 117, India ' Department of Computer Technology, Anna University, India
Abstract: Cognitive workload evaluation is vital in any critical working environment for assessing the user's mental status. Despite the fact that there are many methods for evaluating cognitive strain, the model that uses electroencephalography (EEG) data remains the most promising one. Brain related activities can be used to assess various mental states and also help us to determine mental effort. This study calculates the cognitive workload of people while performing multitasking mental tasks. Here the 'STEW' dataset is used to measure mental effort. 'No task' and 'simultaneous capacity (SIMKAP)-based multitasking activity' are the two tasks in the dataset. For the study we have chosen only the SIMKAP task dataset. The cognitive workload assessment from optimised EEG features using XGBoost classifier (CWAOEX) framework is proposed, in which numerous features from EEG brain signals are retrieved and the grey wolf optimiser (GWO) is utilised to select the best ones. The data is then categorised according to the best feature set. The XGBoost algorithm is employed in the classification step. The recommended method has the classification accuracy of 94.25 in categorising the workload as low, moderate and high which is better than the current methods.
Keywords: XGBoost; grey wolf optimiser; GWO; STEW dataset.
DOI: 10.1504/IJESMS.2023.129988
International Journal of Engineering Systems Modelling and Simulation, 2023 Vol.14 No.2, pp.109 - 115
Received: 20 Jul 2021
Accepted: 17 Nov 2021
Published online: 04 Apr 2023 *