Advanced machine learning techniques for social support detection on social media
<h2>Abstract</h2> The widespread use of social media highlights the need to understand its impact, particularly the role of online social support. In this study, we present a dataset of YouTube comments, initially comprising 66,272 entries, which was refined to 42,695, with a subset of 10,000 comments selected for detailed analysis without additional filtering. The dataset is annotated for three classification tasks: (1) distinguishing supportive from non-supportive comments, (2) determining whether the support is directed at an individual or a group, and (3) further categorizing group support into six subtypes (Nation, LGBTQ, Black Community, Women, Religion, and Other). To address data imbalances in these tasks, we employed K-means clustering to balance the dataset and compared the results with the original unbalanced data. We use state-of-the-art transformer models and zero-shot learning techniques—including GPT-3, GPT-4, and GPT-4o. Our approach, evaluated using macro F1-scores, demonstrates strong performance on the imbalanced dataset, with our transformer-based model (roberta-base) achieving scores of 0.78, 0.84, and 0.80, respectively, across the three classification tasks. These results also demonstrate a macro F1-score improvement of 0.2% for Task 2 and 0.7% for Task 3 compared to previous work that used CNN with GloVe embeddings and traditional machine learning baselines based on TF-IDF and LIWC features.
Authors
- Moein Shahiki Tash (ORCID: https://orcid.org/0009-0003-5767-0566)
- Raúl Monroy (ORCID: https://orcid.org/0000-0002-3465-995X)
- Grigori Sidorov
- Olga Kolesnikova (ORCID: https://orcid.org/0000-0002-1307-1647)
- Zahra Ahani
- Ameeta Agrawal (ORCID: https://orcid.org/0000-0001-6189-099X)
Institutions
- Portland State University (US)
- Instituto Politécnico Nacional (MX)
- Tecnológico de Monterrey (MX)
Publication Details
- Journal
- Heliyon
- Published
- 2025-05-01
- DOI
- https://doi.org/10.1016/j.heliyon.2025.e43437
- Citations
- 3
- Primary Topic
- Mental Health via Writing
- Type
- article
- Field-Weighted Citation Impact
- 7.32
Funders
- Instituto Politécnico Nacional
- Consejo Nacional de Ciencia y Tecnología