arrow
返回

Depression Classification From Tweets Using Small Deep Transfer Learning Language Models

delete2022-01-01
delete13
delete
OA
AI
M
Muhammad Rizwan
M
Muhammad Faheem Mushtaq
U
Urooj Akram
A
Arif Mehmood
I
Imran Ashraf *
B
Benjamín Sahelices *
DOI:10.1109/ACCESS.2022.3223049delete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

En 中文
Depression detection from social media texts such as Tweets or Facebook comments could be very beneficial as early detection of depression may even avoid extreme consequences of long-term depression i.e. suicide. In this study, depression intensity classification is performed using a labeled Twitter dataset. Further, this study makes a detailed performance evaluation of four transformer-based pre-trained small language models, particularly those having less than 15 million tunable parameters i.e. Electra Small Generator (ESG), Electra Small Discriminator (ESD), XtremeDistil-L6 (XDL) and Albert Base V2 (ABV) for classification of depression intensity using Tweets. The models are fine-tuned to get the best performance by applying different hyperparameters. The models are tested by classification of depression intensity of labeled tweets for three label classes i.e. 'severe', 'moderate', and 'mild' by downstream fine-tuning the parameters. Evaluation metrics such as accuracy, F1, precision, recall, and specificity are calculated to evaluate the performance of the models. Comparative analysis of these models is also done with a moderately larger model i.e. DistilBert which has 67 million tunable parameters for the same task with the same experimental settings. Results indicate that ESG outperforms all other models including DistilBert due to its better deep contextualized text representation as it gets the best F1 score of 89% with comparatively less training time. Further optimization of ESG is also proposed to make it suitable for low-powered devices. This study helps to achieve better classification performance of depression detection as well as to choose the best language model in terms of performance and less training time for Twitter-related downstream NLP tasks.
Keyword:
Depression classification
transfer learning
transformer language models
public health

期刊

IEEE Access 封面图
IEEE Access
IF:
3.6
论文数:
9.8W
被引数:
29.4W

机构

Y
Yeungnam University
学者数:
1.0W
论文数: 1.3W
被引数: 1.4W
U
Universidad de Valladolid
学者数:
8.1K
论文数: 6.7K
被引数: 5.9K
I
Islamia University of Bahawalpur
学者数:
4.5K
论文数: 3.6K
被引数: 5.3K
学者 查看更多机构
引用论文

引用论文

299. The constitution of ascorbic acid
err1933-01-01
err0
PREAI
errR. W. Herbert; E. L. Hirst; E. G. V. Percival; R. J. W. Reynolds; F. Smith
err分享
err收藏
A Systematic review of the validity of screening depression through Facebook, Twitter, Instagram, and Snapchat
err2021-05-01
err40
PREAI
errKim, Jiin; Uddin, Zara A.; Lee, Yena; Nasri, Flora; Gill, Hartej; Subramanieapillai, Mehala; Lee, Renna; Udovica, Aleksandra; Lee Phan; Lui, Leanna; Iacobucci, Michelle; Mansur, Rodrigo B.; Rosenblat, Joshua D.; McIntyre, Roger S.
err分享
err收藏
err分享
err收藏
Self-Harm, Suicidal Behaviours, and Cyberbullying in Children and Young People: Systematic Review儿童和青少年的自我伤害,自杀行为和网络欺凌: 系统综述
err2018-04-19
err328
errOAAI
errJohn, Ann; Glendenning, Alexander Charles; Marchant, Amanda; Montgomery, Paul; Stewart, Anne; Wood, Sophie; Lloyd, Keith; Hawton, Keith
err分享
err收藏
学者 查看更多内容