A 5-stage hierarchical mental-health text classifier covering 8 classes: Normal, Depression, Suicidal, Anxiety, Stress, Bipolar, Personality Disorder, Directed Aggression.

This is the Deep Dive tier of the Vibecheck project (Helwan University). The Quick Vibe counterpart is itsLu/mentalbert-v6-flat.

text → Stage 0  (Cardiff RoBERTa, DA binary)
           ↓
       Stage 1A (MentalBERT, Suicidal binary)
           ↓
       Stage 1B (MentalBERT, Normal vs Distress)
           ↓
       Stage 2  (MentalBERT, 5-class Anxiety/Bipolar/Dep/PD/Stress)
           ↓
       Stage 3  (Longformer, Depression vs Suicidal re-scorer)

Each stage is fine-tuned on the full V6 dataset, carved per stage. Routing thresholds are calibrated on the validation set via a joint t1A × t3 grid sweep producing two operating points:

mode is "balanced" (default) or "safety".

mohamedasem318/mental-health-dataset-extended-v6 — ~89k rows from 7 sources (cssrs, jigsaw17, civil_comments, kaggle_bpd, kaggle, huggingface, swmh). V6 replaces V5's OLID-sourced Directed Aggression class with Jigsaw 2017 + Civil Comments threat-labelled data — the change targets real-world detection of explicit violent intent (e.g., "I want to kill my friend"), which OLID's Twitter-insult distribution did not cover. Stratified 70/10/20 split, random_state=42. Per-source reliability constants are recorded in config.json but were not applied to the loss (plain class-weighted CE only — a prior attempt with a per-sample-weighted wrapper destabilised Stage 0 training under fp16/AMP).

If you use this model, please cite the Vibecheck project (Helwan University final-year project, 2026).