Machine Learning to Classify Suicidal Thoughts and Behaviors: Implementation Within the Common Data Elements Used by the Military Suicide Research Consortium

Andrew K. Littlefield, Jeffrey T. Cooke, Courtney L. Bagge, Catherine R. Glenn, Evan M. Kleiman, Ross Jacobucci, Alexander J. Millner, Douglas Steinley

Research output: Contribution to journalArticlepeer-review

Abstract

Suicide rates among military-connected populations have increased over the past 15 years. Meta-analytic studies indicate prediction of suicide outcomes is lacking. Machine-learning approaches have been promoted to enhance classification models for suicide-related outcomes. In the present study, we compared the performance of three primary machine-learning approaches (i.e., elastic net, random forests, stacked ensembles) and a traditional statistical approach, generalized linear modeling (i.e., logistic regression), to classify suicide thoughts and behaviors using data from the Military Suicide Research Consortium’s Common Data Elements (CDE; n = 5,977–6,058 across outcomes). Models were informed by (a) selected items from the CDE or (b) factor scores based on exploratory and confirmatory factor analyses on the selected CDE items. Results indicated similar classification performance across models and sets of features. In this study, we suggest the need for robust evidence before adopting more complex classification models and identify measures that are particularly relevant in classifying suicide-related outcomes.

Original languageEnglish
JournalClinical Psychological Science
DOIs
StateAccepted/In press - 2021

Keywords

  • classification
  • machine learning
  • statistical analysis
  • suicide prevention

Fingerprint Dive into the research topics of 'Machine Learning to Classify Suicidal Thoughts and Behaviors: Implementation Within the Common Data Elements Used by the Military Suicide Research Consortium'. Together they form a unique fingerprint.

Cite this