Machine Learning Aided Depression Detection in Community Dwellers
by Vijay Kumar, Muskan Khajuria * , Anshu Singh
Department of Computer Science and Engineering, National Institute of Technology Hamirpur, Hamirpur, 177005, India
* Author to whom correspondence should be addressed.
Journal of Engineering Research and Sciences, Volume 1, Issue 5, Page # 17-24, 2022; DOI: 10.55708/js0105002
Keywords: Depression, Machine Learning, mental health, SMOTE, ADABoost, SelectKBest
Received: 14 February 2022, Revised: 04 April 2022, Accepted: 11 April 2022, Published Online: 12 May 2022
APA Style
Kumar, V., Khajuria, M., & Singh, A. (2022, May). Machine Learning Aided Depression Detection in Community Dwellers. Journal of Engineering Research and Sciences, 1(5), 17–24. https://doi.org/10.55708/js0105002
Chicago/Turabian Style
Kumar, Vijay, Muskan Khajuria, and Anshu Singh. “Machine Learning Aided Depression Detection in Community Dwellers.” Journal of Engineering Research and Sciences 1, no. 5 (May 2022): 17–24. https://doi.org/10.55708/js0105002.
IEEE Style
V. Kumar, M. Khajuria, and A. Singh, “Machine Learning Aided Depression Detection in Community Dwellers,” Journal of Engineering Research and Sciences, vol. 1, no. 5, pp. 17–24, May 2022, doi: 10.55708/js0105002.
Depression is a mental condition that can have serious negative effects on an individual’s thoughts and nd health problems that could lead to grave heart diseases. Depression detection has become necessary in community dwellers considering the lifestyle being followed. Here we use NHANES dataset to compare the performance of various machine learning algorithms in depression detection. The 2015 dataset was used to train the models and testing was done on data from 2017 to analyze the robustness of the model. Feature extraction was also performed on the dataset to observe relevant features. It was found that ADABoost used wit ic Minority Oversampling Technique (SMOTE) gave the best test results in terms of F1 score.
- J. A. Fain, “Nhanes: Use of a free public data set”, The Diabetes Educa- tor, vol. 43, no. 2, pp. 151–151, 2017, doi:10.1177/0145721717698651, pMID: 28340543.
- N. Chawla, K. Bowyer, L. Hall, W. Kegelmeyer, “Smote: Synthetic minority over-sampling technique”, J. Artif. Intell. Res. (JAIR), vol. 16, pp. 321–357, 2002, doi:10.1613/jair.953.
- M. Hossin, S. M.N, “A review on evaluation metrics for data classifica- tion evaluations”, International Journal of Data Mining & Knowledge Man- agement Process, vol. 5, pp. 01–11, 2015, doi:10.5121/ijdkp.2015.5201.
- C. Goutte, E. Gaussier, “A probabilistic interpretation of preci- sion, recall and f-score, with implication for evaluation”, Lec- ture Notes in Computer Science, vol. 3408, pp. 345–359, 2005, doi: 10.1007/978-3-540-31865-1_25.
- A. P. Bradley, “The use of the area under the roc curve in the evaluation of machine learning algorithms”, Pattern Recognition, vol. 30, no. 7, pp. 1145–1159, 1997, doi:https://doi.org/10.1016/S0031-3203(96) 00142-2.
- D. Su, X. Zhang, K. He, Y. Chen, “Use of machine learning ap- proach to predict depression in the elderly in china: A longitudinal study”, Journal of Affective Disorders, vol. 282, pp. 289–298, 2021, doi: 10.1016/j.jad.2020.12.160.
- C. W. Choi J, Choi J, “redicting depression among community residing older adults: A use of machine learning approch”, 2018.
- H. Jin, S. Wu, P. Di Capua, “Development of a clinical forecasting model to predict comorbid depression among diabetes patients and an application in depression screening policy making”, Preventing Chronic Disease, vol. 12, p. E142, 2015, doi:10.5888/pcd12.150047.
- A. Sau, I. Bhakta, “Artificial neural network (ann) model to predict depression among geriatric population at a slum in kolkata, india”, Journal of Clinical and Diagnostic Research, vol. 11, pp. VC01–VC04, 2017, doi:10.7860/JCDR/2017/23656.9762.
- A. Sharma, W. Verbeke, “Improving diagnosis of depression with xgboost machine learning model and a large biomarkers dutch dataset (n = 11,081)”, Frontiers in Big Data, vol. 3, 2020, doi: 10.3389/fdata.2020.00015.
- S.-E. Cho, Z. W. Geem, K.-S. Na, “Predicting depression in community dwellers using a machine learning algorithm”, Diagnostics, vol. 11, p. 1429, 2021, doi:10.3390/diagnostics11081429.
- E. Victor, Z. M. Aghajan, A. Sewart, R. Christian, “Detecting depres- sion using a framework combining deep multimodal neural networks t automated evaluation”, Psychological Assessment,
1, 2019, doi:10.1037/pas0000724. - J. Dipnall, J. Pasco, M. Berk, L. Williams, S. Dodd, F. Jacka, “Into the bowels of depression: Unravelling medical symptoms associated with depression by applying machine-learning techniques to a community based population sample”, PLOS ONE, vol. 11, p. e0167055, 2016, doi:10.1371/journal.pone.0167055.
- M. Zulfiker, N. Ety, A. A. Biswas, T. Nazneen, M. Uddin, “An in-depth analysis of machine learning approaches to predict depression”, Current Research in Behavioral Sciences, vol. 2, p. 100044, 2021, doi: 10.1016/j.crbeha.2021.100044.
- K. Vaishnavi, U. N. Kamath, B. A. Rao, N. V. S. Reddy, “Predicting mental health illness using machine learning algorithms”, Journal nce Series, vol. 2161, no. 1, p. 012021, 2022, doi:
88/1742-6596/2161/1/012021.