Classifying Multi-Level Stress Responses From Brain Cortical EEG in Nurses and Non-Health Professionals Using Machine Learning Auto Encoder

IEEE J Transl Eng Health Med. 2021 May 5:9:2200109. doi: 10.1109/JTEHM.2021.3077760. eCollection 2021.

Abstract

Objective: Mental stress is a major problem in our society and has become an area of interest for many psychiatric researchers. One primary research focus area is the identification of bio-markers that not only identify stress but also predict the conditions (or tasks) that cause stress. Electroencephalograms (EEGs) have been used for a long time to study and identify bio-markers. While these bio-markers have successfully predicted stress in EEG studies for binary conditions, their performance is suboptimal for multiple conditions of stress.

Methods: To overcome this challenge, we propose using latent based representations of the bio-markers, which have been shown to significantly improve EEG performance compared to traditional bio-markers alone. We evaluated three commonly used EEG based bio-markers for stress, the brain load index (BLI), the spectral power values of EEG frequency bands (alpha, beta and theta), and the relative gamma (RG), with their respective latent representations using four commonly used classifiers.

Results: The results show that spectral power value based bio-markers had a high performance with an accuracy of 83%, while the respective latent representations had an accuracy of 91%.

Keywords: Autoencoder; electroencephalogram; stress classification; support vector machine.

Publication types

  • Research Support, Non-U.S. Gov't

MeSH terms

  • Brain*
  • Electroencephalography*
  • Humans
  • Machine Learning
  • Stress, Psychological / diagnosis

Grants and funding

This work was supported by New South Wales (NSW) Defence Innovation Network through NSW State Government under Grant DINPP2018-02-06.