Electroencephalogram-based mental workload prediction for using Augmented Reality head mounted display in construction assembly: A deep learning approach

Augmented Reality Head-Mounted Displays (AR HMD) are considered a promising technology for assisting on -site construction tasks such as assembly. However, concerns about creating information overload and becoming a distraction to workers outweigh the potential benefits to productivity and task performance. Previous research investigated workers' mental workloads as impacted by AR-based displays for assembly tasks. Within that context, this paper proposes a Long Short-Term Memory (LSTM) based approach to predict mental workload when using AR HMD for construction assembly, to forecast users' cognitive status under such complex working condition. Thirty participants were recruited to finish a wood frame assembly using an AR HMD projecting a 3D conformal model on the workspace. The proposed method provided a reliable prediction of real-time electro-encephalogram (EEG) signal and mental workload. The outcomes validated the feasibility of LSTM model for EEG-based mental workload prediction and specifically provided a usable method for evaluating AR HMD use in construction tasks.