Improving Diagnosis of Depression With XGBOOST Machine Learning Model and a Large Biomarkers Dutch Dataset (n = 11,081)

Abstract

Machine Learning has been on the rise and healthcare is no exception to that. In healthcare, mental health is gaining more and more space. The diagnosis of mental disorders is based upon standardized patient interviews with defined set of questions and scales which is a time consuming and costly process. Our objective was to apply the machine learning model and to evaluate to see if there is predictive power of biomarkers data to enhance the diagnosis of depression cases. In this research paper, we aimed to explore the detection of depression cases among the sample of 11,081 Dutch citizen dataset. Most of the earlier studies have balanced datasets wherein the proportion of healthy cases and unhealthy cases are equal but in our study, the dataset contains only 570 cases of self-reported depression out of 11,081 cases hence it is a class imbalance classification problem. The machine learning model built on imbalance dataset gives predictions biased toward majority class hence the model will always predict the case as no depression case even if it is a case of depression. We used different resampling strategies to address the class imbalance problem. We created multiple samples by under sampling, over sampling, over-under sampling and ROSE sampling techniques to balance the dataset and then, we applied machine learning algorithm “Extreme Gradient Boosting” (XGBoost) on each sample to classify the mental illness cases from healthy cases. The balanced accuracy, precision, recall and F1 score obtained from over-sampling and over-under sampling were more than 0.90.

A Machine Learning Prediction Model of Respiratory Failure Within 48 Hours of Patient Admission for COVID-19: Model Development and Validation (Preprint)

“…We evaluated three machine learning models: XGBoost, XGBoost + SMOTEENN, and logistic regression [38][39][40]. XGBoost is widely used due to its high efficiency and predictability, and it has been used to predict health care outcomes in patients with [41,42] and without [43][44][45] COVID-19. In our study, XGBoost was the most accurate prediction model, with an accuracy of 0.919 (SD 0.028) and precision of 0.521 (SD 0.329; Figure 1), similar to the findings of another study that examined combined outcomes [46].…”

Prediction of acute suicidal ideation in young adults using multi-dimensional scales: A graph neural network approach

“…Next, the nal ensemble GIN model was obtained using the three best models. Speci cally, the sigmoid prediction scores from the best models were averaged to obtain the nal prediction score of the ensemble model, which is a process known as "soft voting" 51,52 .…”

Judging Mental Health Disorders Using Decision Tree Models

“…Thus, it is measured, what is a time bounding and expensive procedure. The purpose of doing so, is to make an application of the machine learning prototype and assessment of discovering whether a speculative ability of patient data is there for augmenting the detection of the events related to dejection [6,10,47,48]. In this research, there is a target to create a podium which will be able to categorize the dataset into reappearance and non-reappearance.…”