Finetuning Large Language Models for Suicide Risk Level Assessment on Social Media

Suicide remains a pressing global issue, and early detection of individuals at risk is vital for prevention. Traditional methods of suicide risk assessment rely heavily on questionnaires and interviews. However, social media offers a rich source of realtime data, providing an opportunity for automated suicide risk detection. In this study, we propose a novel method leveraging large language models (LLMs) to assess suicide risk based on social media activity. Our approach integrates RoBERTa, Llama 3.1, and Gemma 2 through a voting mechanism to enhance predictive performance. Results indicate that our model achieves an F1 score of 0.7006, outperforming traditional machine learning and deep learning models. We also observe that while fine-tuning large models on limited data yields better results than pre-trained models, increasing model size does not always improve performance. The use of a voting mechanism significantly improves the accuracy of the predictions, demonstrating the utility of multi-model approaches in suicide risk detection.

Authors

Institutions

Publication Details

Published
2024-12-15
DOI
https://doi.org/10.1109/bigdata62323.2024.10825211
Primary Topic
Mental Health via Writing
Type
article
Field-Weighted Citation Impact
0.00
Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

Finetuning Large Language Models for Suicide Risk Level Assessment on Social Media

Zihan Wei, Zhigang Wei, Xixian Chen, Lei Cao
Mental Health via Writing
article

Finetuning Large Language Models for Suicide Risk Level Assessment on Social Media

Zihan Wei, Zhigang Wei, Xixian Chen, Lei Cao
article en

Abstract

Suicide remains a pressing global issue, and early detection of individuals at risk is vital for prevention. Traditional methods of suicide risk assessment rely heavily on questionnaires and interviews. However, social media offers a rich source of realtime data, providing an opportunity for automated suicide risk detection. In this study, we propose a novel method leveraging large language models (LLMs) to assess suicide risk based on social media activity. Our approach integrates RoBERTa, Llama 3.1, and Gemma 2 through a voting mechanism to enhance predictive performance. Results indicate that our model achieves an F1 score of 0.7006, outperforming traditional machine learning and deep learning models. We also observe that while fine-tuning large models on limited data yields better results than pre-trained models, increasing model size does not always improve performance. The use of a voting mechanism significantly improves the accuracy of the predictions, demonstrating the utility of multi-model approaches in suicide risk detection.

Beijing Normal University (CN), Beijing University of Chemical Technology (CN)
Gender equality
Openalex Percentile: Top 66%
Mental Health via Writing
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.