Saved in:
Bibliographic Details
Main Authors: Lin, Emily, Sun, Jian, Chen, Hsingyu, Mahoor, Mohammad H.
Format: Preprint
Published: 2024
Subjects:
Online Access:https://arxiv.org/abs/2402.02262
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866910756562272256
author Lin, Emily
Sun, Jian
Chen, Hsingyu
Mahoor, Mohammad H.
author_facet Lin, Emily
Sun, Jian
Chen, Hsingyu
Mahoor, Mohammad H.
contents Suicide remains a pressing global health concern, necessitating innovative approaches for early detection and intervention. This paper focuses on identifying suicidal intentions in posts from the SuicideWatch subreddit by proposing a novel deep-learning approach that utilizes the state-of-the-art RoBERTa-CNN model. The robustly Optimized BERT Pretraining Approach (RoBERTa) excels at capturing textual nuances and forming semantic relationships within the text. The remaining Convolutional Neural Network (CNN) head enhances RoBERTa's capacity to discern critical patterns from extensive datasets. To evaluate RoBERTa-CNN, we conducted experiments on the Suicide and Depression Detection dataset, yielding promising results. For instance, RoBERTa-CNN achieves a mean accuracy of 98% with a standard deviation (STD) of 0.0009. Additionally, we found that data quality significantly impacts the training of a robust model. To improve data quality, we removed noise from the text data while preserving its contextual content through either manually cleaning or utilizing the OpenAI API.
format Preprint
id arxiv_https___arxiv_org_abs_2402_02262
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Data Quality Matters: Suicide Intention Detection on Social Media Posts Using RoBERTa-CNN
Lin, Emily
Sun, Jian
Chen, Hsingyu
Mahoor, Mohammad H.
Computation and Language
Artificial Intelligence
Suicide remains a pressing global health concern, necessitating innovative approaches for early detection and intervention. This paper focuses on identifying suicidal intentions in posts from the SuicideWatch subreddit by proposing a novel deep-learning approach that utilizes the state-of-the-art RoBERTa-CNN model. The robustly Optimized BERT Pretraining Approach (RoBERTa) excels at capturing textual nuances and forming semantic relationships within the text. The remaining Convolutional Neural Network (CNN) head enhances RoBERTa's capacity to discern critical patterns from extensive datasets. To evaluate RoBERTa-CNN, we conducted experiments on the Suicide and Depression Detection dataset, yielding promising results. For instance, RoBERTa-CNN achieves a mean accuracy of 98% with a standard deviation (STD) of 0.0009. Additionally, we found that data quality significantly impacts the training of a robust model. To improve data quality, we removed noise from the text data while preserving its contextual content through either manually cleaning or utilizing the OpenAI API.
title Data Quality Matters: Suicide Intention Detection on Social Media Posts Using RoBERTa-CNN
topic Computation and Language
Artificial Intelligence
url https://arxiv.org/abs/2402.02262