Prediction of mental fatigue states based on the CNN-LSTM architecture for BCG signals

To achieve feature fusion of ballistocardiogram (BCG) signals for the purpose of predicting mental fatigue states, a prediction model based on convolutional neural network-long short-term memory (CNN-LSTM) architecture is built in this paper. The model combines the BCG signal Heart Rate Variability (HRV) parameter values with multiscale features to achieve accurate prediction of the BCG signal state. The results show that the fatigue classification proposed in this paper is highly consistent with the subjective evaluation, and all the evaluation indexes are better than the traditional methods. It achieved an accuracy of 95%, a precision of 96%, a linear regression of 0.9, and a G-mean of 0.93. In addition, it is able to automatically accelerate the learning of features without human intervention. There is no need to perform a tedious feature extraction process compared to traditional models. The accuracy and real-time performance of mental fatigue monitoring can be improved by using this model.