Artificial Intelligence-Based Prediction Models for Suicide Risk Assessment and Prevention: A Retrospective Observational Study
Keywords:
Artificial intelligence, machine learning, suicide prediction, suicide prevention, mental health, hospital medical records, XGBoost, psychiatric risk assessment.Abstract
Background: Suicide remains one of the leading causes of preventable mortality worldwide and represents a major public health challenge. Despite advances in psychiatric assessment and mental health services, accurately identifying individuals at imminent risk of suicide remains difficult because suicidal behavior arises from a complex interaction of biological, psychological, social, and environmental factors. Conventional clinical risk assessment methods primarily rely on clinician judgment and standardized screening tools, which often demonstrate limited predictive accuracy and considerable inter-observer variability. Recent developments in artificial intelligence (AI) and machine learning (ML) have enabled the integration of large-scale hospital medical records, demographic variables, psychiatric history, medication profiles, laboratory findings, and behavioral characteristics to generate individualized suicide risk predictions. AI-driven predictive models have demonstrated superior performance compared with traditional statistical approaches by identifying complex nonlinear relationships and hidden interactions among multiple risk factors. However, evidence regarding the applicability of these models in routine clinical practice remains limited, particularly in retrospective hospital-based settings.
Objective: To evaluate the performance of artificial intelligence-based prediction models for suicide risk assessment and prevention using retrospective clinical data and to identify the most influential predictors associated with high suicide risk.
Materials and Methods: A retrospective observational study was conducted over four months using medical records of 100 patients who underwent psychiatric evaluation at PMCH, Patna. Demographic characteristics, psychiatric diagnoses, previous suicide attempts, substance use disorders, family history of mental illness, medication adherence, hospitalization history, depression severity scores, anxiety scores, and psychosocial variables were extracted from hospital medical records maintained in the Medical Record Department (MRD). Missing values were managed through multiple imputation where appropriate. Four predictive models—Logistic Regression, Random Forest, Support Vector Machine (SVM), and Extreme Gradient Boosting (XGBoost)—were developed and internally validated using five-fold cross-validation. Model performance was evaluated using accuracy, sensitivity, specificity, precision, F1-score, area under the receiver operating characteristic curve (AUC), and calibration statistics. Statistical significance was considered at p <0.05.
Results: Among the 100 patients included, 36 were categorized as high suicide risk based on documented psychiatric evaluation. Previous suicide attempt, severe depressive symptoms, substance use disorder, medication non-adherence, unemployment, and multiple psychiatric hospitalizations were significantly associated with high suicide risk (p<0.05). XGBoost demonstrated the highest predictive performance with an AUC of 0.94, sensitivity of 91.7%, specificity of 89.1%, accuracy of 90.0%, and F1-score of 0.90. Random Forest achieved an AUC of 0.91, while SVM and Logistic Regression demonstrated AUC values of 0.88 and 0.84, respectively. Feature importance analysis identified previous suicide attempt, depression severity score, medication adherence, substance use disorder, psychiatric hospitalization history, and age as the strongest predictors.
Conclusion: Artificial intelligence-based prediction models demonstrated excellent discrimination in identifying patients at elevated suicide risk. Tree-based ensemble algorithms, particularly XGBoost, outperformed conventional logistic regression by effectively modeling complex clinical interactions. Integration of AI-assisted decision support into psychiatric practice may facilitate early identification of vulnerable individuals, optimize allocation of mental health resources, and strengthen suicide prevention strategies. Prospective multicenter validation studies are warranted before widespread clinical implementation.
References
World Health Organization. Suicide worldwide in 2019: global
health estimates. Geneva: World Health Organization; 2021.
Turecki G, Brent DA. Suicide and suicidal behaviour. Lancet.
;387(10024):1227-1239.
Franklin JC, Ribeiro JD, Fox KR, Bentley KH, Kleiman
EM, Huang X, et al. Risk factors for suicidal thoughts and
behaviors: a meta-analysis of 50 years of research. Psychol
Bull. 2017;143(2):187-232.
Posner K, Brown GK, Stanley B, Brent DA, Yershova KV,
Oquendo MA, et al. The Columbia–Suicide Severity Rating
Scale: initial validity and internal consistency findings. Am
J Psychiatry. 2011;168(12):1266-1277.
Kessler RC, Abelson JM, Demler O, Escobar JI, Gibbon M,
Guyer ME, et al. Clinical calibration of DSM-IV diagnoses
in the World Mental Health surveys. Int J Methods Psychiatr
Res. 2004;13(2):122-139.
Obermeyer Z, Emanuel EJ. Predicting the future—big data,
machine learning, and clinical medicine. N Engl J Med.
;375(13):1216-1219.
Kessler RC, Warner CH, Ivany C, Petukhova MV, Rose
S, Bromet EJ, et al. Predicting suicides after psychiatric
hospitalization in US Army soldiers: results from the Army
Study to Assess Risk and Resilience in Servicemembers.
JAMA Psychiatry. 2015;72(1):49-57.
Simon GE, Shortreed SM, Johnson E, Beck A, Coleman KJ,
Rossom RC, et al. Integrated primary care and populationbased
suicide prevention. Psychiatr Serv. 2018;69(4):443-449.
Belsher BE, Smolenski DJ, Pruitt LD, Bush NE, Beech EH,
Workman DE, et al. Prediction models for suicide attempts and
deaths: a systematic review and simulation. JAMA Psychiatry.
;76(6):642-651.
Rajkomar A, Dean J, Kohane I. Machine learning in medicine.
N Engl J Med. 2019;380(14):1347-1358.
Shatte ABR, Hutchinson DM, Teague SJ. Machine learning in
mental health: a scoping review of methods and applications.
Psychol Med. 2019;49(9):1426-1448.
Dwyer DB, Falkai P, Koutsouleris N. Machine learning
approaches for clinical psychology and psychiatry. Annu Rev
Clin Psychol. 2018;14:91-118.
McCarthy JF, Bossarte RM, Katz IR, Thompson C, Kemp
J, Hannemann CM, et al. Predictive modeling and targeted
suicide prevention: advances and challenges. PsychiatrClin
North Am. 2018;41(3):447-458.
Ribeiro JD, Franklin JC, Fox KR, Bentley KH, Kleiman EM,
Chang BP, et al. Self-injurious thoughts and behaviors as risk
factors for future suicide attempts: a meta-analysis. Psychol
Med. 2016;46(2):225-236.
Oquendo MA, Currier D, Mann JJ. Prospective studies of
suicidal behavior in major depressive and bipolar disorders.
PsychiatrClin North Am. 2006;29(4):1001-1024.
Hoertel N, Sánchez-Rico M, de Maricourt P, et al. Association
between medication adherence and suicidal behavior among
psychiatric patients. J Clin Psychiatry. 2021;82(3):20m13691.
Borges G, Walters EE, Kessler RC. Associations of substance
use, abuse, and dependence with subsequent suicidal behavior.
Am J Epidemiol. 2000;151(8):781-789.
Madsen T, Erlangsen A, Nordentoft M. Risk factors and
prevention of suicide after psychiatric admission. J Clin
Psychiatry. 2017;78(9):e1102-e1110.
Chen T, Guestrin C. XGBoost: a scalable tree boosting
system. Proceedings of the 22nd ACM SIGKDD International
Conference on Knowledge Discovery and Data Mining.
:785-794.
Breiman L. Random forests. Mach Learn. 2001;45:5-32.
Steyerberg EW. Clinical prediction models: a practical
approach to development, validation, and updating. New
York: Springer; 2019.
Lundberg SM, Lee SI. A unified approach to interpreting
model predictions. Advances in Neural Information
Processing Systems. 2017;30:4765-4774.
Vellido A. The importance of interpretability and visualization
in machine learning for applications in medicine. Neural
Comput Appl. 2019;31:1-15.
Char DS, Shah NH, Magnus D. Implementing machine
learning in health care—addressing ethical challenges. N
Engl J Med. 2018;378(11):981-983.
Topol EJ. High-performance medicine: the convergence of
human and artificial intelligence. Nat Med. 2019;25(1):44-56.
Downloads
Published
How to Cite
Issue
Section
License

This work is licensed under a Creative Commons Attribution-NonCommercial-NoDerivatives 4.0 International License.
