Exploring predictors of welfare dependency 1, 3, and 5 years after mental health-related absence in danish municipalities between 2010 and 2012 using flexible machine learning modelling

Abstract

Background Using XGBoost (XGB), this study demonstrates how flexible machine learning modelling can complement traditional statistical modelling (multinomial logistic regression) as a sensitivity analysis and predictive modelling tool in occupational health research. Design The study predicts welfare dependency for a cohort at 1, 3, and 5 years of follow-up using XGB and multinomial logistic regression (MLR). The models’ predictive ability is evaluated using tenfold cross-validation (internal validation) and geographical validation (semi-external validation). In addition, we calculate and graphically assess Shapley additive explanation (SHAP) values from the XGB model to examine deviation from linearity assumptions, including interactions. The study population consists of all 20–54 years old on long-term sickness absence leave due to self-reported common mental disorders (CMD) between April 26, 2010, and September 2012 in 21 (of 98) Danish municipalities that participated in the Danish Return to Work program. The total sample of 19.664 observations is split geospatially into a development set (n = 9.756) and a test set (n = 9.908). Results There were no practical differences in the XGB and MLR models’ predictive ability. Industry, job skills, citizenship, unemployment insurance, gender, and period had limited importance in predicting welfare dependency in both models. On the other hand, welfare dependency history and reason for sickness absence were strong predictors. Graphical SHAP-analysis of the XGB model did not indicate substantial deviations from linearity assumptions implied by the multinomial regression model. Conclusion Flexible machine learning models like XGB can supplement traditional statistical methods like multinomial logistic regression in occupational health research by providing a benchmark for predictive performance and traditional statistical models' ability to capture important associations for a given set of predictors as well as potential violations of linearity. Trial registration ISRCTN43004323.

Analysis of Health-Related Quality of Life in Elderly Patients with Stroke Complicated by Hypertension in China Using the EQ-5D-3L Scale

“…The optimal split is selected by a greedy algorithm to generate multiple decision trees and combine their predictions by weighting them to build a stronger model. 27,28 In this study, machine learning models (including Random Forest and XGBoost) were used to predict the health utility values of elderly hypertensive stroke patients, and the importance of each feature in the prediction was analyzed. In view of the small sample size of the study, deep learning models may be overfitted with small samples, so machine learning models are chosen instead of deep learning models in this study.…”

Interpretable machine learning in predicting drug-induced liver injury among tuberculosis patients: model development and validation study

“…Nevertheless, one of the primary challenges in implementing ML algorithms in clinical settings is interpreting the outcomes of the models [11,12]. The Shapley Additive exPlanations (SHAP) framework [13] provides insights into the influence of various features on model predictions and the effect of these features on the DILI status in individuals, thus bridging the interpretability gap.…”

Interpretable Machine Learning in Predicting Drug-Induced Liver Injury among Tuberculosis Patients: Model Development and Validation Study

“…However, one of the prevailing concerns in implementing ML in clinical settings is the challenge of interpreting model outcomes [11]. Addressing this, the SHapley Additive exPlanations (SHAP) framework [12] offers insights into how different features in uence model predictions, bridging the interpretability gap.…”