Table 7.
Binary classification scores using all posts of 70 depressed users and 5 of their matched control usersa.
|
|
Positive predictive value, mean (SD) | Sensitivity, mean (SD) | F1-score, mean (SD) |
| SVMb using TF-IDFc | 0.400 (N/Ad) | 0.029 (N/A) | 0.053 (N/A) |
| SVM using word embeddings | 0.309 (N/A) | 0.471 (N/A) | 0.372 (N/A) |
| SVM using TF-IDF and word embeddings | 0.200 (N/A) | 0.014 (N/A) | 0.027 (N/A) |
| BERTe LMf | 0.615 (0.028) | 0.290 (0.022) | 0.379 (0.017) |
| ALBERTg LM | 0.555 (0.030) | 0.281 (0.009) | 0.354 (0.006) |
| BioBERT LM | 0.627 (0.034) | 0.252 (0.021) | 0.331 (0.027) |
| Longformer LM | 0.624 (0.108) | 0.286 (0.038) | 0.363 (0.059) |
| MentalBERT LM | 0.572 (0.002) | 0.329 (0.043) | 0.400 (0.040) |
| MentalRoBERTa LM | 0.562 (0.027) | 0.343 (0.000) | 0.419 (0.010) |
| Naive baseline—all depression | 0.167 (N/A) | 1.000 (N/A) | 0.286 (N/A) |
aLanguage model experiments were run 3 times each, therefore both mean and SD scores are provided.
bSVM: support vector machine.
cTF-IDF: term frequency–inverse document frequency.
dN/A: not applicable.
eBERT: Bidirectional Encoder Representations from Transformers.
fLM: language model.
gALBERT: A Lite Bidirectional Encoder Representations from Transformers.