Interpretable Depression Detection from Social Media Text Using LLM-Derived Embeddings
Summary
An investigation evaluates interpretable depression detection from social media text using LLM-derived embeddings. It compares zero-shot LLMs with supervised classifiers trained on conventional text embeddings and LLM-generated summary embeddings across three tasks: binary depression classification, depression severity classification, and differential diagnosis (depression, PTSD, anxiety). The study used five social media-based depression datasets (MHB, CAMS, HelaDepDet, RMHD, DepressionEmo) and a general-domain dataset (AITA). Experiments show zero-shot LLMs excel in binary classification but struggle with fine-grained ordinal classifications like severity. Classifiers trained on LLM-generated summary embeddings, however, achieved competitive or superior performance compared to models using traditional text embeddings, particularly in severity classification (Logistic Regression with LLM summary embeddings achieved 58% accuracy). Zero-shot LLMs achieved 65% accuracy in differential diagnosis. This highlights the potential of hybrid approaches combining LLM generalization with interpretable classifiers.
Key takeaway
For Machine Learning Engineers developing mental health screening tools, consider integrating LLM-generated summary embeddings into your classification pipelines. While zero-shot LLMs perform well for binary depression detection, training supervised models on these context-aware summaries significantly improves accuracy for fine-grained tasks like severity assessment and differential diagnosis. This hybrid approach offers enhanced interpretability and more consistent predictions than relying solely on raw text embeddings or zero-shot LLMs for complex classifications.
Key insights
LLM-generated summary embeddings enhance traditional classifiers for nuanced mental health detection from social media, outperforming raw text embeddings.
Principles
- Zero-shot LLMs generalize well for binary classification.
- LLM-derived summaries improve classifier performance.
- Fine-grained ordinal tasks challenge zero-shot LLMs.
Method
Prompt an LLM (GPT-4o) to summarize a social media post's mental state, then extract a 768-dimensional embedding from this summary using all-mpnet-base-v2, and train traditional classifiers.
In practice
- Use LLM summaries for mental health feature extraction.
- Apply zero-shot LLMs for initial binary screening.
- Combine LLM summaries with Logistic Regression for severity.
Topics
- Depression Detection
- Large Language Models
- Social Media Analysis
- Clinical NLP
- Text Embeddings
- Mental Health Screening
Best for: NLP Engineer, AI Scientist, Machine Learning Engineer, Research Scientist
Related on AIssential
See Counsel's argued verdicts on the open AI decisions leaders are weighing →
Editorial summary, takeaway, and curation by AIssential. Original article published by cs.CL updates on arXiv.org.