A Diabetes Risk Predicting Method With Multi-Strategy Counterfactual-Based Data Augmentation

Abstract

Diabetes is a chronic disease that poses a serious threat to health, and its early risk prediction has been a hot research topic in the field of medical artificial intelligence. Routine medical checkups are the most common way to monitor people’s health status, and the data from medical checkups contain rich diagnostic information, which is valuable for diabetes risk prediction. Currently, most of the available studies on diabetes risk prediction are based on publicly available datasets, and the models and algorithms do not work well on real clinical datasets. Real routine checkup data are characterized by complex information, diverse features, high redundancy and poor balance, which pose great challenges for diabetes risk prediction. To address this problem, this paper proposes a multi-strategy data augmentation-based diabetes risk prediction method, after completing data pre-processing and feature selection, a counterfactual-based data balancing strategy is used to augment a minority class of instances, and a density clustering-based supplemental counterfactual data augmentation strategy is proposed to address the problem of insufficient representation of generated instances in the counterfactual method. Moreover, the uncertainty-weighted method is used in the model training phase. Based on the real checkup dataset, five machine learning methods including Logistic Regression (LR), SVM, Decision Tree, Random Forest and Gradient Boosting are used to model and use 5-fold cross-validation to carry out diabetes risk assessment and prediction. The experimental results showed that the sensitivity and precision of the models were significantly improved compared with the existing methods, and the sensitivity of the LR model for diabetes risk prediction on the real routine checkup dataset reached more than 90[Formula: see text], which meet the requirements of clinical application.