Sentiment Informed Sentence BERT-Ensemble Algorithm for Depression Detection
arXiv:2409.13713 · doi:10.3390/bdcc8090112
Abstract
The World Health Organisation (WHO) revealed approximately 280 million people in the world suffer from depression. Yet, existing studies on early-stage depression detection using machine learning (ML) techniques are limited. Prior studies have applied a single stand-alone algorithm, which is unable to deal with data complexities, prone to overfitting, and limited in generalization. To this end, our paper examined the performance of several ML algorithms for early-stage depression detection using two benchmark social media datasets (D1 and D2). More specifically, we incorporated sentiment indicators to improve our model performance. Our experimental results showed that sentence bidirectional encoder representations from transformers (SBERT) numerical vectors fitted into the stacking ensemble model achieved comparable F1 scores of 69% in the dataset (D1) and 76% in the dataset (D2). Our findings suggest that utilizing sentiment indicators as an additional feature for depression detection yields an improved model performance, and thus, we recommend the development of a depressive term corpus for future work.
References in corpus (6)
- Utilizing Neural Networks and Linguistic Metadata for Early Detection of Depression Indications in Text Sequences
- A Text Classification Framework for Simple and Effective Early Depression Detection Over Social Media Streams
- Calibration of Transformer-based Models for Identifying Stress and Depression in Social Media
- The Use of a Large Language Model for Cyberbullying Detection
- Exploring Social Media for Early Detection of Depression in COVID-19 Patients
- Data set creation and empirical analysis for detecting signs of depression from social media postings