Characterisation of cognitive load using machine learning classifiers of electroencephalogram data

Date published

2023-10-17

Free to read from

Supervisor/s

Journal Title

Journal ISSN

Volume Title

Publisher

MDPI

Department

Type

Article

ISSN

1424-8220

Format

Citation

Wang Q, Smythe D, Cao J, et al., (2023) Characterisation of cognitive load using machine learning classifiers of electroencephalogram data. Sensors, Volume 23, Issue 20, Article number 8528

Abstract

A high cognitive load can overload a person, potentially resulting in catastrophic accidents. It is therefore important to ensure the level of cognitive load associated with safety-critical tasks (such as driving a vehicle) remains manageable for drivers, enabling them to respond appropriately to changes in the driving environment. Although electroencephalography (EEG) has attracted significant interest in cognitive load research, few studies have used EEG to investigate cognitive load in the context of driving. This paper presents a feasibility study on the simulation of various levels of cognitive load through designing and implementing four driving tasks. We employ machine learning-based classification techniques using EEG recordings to differentiate driving conditions. An EEG dataset containing these four driving tasks from a group of 20 participants was collected to investigate whether EEG can be used as an indicator of changes in cognitive load. The collected dataset was used to train four Deep Neural Networks and four Support Vector Machine classification models. The results showed that the best model achieved a classification accuracy of 90.37%, utilising statistical features from multiple frequency bands in 24 EEG channels. Furthermore, the Gamma and Beta bands achieved higher classification accuracy than the Alpha and Theta bands during the analysis. The outcomes of this study have the potential to enhance the Human–Machine Interface of vehicles, contributing to improved safety.

Description

Software Description

Software Language

Github

Keywords

electroencephalography, machine learning, Deep Neural Network, Support Vector Machine, cognitive load classification

DOI

Rights

Attribution 4.0 International

Relationships

Relationships

Supplements

Funder/s