Background: Suicide is a serious social and public health problem. Social stigma and prejudice reduce the accessibility of mental health care services for high-risk groups, resulting in them not receiving interventions and committing suicide. A suicide prediction model is necessary to identify high-risk groups in the general population.
Methods: We used national medical check-up data from 2009 to 2015 in Korea. The latest medical check-up data for each subject was set as an index point. Analysis was undertaken for an overall follow-up period (index point to the final tracking period) as well as for a one-year follow-up period. The training set was cross-validated fivefold. The predictive model was trained using a random forest algorithm, and its performance was measured using a separate test set not included in the training.
Results: The analysis covered 372,813 individuals, with an average (SD) overall follow-up duration of 1.52 (1.52) years. When we predicted suicide during the overall follow-up period, the area under the receiver operating characteristic curve (AUC) was 0.849, sensitivity was 0.817, and specificity was 0.754. The performance of the predicted suicide risk model for one year from the index point was AUC 0.818, sensitivity 0.788, and specificity 0.657.
Conclusions: This is probably the first suicide predictive model using machine learning based on medical check-up data from the general population. It could be used to screen high-risk suicidal groups from the population through routine medical check-ups. Future studies may test preventive interventions such as exercise and alcohol in these high-risk groups.
Keywords: Artificial intelligence; Big data; Machine learning; Suicide.
Copyright © 2020 Elsevier Ltd. All rights reserved.