Predicting Anxiety, Depression and Stress in Modern Life using Machine Learning Algorithms

Abstract In the fast-paced modern world, psychological health issues like anxiety, depression and stress have become very common among the masses. In this paper, predictions of anxiety, depression and stress were made using machine learning algorithms. In order to apply these algorithms, data were collected from employed and unemployed individuals across different cultures and communities through the Depression, Anxiety and Stress Scale questionnaire (DASS 21). Anxiety, depression and stress were predicted as occurring on five levels of severity by five different machine learning algorithms – because these are highly accurate, they are particularly suited to predicting psychological problems. After applying the different methods, it was found that classes were imbalanced in the confusion matrix. Thus, the f1 score measure was added, which helped identify the best accuracy model among the five applied algorithms as the Random Forest classifier. Furthermore, the specificity parameter revealed that the algorithms were also especially sensitive to negative results.

[1]  D. Leightley,et al.  Identifying probable post-traumatic stress disorder: applying supervised machine learning to data from a UK military cohort , 2018, Journal of Mental Health.

[2]  Jingcheng Du,et al.  Extracting psychiatric stressors for suicide from social media using deep learning , 2018, BMC Medical Informatics and Decision Making.

[3]  Dipti Prasad Mukherjee,et al.  Improved Random Forest for Classification , 2018, IEEE Transactions on Image Processing.

[4]  Christophe Giraud-Carrier,et al.  Validating Machine Learning Algorithms for Twitter Data Against Established Measures of Suicidality , 2016, JMIR mental health.

[5]  Thomas G. Dietterich Multiple Classifier Systems , 2000, Lecture Notes in Computer Science.

[6]  Mario Chica-Olmo,et al.  An assessment of the effectiveness of a random forest classifier for land-cover classification , 2012 .

[7]  Zhi-Hua Zhou,et al.  Exploratory Under-Sampling for Class-Imbalance Learning , 2006, Sixth International Conference on Data Mining (ICDM'06).

[8]  Svetha Venkatesh,et al.  A Framework for Classifying Online Mental Health-Related Communities With an Interest in Depression , 2016, IEEE Journal of Biomedical and Health Informatics.

[9]  Arkaprabha Sau,et al.  Predicting anxiety and depression in elderly patients using machine learning technology , 2017 .

[10]  S. Sawang,et al.  Using the Depression Anxiety Stress Scale 21 (DASS-21) across cultures. , 2013, International journal of psychology : Journal international de psychologie.

[11]  Christopher M. Danforth,et al.  Forecasting the onset and course of mental illness with Twitter data , 2016, Scientific Reports.

[12]  R. Spitzer,et al.  The PHQ-9 , 2001, Journal of General Internal Medicine.