Back to Search Start Over

Harnessing the Power of Hugging Face Transformers for Predicting Mental Health Disorders in Social Networks

Authors :
Pourkeyvan, Alireza
Safa, Ramin
Sorourkhah, Ali
Source :
IEEE Access, vol. 12, pp. 28025-28035, 2024
Publication Year :
2023

Abstract

Early diagnosis of mental disorders and intervention can facilitate the prevention of severe injuries and the improvement of treatment results. Using social media and pre-trained language models, this study explores how user-generated data can be used to predict mental disorder symptoms. Our study compares four different BERT models of Hugging Face with standard machine learning techniques used in automatic depression diagnosis in recent literature. The results show that new models outperform the previous approach with an accuracy rate of up to 97%. Analyzing the results while complementing past findings, we find that even tiny amounts of data (like users' bio descriptions) have the potential to predict mental disorders. We conclude that social media data is an excellent source of mental health screening, and pre-trained models can effectively automate this critical task.<br />Comment: 19 pages, 5 figures

Details

Database :
arXiv
Journal :
IEEE Access, vol. 12, pp. 28025-28035, 2024
Publication Type :
Report
Accession number :
edsarx.2306.16891
Document Type :
Working Paper
Full Text :
https://doi.org/10.1109/ACCESS.2024.3366653