We propose a deep learning architecture and test three other machine learning models to automatically detect individuals that will attempt suicide within (1) 30 days and (2) six months, using their social media post data provided in the CLPsych 2021 shared task. Additionally, we create and extract three sets of handcrafted features for suicide risk detection based on the three-stage theory of suicide and prior work on emotions and the use of pronouns among persons exhibiting suicidal ideations. Extensive experimentations show that some of the traditional machine learning methods outperform the baseline with an F1 score of 0.741 and F2 score of 0.833 on subtask 1 (prediction of a suicide attempt 30 days prior). However, the proposed deep learning method outperforms the baseline with F1 score of 0.737 and F2 score of 0.843 on subtask 2 (prediction of suicide 6 months prior).
@article{arxiv.2105.03315,
title = {Learning Models for Suicide Prediction from Social Media Posts},
author = {Ning Wang and Fan Luo and Yuvraj Shivtare and Varsha D. Badal and K. P. Subbalakshmi and R. Chandramouli and Ellen Lee},
journal= {arXiv preprint arXiv:2105.03315},
year = {2021}
}
Comments
This work is accepted to CLPsych 2021, to be held in conjunction with NAACL 2021