Text Classification with Limited Training Data: Suicide Risk Detection on Social Media
In this work, we evaluate the effectiveness of several machine learning models for text classification on small datasets, focusing on a collection of Reddit posts labeled for suicidal behavior. Unlike with larger datasets, where fine-tuning complex models is typically effective, we demonstrate that carefully engineered prompts can achieve superior classification accuracy when training data is limited. Our findings highlight the potential of prompt-based approaches for effective use in resource-constrained scenarios, offering insights for researchers tackling similar small dataset challenges in text classification.
Authors
- Stefan Pasch
- Jannic Cutura
Institutions
- European Central Bank (DE)
- Hankuk University of Foreign Studies (KR)
Publication Details
- Published
- 2024-12-15
- DOI
- https://doi.org/10.1109/bigdata62323.2024.10825333
- Citations
- 3
- Primary Topic
- Mental Health via Writing
- Type
- article
- Field-Weighted Citation Impact
- 1.93