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

基于多任务深度学习的新浪微博用户抑郁检测

A Multitask Deep Learning Approach for User Depression Detection on Sina Weibo

College of Cybersecurity(网络空间安全学院);Sichuan University(四川大学)

Yiding Wang, Zhenyi Wang, Chenghao Li, Yilin Zhang, Haizhou Wang

摘要:

近年来,由于抑郁症的精神负担,危及生命的人数正在迅速增加。在线社交网络(OSN)为研究人员提供了另一种检测抑郁症患者的视角。然而,基于机器学习的抑郁症检测的现有研究仍然留下相对较低的分类性能,这表明在其特征工程的改进方面存在显着的改进潜力。在本文中,我们在新浪微博(一个领先的OSN,在中国社区中拥有最多的活跃用户)上手动构建了一个大型数据集,即微博用户抑郁检测数据集(WU 3D)。其中包括2万多名正常用户和1万多名抑郁用户,这两种用户都是由专业人员手动标记并重新检查的。通过分析用户的文本、社交行为和发布的图片,总结并提出了十个统计特征。同时,使用流行的预训练模型XLNet提取基于文本的单词特征。此外,提出了一种新的深度神经网络分类模型,即FusionNet(FN),并同时使用上述提取的特征进行训练,这些特征被视为多个分类任务。实验结果表明,FusionNet在测试数据集上获得了最高的F1-Score,为0.9772。与现有的研究相比,我们提出的方法具有更好的分类性能和不平衡的训练样本的鲁棒性。我们的工作还提供了一种新的方法来检测其他OSN平台上的抑郁症。

英文摘要:

In recent years, due to the mental burden of depression, the number of people who endanger their lives has been increasing rapidly. The online social network (OSN) provides researchers with another perspective for detecting individuals suffering from depression. However, existing studies of depression detection based on machine learning still leave relatively low classification performance, suggesting that there is significant improvement potential for improvement in their feature engineering. In this paper, we manually build a large dataset on Sina Weibo (a leading OSN with the largest number of active users in the Chinese community), namely Weibo User Depression Detection Dataset (WU3D). It includes more than 20,000 normal users and more than 10,000 depressed users, both of which are manually labeled and rechecked by professionals. By analyzing the user's text, social behavior, and posted pictures, ten statistical features are concluded and proposed. In the meantime, text-based word features are extracted using the popular pretrained model XLNet. Moreover, a novel deep neural network classification model, i.e. FusionNet (FN), is proposed and simultaneously trained with the above-extracted features, which are seen as multiple classification tasks. The experimental results show that FusionNet achieves the highest F1-Score of 0.9772 on the test dataset. Compared to existing studies, our proposed method has better classification performance and robustness for unbalanced training samples. Our work also provides a new way to detect depression on other OSN platforms.

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

↑