Abstract
Background: Only a fraction of the produced social media data is usable in mental health assessment. So the problem of sufficient training data for deep learning approaches arises. Data sufficiency can be presented in terms of number of users or the number of posts per user.
Objective: We examine the data need of machine learning and deep learning models for a practical system and let researcher choose best fitting models depending on the dataset type available with them. We perform distinct experiments to find the effect of these issues on depression classification by various approaches.
Methods: We explored various machine learning and deep learning techniques on various data set versions, taken from Twitter and Reddit, with varying numbers of users and posts per user. Diagnosed and control users are taken in different ratios to assess the impact of an imbalanced dataset.
Results: The results reveal that SVM achieved 68% accuracy in depression classification for 70 users each from diagnosed and control group. It decreases for 150 users from each group, but then regains performance for 350 and 550 users from each group. Whereas Naive Bayes got 64% for the same dataset fragment (1). We observed that accuracy decreases for 150 diagnosed users, but then regains performance for 350 and 550 users. However from deep learning algorithms, HAN and BiLSTM perform better, compared to other algorithms, as the imbalance ratio increases.
Conclusion: We found, mainly, that classification accuracy increases with the number of users, number of posts per user and imbalance in the number of diagnosed versus control users. We also found that posts from Reddit have better accuracy compared to tweets.
Keywords: Mental health, neural network, depression, word embedding, machine learning, psycholinguistic.
Graphical Abstract
[http://dx.doi.org/10.1109/CloudCom.2014.69]
[http://dx.doi.org/10.1016/j.trc.2016.12.008]
[http://dx.doi.org/10.1145/2830657.2830660]
[http://dx.doi.org/10.1007/s11882-012-0299-7] [PMID: 22976493]
[http://dx.doi.org/10.1016/j.imu.2017.12.005]
[http://dx.doi.org/10.1016/j.amepre.2013.10.008] [PMID: 24439350]
[http://dx.doi.org/10.2139/ssrn.3511013]
[http://dx.doi.org/10.1016/B978-0-12-822468-7.00010-9]
[http://dx.doi.org/10.3390/healthcare10020291]
[http://dx.doi.org/10.1007/978-3-319-06269-3_10]
[http://dx.doi.org/10.1038/s41386-018-0247-x] [PMID: 30356094]
[http://dx.doi.org/10.1109/ICME.2014.6890213]
[http://dx.doi.org/10.1145/2647868.2654945]
[http://dx.doi.org/10.1038/srep45141] [PMID: 28127051]
[http://dx.doi.org/10.1109/TKDE.2018.2885515]
[http://dx.doi.org/10.1166/jmihi.2020.3169]
[http://dx.doi.org/10.2196/28754] [PMID: 34269683]
[http://dx.doi.org/10.18653/v1/W18-0609]
[http://dx.doi.org/10.18653/v1/D19-6208]
In: K, Azzopardi, B. Stein, N. Fuhr, P. Mayr, C. Hauff, D. Hiemstra, (eds) Advances in Information Retrieval. ECIR 2019. Lecture Notes in Computer Science, vol 11438. Springer, Cham 2019. [http://dx.doi.org/10.1007/978-3-030-15719-7_48]
[http://dx.doi.org/10.1109/BIBM.2018.8621230]
[http://dx.doi.org/10.18653/v1/W18-0607]
[http://dx.doi.org/10.2196/19824] [PMID: 34383688]
[http://dx.doi.org/10.3115/v1/W14-3207]
[http://dx.doi.org/10.1007/978-3-319-98932-7_30]
[http://dx.doi.org/10.1007/978-3-319-44564-9_3]