文献类型：期刊文献

英文题名：A hybrid depression detection model and correlation analysis for social media based on attention mechanism

作者：Liu, Jiacheng[1];Chen, Wanzhen[2];Wang, Liangxu[3];Ding, Fangyikuang[2]

机构：[1]East China Univ Sci & Technol, Sch Informat Sci & Engn, Shanghai 200237, Peoples R China;[2]East China Univ Sci & Technol, Sch Social & Publ Adm, Shanghai 200237, Peoples R China;[3]Shanghai Normal Univ, Sch Environm & Geog Sci, Shanghai 200234, Peoples R China

年份：2024

卷号：15

期号：7

起止页码：2631

外文期刊名：INTERNATIONAL JOURNAL OF MACHINE LEARNING AND CYBERNETICS

收录：;EI(收录号：20235215276588);WOS:【SCI-EXPANDED(收录号:WOS:001129839600001)】；

基金：The authors would like to express gratitude to all individuals who contribute open-source materials on the Internet, with special thanks to the providers of the WU3D dataset used in this article, which is available for access at https://github.com/aidenwang9867/Weibo-User-Depression-Detection-Dataset. We also extend our gratitude to the providers of the initial pre-trained models. The pre-trained models used in this paper can be found at https://github.com/brightmart/albert_zh.

语种：英文

外文关键词：Depression detection; Weibo; Attention mechanism; Individual interpretability visualization

摘要：As a serious mental illness, depression can be extremely harmful to the physical and mental health of individuals. However, detecting depression can be challenging due to the reluctance of the depressed to actively express themselves. Fortunately, in modern society, online social platforms provide an opportunity for genuine self-expression in our daily lives. Leveraging the power of social data, we can identify potentially depressed users efficiently and accurately. This lays a strong foundation for subsequent interventions. In this paper, we propose a hybrid model that comprehensively considers the features and post texts of users by utilizing a simplified multi-head attention mechanism for detecting user depression. Compared to traditional models, such as Decision Tree and Random Forest, the simplified multi-head attention mechanism achieves higher classification accuracy while offering enhanced interpretability at the individual level. To verify the validity of our model, we apply it to the Weibo User Depression Detection Dataset (WU3D) containing approximately 1,150,000 posts from around 21,000 users. The dataset has been classified by human experts as either depressed or not. The results show that our model both explains the association of each feature of a single user well and achieves better performance than traditional methods. Notably, the final F1-score of our new model on the test set is 0.9473. Furthermore, by visualizing attention scores, we conduct a correlation analysis between different features and results for individual users, which can facilitate further analysis of the behavior and psychological characteristics of specific depressed users by professionals.