TY - JOUR PY - 2020// TI - Predicting future suicidal behaviour in young adults, with different machine learning techniques: a population-based longitudinal study JO - Journal of affective disorders A1 - van Mens, Kasper A1 - de Schepper, Cwm A1 - Wijnen, Ben A1 - Koldijk, Saskia J. A1 - Schnack, Hugo A1 - de Looff, Peter A1 - Lokkerbol, Joran A1 - Wetherall, Karen A1 - Cleare, Seonaid A1 - O'Connor, Rory C. A1 - De Beurs, Derek SP - 169 EP - 177 VL - 271 IS - N2 - BACKGROUND: The predictive accuracy of suicidal behaviour has not improved over the last decades. We aimed to explore the potential of machine learning to predict future suicidal behaviour using population-based longitudinal data. METHOD: Baseline risk data assessed within the Scottish wellbeing study, in which 3508 young adults (18-34 years) completed a battery of psychological measures, were used to predict both suicide ideation and suicide attempts at one-year follow-up. The performance of the following algorithms was compared: regular logistic regression, K-nearest neighbors, classification tree, random forests, gradient boosting and support vector machine. RESULTS: At one year follow up, 2428 respondents (71%) finished the second assessment. 336 respondents (14%) reported suicide ideation between baseline and follow up, and 50 (2%) reported a suicide attempt. All performance metrics were highly similar across methods. The random forest algorithm was the best algorithm to predict suicide ideation (AUC 0.83, PPV 0.52, BA 0.74) and the gradient boosting to predict suicide attempt (AUC 0.80, PPV 0.10, BA 0.69). LIMITATIONS: The number of respondents with suicidal behaviour at follow up was small. We only had data on psychological risk factors, limiting the potential of the more complex machine learning algorithms to outperform regular logistical regression. CONCLUSIONS: When applied to population-based longitudinal data containing multiple psychological measurements, machine learning techniques did not significantly improve the predictive accuracy of suicidal behaviour. Adding more detailed data on for example employment, education or previous health care uptake, might result in better performance of machine learning over regular logistical regression.
Language: en
LA - en SN - 0165-0327 UR - http://dx.doi.org/10.1016/j.jad.2020.03.081 ID - ref1 ER -