CoolFace
Apppublic

absiddiq735/Mental_Health_Risk_Analysis_Detector

sourceHugging Facemitupdated 3mo agoView on Hugging Face
0likes
App README

๐Ÿง  Mental Health Sentiment & Depression Severity Analyser

Thesis Project โ€” Multi-Task DistilBERT for Joint Sentiment & Depression Severity Detection

Model Architecture

ComponentDetail
Base encoderdistilbert-base-uncased
MAX_LEN256 tokens
Sentiment head768โ†’512โ†’256โ†’3 (LayerNorm + GELU, 3-layer MLP)
Cross-task gate2-layer MLP 3โ†’128โ†’256 (Sigmoid)
Severity branch768โ†’512โ†’256 (LayerNorm + Residual)
Severity refine256โ†’256โ†’256 (2-layer + Residual)
CORAL layer4-class ordinal (3 boundaries)
ThresholdsLearnable per-boundary CORAL thresholds
LossUncertainty weighting (Kendall et al., 2018) + ordinal smoothing

Labels

Sentiment (3-class): negative ยท neutral ยท positive

Severity (4-class, PHQ-9): minimum ยท mild ยท moderate ยท severe

Training Datasets (6 sources)

  • โ€”Combined_Data.csv (sentiment for mental health)
  • โ€”Redditdepressiondataset.csv
  • โ€”combined.json (DepressionEmo)
  • โ€”reddit-depression-cleaned (HuggingFace)
  • โ€”mentalhealthcounseling_conversations (HuggingFace)
  • โ€”Suicide_Detection.csv

Disclaimer

Research prototype for academic/thesis purposes only. Not a clinical diagnostic tool.