Abstract:BACKGROUND:Suicide is a major public health concern globally. Accurately predicting suicidal behavior remains challenging. This study aimed to use machine learning approaches to examine the potential of the Swedish national registry data for prediction of suicidal behavior.METHODS AND FINDINGS:The study sample consisted of 541,300 inpatient and outpatient visits by 126,205 Sweden-born patients (54% female and 46% male) aged 18 to 39 (mean age at the visit: 27.3) years to psychiatric specialty care in Sweden between January 1, 2011 and December 31, 2012. The most common psychiatric diagnoses at the visit were anxiety disorders (20.0%), major depressive disorder (16.9%), and substance use disorders (13.6%). A total of 425 candidate predictors covering demographic characteristics, socioeconomic status (SES), electronic medical records, criminality, as well as family history of disease and crime were extracted from the Swedish registry data. The sample was randomly split into an 80% training set containing 433,024 visits and a 20% test set containing 108,276 visits. Models were trained separately for suicide attempt/death within 90 and 30 days following a visit using multiple machine learning algorithms. Model discrimination and calibration were both evaluated. Among all eligible visits, 3.5% (18,682) were followed by a suicide attempt/death within 90 days and 1.7% (9,099) within 30 days. The final models were based on ensemble learning that combined predictions from elastic net penalized logistic regression, random forest, gradient boosting, and a neural network. The area under the receiver operating characteristic (ROC) curves (AUCs) on the test set were 0.88 (95% confidence interval [CI] = 0.87-0.89) and 0.89 (95% CI = 0.88-0.90) for the outcome within 90 days and 30 days, respectively, both being significantly better than chance (i.e., AUC = 0.50) (p < 0.01). Sensitivity, specificity, and predictive values were reported at different risk thresholds. A limitation of our study is that our models have not yet been externally validated, and thus, the generalizability of the models to other populations remains unknown.CONCLUSIONS:By combining the ensemble method of multiple machine learning algorithms and high-quality data solely from the Swedish registers, we developed prognostic models to predict short-term suicide attempt/death with good discrimination and calibration. Whether novel predictors can improve predictive performance requires further investigation.

A machine learning approach for predicting suicidal ideation in post stroke patients

Prediction of Poststroke Depression Based on the Outcomes of Machine Learning Algorithms

Explainable Risk Prediction of Post-Stroke Adverse Mental Outcomes Using Machine Learning Techniques in a Population of 1780 Patients

Prediction of Suicidal Ideation among Korean Adults Using Machine Learning: A Cross-Sectional Study

Prediction-Driven Decision Support for Patients With Mild Stroke: A Model Based on Machine Learning Algorithms

Applying Machine Learning Approaches to Suicide Prediction Using Healthcare Data: Overview and Future Directions

Neural activity during inhibitory control predicts suicidal ideation with machine learning

A machine learning approach predicts future risk to suicidal ideation from social media data

Identifying momentary suicidal ideation using machine learning in patients at high-risk for suicide

Predicting new-onset post-stroke depression from real-world data using machine learning algorithm

Constructing prediction models and analyzing factors in suicidal ideation using machine learning, focusing on the older population

Machine Learning–Based Prediction of Suicidal Thinking in Adolescents by Derivation and Validation in 3 Independent Worldwide Cohorts: Algorithm Development and Validation Study

Prediction of Suicidal Thoughts and Suicide Attempts in People Who Gamble Based on Biological-Psychological-Social Variables: A Machine Learning Study

Predicting Suicide Attempt or Suicide Death Following a Visit to Psychiatric Specialty Care: A Machine Learning Study Using Swedish National Registry Data

Machine Learning–Based Suicide Risk Prediction Model for Suicidal Trajectory on Social Media Following Suicidal Mentions: Independent Algorithm Validation

Identifying the suicidal ideation risk group among older adults in rural areas: Developing a predictive model using machine learning methods

Risk Prediction Model for Non-Suicidal Self-Injury in Chinese Adolescents with Major Depressive Disorder Based on Machine Learning

Development of an AI-Based Suicide Ideation Prediction Model for People with Disabilities

Highlighting Psychological Pain Avoidance and Decision-Making Bias As Key Predictors of Suicide Attempt in Major Depressive Disorder-a Novel Investigative Approach Using Machine Learning

Prediction of suicidal ideation in children and adolescents using machine learning and deep learning algorithm: A case study in South Korea where suicide is the leading cause of death

Machine learning in the prediction of post-stroke cognitive impairment: a systematic review and meta-analysis