arXiv preprint ·

Multidimensional Suicide Risk Assessment with Machine Learning using Electronic Health Records

Nathan Scollar, Kemal Davaslioglu, Nathan Conger, Sastry Kompella, Yalin E. Sagduyu and Nathaniel D. Bastian

Published
Source
arXiv preprint, posted before peer review
Original
Read it at arXiv
Topics
Machine Learning

In brief

Reading Suicide Risk Across the EHR

The authors build a multidimensional machine-learning framework using MIMIC-IV data and a composite outcome based on documented suicidal ideation, attempts, prior suicidal behavior, and intentional self-harm. They compare three classifiers, use age-matched resampling and isotonic calibration, and find that calibrated XGBoost performs best in the matched cohort.

What they found: In the matched cohort, calibrated XGBoost achieves the strongest performance among the evaluated classifiers. SHAP interpretation links elevated predicted probabilities to diagnostic burden, depression-related features, age, personality-disorder history, violence-related social determinants, and medical comorbidity. The results demonstrate a practical framework for retrospective modeling from broad EHR-derived information.

AI-written summary. Check anything you rely on against the original paper.

Abstract

Suicide prevention remains a critical public health challenge, and electronic health records (EHRs) provide an opportunity to characterize suicide-related risk indicators using documented clinical, behavioral, medical, and social-contextual information. This paper presents a multidimensional machine learning framework to model and estimate suicide-related outcomes using MIMIC-IV EHR data. We construct clinically motivated diagnosis categories and define a composite outcome from documented codes corresponding to suicidal ideation, suicide attempt, prior suicidal behavior, and intentional self-harm. To address severe class imbalance and reduce age-related confounding, we apply age-matched resampling that retains all positive patients and samples comparison-cohort patients from the same age group. Logistic regression, random forest, and XGBoost classifiers are evaluated before and after resampling with isotonic calibration. The calibrated XGBoost model achieves the strongest matched-cohort performance, outperforming the other evaluated classifiers. SHAP-based interpretation shows that diagnostic burden, depression-related features, age, personality-disorder history, social determinants related to violence, and medical comorbidity contribute to elevated predicted probabilities. Our results demonstrate that combining broad EHR-derived features, age-matched cohort construction, calibration, and feature-level interpretation provides a practical framework for retrospective suicide-related risk modeling.

The authors’ abstract, as arXiv lists it. arXiv’s listings, abstracts included, are free for anyone to reuse.

Read it on ReadSearch

Each opens after you create an account, and brings you back to this paper.

Already a reader? Sign in

All new papers in Machine Learning

Get the few papers worth your time in Machine Learning every morning.

ReadSearch reads the new papers in your field each day and brings you the few worth reading, each with its key finding in a sentence.

Papers belong to their authors and publishers; every page here links to the original. All research topics