Predicting future suicidal behaviour in young adults, with different machine learning techniques: a population-based longitudinal study

van Mens, K. et al. (2020) Predicting future suicidal behaviour in young adults, with different machine learning techniques: a population-based longitudinal study. Journal of Affective Disorders, 271, pp. 169-177. (doi: 10.1016/j.jad.2020.03.081) (PMID:32479313)

212741.pdf - Published Version
Available under License Creative Commons Attribution Non-commercial No Derivatives.



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 behavior. 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.

Item Type:Articles
Glasgow Author(s) Enlighten ID:O'Connor, Professor Rory and Cleare, Dr Seonaid and Wetherall, Miss Karen and De Beurs, Mr Derek
Authors: van Mens, K., de Schepper, C.W.M., Wijnen, B., Koldijk, S. J., Schnack, H., de Looff, P., Lokkerbol, J., Wetherall, K., Cleare, S., O'Connor, R. C., and De Beurs, D.
College/School:College of Medical Veterinary and Life Sciences > School of Health & Wellbeing > Mental Health and Wellbeing
Journal Name:Journal of Affective Disorders
ISSN (Online):1573-2517
Published Online:18 April 2020
Copyright Holders:Copyright © 2020 The Authors
First Published:First published in Journal of Affective Disorders 271: 169-177
Publisher Policy:Reproduced under a Creative Commons License

University Staff: Request a correction | Enlighten Editors: Update this record