Detection of child depression using machine learning methods
Article
Article Title | Detection of child depression using machine learning methods |
---|---|
ERA Journal ID | 39745 |
Article Category | Article |
Authors | Haque, Umme Marzia (Author), Kabir, Enamul (Author) and Khanam, Rasheda (Author) |
Journal Title | PLoS One |
Journal Citation | 16 (12), pp. 1-13 |
Article Number | e0261131 |
Number of Pages | 13 |
Year | 2021 |
Publisher | Public Library of Science (PLoS) |
Place of Publication | United States |
ISSN | 1932-6203 |
Digital Object Identifier (DOI) | https://doi.org/10.1371/journal.pone.0261131 |
Web Address (URL) | https://journals.plos.org/plosone/article?id=10.1371/journal.pone.0261131 |
Abstract | Background: Mental health problems, such as depression in children have far-reaching negative effects on child, family and society as whole. It is necessary to identify the reasons that contribute to this mental illness. Detecting the appropriate signs to anticipate mental illness as depression in children and adolescents is vital in making an early and accurate diagnosis to avoid severe consequences in the future. There has been no research employing machine learning (ML) approaches for depression detection among children and adolescents aged 4–17 years in a precisely constructed high prediction dataset, such as Young Minds Matter (YMM). As a result, our objective is to 1) create a model that can predict depression in children and adolescents aged 4–17 years old, 2) evaluate the results of ML algorithms to determine which one outperforms the others and 3) associate with the related issues of family activities and socioeconomic difficulties that contribute to depression. Methods: The YMM, the second Australian Child and Adolescent Survey of Mental Health and Wellbeing 2013–14 has been used as data source in this research. The variables of yes/no value of low correlation with the target variable (depression status) have been eliminated. The Boruta algorithm has been utilized in association with a Random Forest (RF) classifier to extract the most important features for depression detection among the high correlated variables with target variable. The Tree-based Pipeline Optimization Tool (TPOTclassifier) has been used to choose suitable supervised learning models. In the depression detection step, RF, XGBoost (XGB), Decision Tree (DT), and Gaussian Naive Bayes (GaussianNB) have been used. Results: Unhappy, nothing fun, irritable mood, diminished interest, weight loss/gain, insomnia or hypersomnia, psychomotor agitation or retardation, fatigue, thinking or concentration problems or indecisiveness, suicide attempt or plan, presence of any of these five symptoms have been identified as 11 important features to detect depression among children and adolescents. Although model performance varied somewhat, RF outperformed all other algorithms in predicting depressed classes by 99% with 95% accuracy rate and 99% precision rate in 315 milliseconds (ms). Conclusion: This RF-based prediction model is more accurate and informative in predicting child and adolescent depression that outperforms in all four confusion matrix performance measures s well as execution duration. |
Keywords | children; depression; mental health; machine learning methods |
Related Output | |
Is part of | Enhancing early detection of psychological disorders in children and adolescents using ML: a comprehensive mental health assessment |
ANZSRC Field of Research 2020 | 420699. Public health not elsewhere classified |
Public Notes | This article is part of a UniSQ Thesis by publication. See Related Output. |
Byline Affiliations | School of Sciences |
School of Business | |
Institution of Origin | University of Southern Queensland |
https://research.usq.edu.au/item/q6z39/detection-of-child-depression-using-machine-learning-methods
Download files
247
total views106
total downloads5
views this month1
downloads this month