EEG-Based Mental Workload Classification Method Based on Hybrid Deep Learning Model Under IoT.

Journal: IEEE journal of biomedical and health informatics
Published Date:

Abstract

Automatically detecting human mental workload to prevent mental diseases is highly important. With the development of information technology, remote detection of mental workload is expected. The development of artificial intelligence and Internet of Things technology will also enable the identification of mental workload remotely based on human physiological signals. In this article, a method based on the spatial and time-frequency domains of electroencephalography (EEG) signals is proposed to improve the classification accuracy of mental workload. Moreover, a hybrid deep learning model is presented. First, the spatial domain features of different brain regions are proposed. Simultaneously, EEG time-frequency domain information is obtained based on wavelet transform. The spatial and time-frequency domain features are input into two types of deep learning models for mental workload classification. To validate the performance of the proposed method, the Simultaneous Task EEG Workload public database is used. Compared with the existing methods, the proposed approach shows higher classification accuracy. It provides a novel means of assessing mental workload.

Authors

  • Shiliang Shao
  • Guangjie Han
  • Ting Wang
    CAS Key Laboratory of Separation Sciences for Analytical Chemistry, Dalian Institute of Chemical Physics, Chinese Academy of Sciences, Dalian 116023, China.
  • Chuan Lin
  • Chunhe Song
  • Chen Yao
    Division of Vascular Surgery, the First Affiliated Hospital, Sun Yat-sen University, Guangzhou 510800, China; National-Guangdong Joint Engineering Laboratory for Diagnosis and Treatment of Vascular Disease, First Affiliated Hospital, Sun Yat-sen University, Guangzhou 510080, China. Electronic address: yaochen@mail.sysu.edu.cn.