Deep learning and large language models (LLMs) are being investigated to detect depression from social media data. However, traditional deep learning methods such as Long Short-Term Memory (LSTM) are limited by their reliance on feature engineering, resulting in time consuming and hard to capture complex patterns arising from media data. To address this issue, we propose a novel deep-learning approach that is based on Bidirectional Encoder Representations from Transformers (BERT) framework with Transformer Regression (called BERT-TR) for depression severity prediction from media data. In addition, we employ different techniques such as under sampling, oversampling and weighted loss function to ensure robust and generalizable predictions for depression. The data from X, derived from clinical assessments aligned with the fifth edition of the (DSM-5-TR) criteria, is used for investigations. We theoretically and empirically show that BERT-TR can predict depression severity from media data effectively and efficiently.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

A Deep-Learning Approach Based on BERT Model for Depression Severity Prediction

  • Ngonidzashe Mathew Kanyangarara,
  • Madhu Chetty,
  • Jiangang Ma

摘要

Deep learning and large language models (LLMs) are being investigated to detect depression from social media data. However, traditional deep learning methods such as Long Short-Term Memory (LSTM) are limited by their reliance on feature engineering, resulting in time consuming and hard to capture complex patterns arising from media data. To address this issue, we propose a novel deep-learning approach that is based on Bidirectional Encoder Representations from Transformers (BERT) framework with Transformer Regression (called BERT-TR) for depression severity prediction from media data. In addition, we employ different techniques such as under sampling, oversampling and weighted loss function to ensure robust and generalizable predictions for depression. The data from X, derived from clinical assessments aligned with the fifth edition of the (DSM-5-TR) criteria, is used for investigations. We theoretically and empirically show that BERT-TR can predict depression severity from media data effectively and efficiently.