Marquez Salazar, Diego (2025) Fine-Tuned DistilBERT and Interactive Dashboard for Early Detection of Suicidal Ideation in Social Media Posts. Masters thesis, Dublin, National College of Ireland.
Preview |
PDF (Master of Science)
Download (847kB) | Preview |
Preview |
PDF (Configuration Manual)
Download (411kB) | Preview |
Abstract
The early detection of suicidal ideation in social media content can save lives by enabling timely interventions. In this work, we present the development of a fine-tuned DistilBERT model on a pre-processed dataset to classify texts as indicative of suicide risk or not. The model was integrated into an interactive dashboard built with Streamlit, which allows both manual input of individual texts and bulk analysis through CSV file uploads. The tool displays classification results along with estimated probabilities and provides additional visualizations, such as word frequency distributions, to help uncover patterns in the data. Experimental evaluation using 5-fold cross-validation demonstrated outstanding performance, achieving accuracy above 93% and ROC-AUC scores exceeding 0.90 This solution offers a practical and socially impactful contribution, aimed at mental health professionals, researchers, and organizations seeking to monitor and prevent suicide related risks within online communities.
Actions (login required)
![]() |
View Item |
Tools
Tools