Learning Models for Suicide Prediction from Social Media Posts
This work addresses suicide risk detection for mental health applications, but it is incremental as it builds on existing shared tasks and methods.
The paper tackles the problem of predicting suicide attempts from social media posts using machine learning models, achieving F1 scores of 0.741 for 30-day prediction and 0.737 for 6-month prediction.
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).