Back to Search Start Over

Sentiment Informed Sentence BERT-Ensemble Algorithm for Depression Detection.

Authors :
Ogunleye, Bayode
Sharma, Hemlata
Shobayo, Olamilekan
Source :
Big Data & Cognitive Computing; Sep2024, Vol. 8 Issue 9, p112, 16p
Publication Year :
2024

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. [ABSTRACT FROM AUTHOR]

Details

Language :
English
ISSN :
25042289
Volume :
8
Issue :
9
Database :
Complementary Index
Journal :
Big Data & Cognitive Computing
Publication Type :
Academic Journal
Accession number :
180017165
Full Text :
https://doi.org/10.3390/bdcc8090112