Related Experiment Video
Updated: Sep 20, 2025
A Machine Learning Approach to Design an Efficient Selective Screening of Mild Cognitive Impairment
Published on: January 11, 2020
Application of machine learning algorithms to identify risk factors for depression in type 2 diabetes mellitus
Yu-Wen Su1, Wayne Huey-Herng Sheu2,3,4, Chii-Min Hwu3
1Graduate Institute of Biomedical Electronics and Bioinformatics, National Taiwan University, Taipei, Taiwan, ROC.
Background:
We analyzed variables reported during routine clinical practice using a registrational database to estimate risk factors for depression in people with type 2 diabetes mellitus.
Methods:
A Patient Health Questionnaire (PHQ-9) score of 15 was selected as the cut-off for clinically meaningful depression. Missing data was either filled in with a median value, the k -nearest neighbors' method, or the entire variable was removed. Logistic regression, random forest, and decision tree machine learning models were used to decide which factors were most relevant to depression. The accuracy of each algorithm was evaluated with a testing set.
Results:
When all variables were included in the logistic regression model, the area under the receiver operating characteristic curve was 0.81. In the random forest model, the most important factor was quality of life (QoL). Upon removing QoL-related variables, bloating, and autoimmune disease became the greatest contributing factors. Model accuracy was 83.1%. In the decision tree model, QoL was also observed as the most decisive factor. Upon removing QoL variables, bloating was the first node. Model accuracy was 82.5%.
Conclusion:
QoL, bloating, and autoimmune disease were the most important factors associated with depression in type 2 diabetes mellitus patients.
