arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2302.02759cs.CLcs.AIcs.LG
收藏

使用混合神经网络 SBERT-CNN 检测患有抑郁症的 Reddit 用户

Detecting Reddit Users with Depression Using a Hybrid Neural Network SBERT-CNN

University of Washington(华盛顿大学);Mayo Clinic(梅奥诊所)

Ziyi Chen, Ren Yang, Sunyang Fu, Nansu Zong, Hongfang Liu, Ming Huang

AI总结:

本研究提出结合预训练 SBERT 和 CNN 的混合深度学习模型,通过分析 Reddit 帖子自动识别抑郁症用户,在 SMHD 数据集上达到 0.86 的准确率和 F1 分数,优于现有方法。

摘要:

抑郁症是一种广泛存在的心理健康问题,影响着全球约 3.8% 的人口。它也是全球导致残疾的主要原因之一。最近,个人使用社交媒体平台(如 Reddit)表达他们的困难和健康问题(如抑郁症),并在在线社区中寻求其他用户的支持变得越来越普遍。这为通过解析数百万条帖子进行潜在干预,从而自动识别患有抑郁症的社交媒体用户提供了巨大机会。由于其易用性、高效处理以及在许多 NLP 任务上的最先进结果,深度学习方法已开始在机器学习和自然语言处理(NLP)领域占据主导地位。在这项工作中,我们提出了一种混合深度学习模型,它结合了预训练的句子 BERT(SBERT)和卷积神经网络(CNN),通过用户的 Reddit 帖子来检测患有抑郁症的个人。句子 BERT 用于学习每个帖子中语义信息的有意义表示。CNN 能够进一步转换这些嵌入,并实现用户行为模式的时间识别。我们利用自我报告的心理健康诊断(SMHD)数据训练并评估了该模型识别患有抑郁症的 Reddit 用户的性能。该混合深度学习模型达到了 0.86 的准确率和 0.86 的 F1 分数,并优于文献中其他机器学习模型记录的最先进结果(F1 分数为 0.79)。结果表明了混合模型识别抑郁症患者的可行性。尽管该混合模型被验证可以通过 Reddit 帖子检测抑郁症,但它可以很容易地进行微调,并应用于其他文本分类任务和不同的临床应用。

英文摘要:

Depression is a widespread mental health issue, affecting an estimated 3.8% of the global population. It is also one of the main contributors to disability worldwide. Recently it is becoming popular for individuals to use social media platforms (e.g., Reddit) to express their difficulties and health issues (e.g., depression) and seek support from other users in online communities. It opens great opportunities to automatically identify social media users with depression by parsing millions of posts for potential interventions. Deep learning methods have begun to dominate in the field of machine learning and natural language processing (NLP) because of their ease of use, efficient processing, and state-of-the-art results on many NLP tasks. In this work, we propose a hybrid deep learning model which combines a pretrained sentence BERT (SBERT) and convolutional neural network (CNN) to detect individuals with depression with their Reddit posts. The sentence BERT is used to learn the meaningful representation of semantic information in each post. CNN enables the further transformation of those embeddings and the temporal identification of behavioral patterns of users. We trained and evaluated the model performance to identify Reddit users with depression by utilizing the Self-reported Mental Health Diagnoses (SMHD) data. The hybrid deep learning model achieved an accuracy of 0.86 and an F1 score of 0.86 and outperformed the state-of-the-art documented result (F1 score of 0.79) by other machine learning models in the literature. The results show the feasibility of the hybrid model to identify individuals with depression. Although the hybrid model is validated to detect depression with Reddit posts, it can be easily tuned and applied to other text classification tasks and different clinical applications.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑