Comparison of back propagation artificial neural network and logistic regression models for predicting cumulative fatigue among petrochemical employees: a web-based questionnaire survey

Objective To compare the performance of back propagation artificial neural network (BP-ANN) and logistic regression models in predicting the risk of cumulative fatigue among petrochemical employees and evaluate the prediction models for early identification and intervention of cumulative fatigue in these employees.

Methods A total of 4 066 employees of a petrochemical enterprise in Jiangsu province were selected by cluster sampling for a cross-sectional study from July to October 2021. A web-based questionnaire was designed to investigate the general information, work conditions, occupational stress, and prevalence of cumulative fatigue among employees. SPSS 23.0 was used to analyze the influencing factors of cumulative fatigue among petrochemical employees through non-parametric tests. BP-ANN and logistic regression models were established and receiver operating characteristic (ROC) curves were plotted to compare the prediction performance of models based on sensitivity, specificity, Youden index, and area under the curve (AUC) value.

Results A total of 3 763 valid questionnaires were collected, which showed that 63.2% (2 377/3 763) of petrochemical employees experienced cumulative fatigue. The comprehensive analysis of the two models indicated that age, weekly working hours, shift work, occupational stress, and regular exercise were the key factors influencing employees’ cumulative fatigue. Both models had AUC values > 0.7. Sensitivity was comparable between the BP-ANN and logistic models (BP 75.7%, logistic 75.6%). The BP-ANN model achieved higher specificity (BP 76.3%, logistic 69.3%), Youden index (BP 51.9%, logistic 44.9%), and AUC value (BP 84.4%, logistic 79.9%) than the logistic model. Both models can be used to predict the incidence of cumulative fatigue among petrochemical employees, while the overall prediction performance of the BP-ANN model was superior to that of the logistic model.

Conclusions Both BP-ANN and logistic regression models can be adopted to predict cumulative fatigue among petrochemical employees, with the BP-ANN model showing better prediction performance.