| Literature DB >> 36220875 |
Hiroe Seto1,2, Asuka Oyama3, Shuji Kitora1, Hiroshi Toki1,4, Ryohei Yamamoto1,5,6, Jun'ichi Kotoku1,7, Akihiro Haga1,8, Maki Shinzawa5, Miyae Yamakawa9, Sakiko Fukui9,10, Toshiki Moriyama1,5,6.
Abstract
We sought to verify the reliability of machine learning (ML) in developing diabetes prediction models by utilizing big data. To this end, we compared the reliability of gradient boosting decision tree (GBDT) and logistic regression (LR) models using data obtained from the Kokuho-database of the Osaka prefecture, Japan. To develop the models, we focused on 16 predictors from health checkup data from April 2013 to December 2014. A total of 277,651 eligible participants were studied. The prediction models were developed using a light gradient boosting machine (LightGBM), which is an effective GBDT implementation algorithm, and LR. Their reliabilities were measured based on expected calibration error (ECE), negative log-likelihood (Logloss), and reliability diagrams. Similarly, their classification accuracies were measured in the area under the curve (AUC). We further analyzed their reliabilities while changing the sample size for training. Among the 277,651 participants, 15,900 (7978 males and 7922 females) were newly diagnosed with diabetes within 3 years. LightGBM (LR) achieved an ECE of 0.0018 ± 0.00033 (0.0048 ± 0.00058), a Logloss of 0.167 ± 0.00062 (0.172 ± 0.00090), and an AUC of 0.844 ± 0.0025 (0.826 ± 0.0035). From sample size analysis, the reliability of LightGBM became higher than LR when the sample size increased more than [Formula: see text]. Thus, we confirmed that GBDT provides a more reliable model than that of LR in the development of diabetes prediction models using big data. ML could potentially produce a highly reliable diabetes prediction model, a helpful tool for improving lifestyle and preventing diabetes.Entities:
Mesh:
Year: 2022 PMID: 36220875 PMCID: PMC9553945 DOI: 10.1038/s41598-022-20149-z
Source DB: PubMed Journal: Sci Rep ISSN: 2045-2322 Impact factor: 4.996
Characteristics of variables for all subjects and those in the positive and negative groups. Values are presented as median [Q1, Q3] for continuous variables, and the number n and percentage in brackets for categorical variables. BMI body mass index, SBP systolic blood pressure, TG triglyceride cholesterol, HDL-C high-density lipoprotein cholesterol, LDL-C low-density lipoprotein cholesterol, ALT alanine aminotransferase, HbA1c glycated hemoglobin A1c, UP urinary protein, HTN hypertension, DLP dyslipidemia, MH medical history.
| All (N = 277,651) | Negative (N = 261,751) | Positive (N = 15,900) | |
|---|---|---|---|
| Age (years) | 68.0 [63.0, 75.0] | 68.0 [63.0, 74.0] | 71.0 [66.0, 77.0] |
| BMI (kg/m | 22.3 [20.4, 24.4] | 22.3 [20.4, 24.3] | 23.5 [21.4, 25.7] |
| SBP (mmHg) | 128.0 [118.0, 139.0] | 128.0 [117.0, 138.0] | 132.0 [122.0, 142.0] |
| TG (mg/dL) | 93.0 [69.0, 128.0] | 92.0 [68.0, 127.0] | 107.0 [79.0, 149.0] |
| HDL-C (mg/dL) | 63.0 [52.0, 73.0] | 63.0 [53.0, 74.0] | 57.0 [48.0, 68.0] |
| LDL-C (mg/dL) | 124.0 [105.0, 145.0] | 125.0 [106.0, 145.0] | 121.0 [102.0, 143.0] |
| ALT (IU/L) | 17.0 [13.0, 22.0] | 17.0 [13.0, 22.0] | 19.0 [14.0, 26.0] |
| HbA1c (%) | 5.5 [5.3, 5.8] | 5.5 [5.3, 5.7] | 6.0 [5.7, 6.2] |
| Female | 169,779 (61.1) | 161,857 (61.8) | 7922 (49.8) |
| Smoking | 29,998 (10.8) | 28,010 (10.7) | 1988 (12.5) |
| UP | |||
| Negative | 262,955 (94.7) | 248,453 (94.9) | 14,502 (91.4) |
| Positive | 14,696 (5.3) | 13,298 (5.1) | 1398 (8.8) |
| Anti-HTN drug | 99,612 (35.9) | 91,230 (34.9) | 8382 (52.7) |
| Anti-DLP drug | 67,923 (24.5) | 62,747 (24.0) | 5176 (32.6) |
| MH stroke | 9605 (3.5) | 8733 (3.3) | 872 (5.5) |
| MH heart disease | 16,462 (5.9) | 14,972 (5.7) | 1490 (9.4) |
| MH renal failure | 1043 (0.4) | 954 (0.4) | 89 (0.6) |
Figure 1Distributions of all the continuous variables. The histograms of the negative group are denoted by a blue color, whereas those of the positive group are denoted by an orange color. The distributions are truncated at 100 for ALT and 500 for TG for better presentation.
Figure 2Reliability diagrams for the LR model (left) and LightGBM (right). The horizontal and vertical axes are the predicted probabilities and fraction of positives. There are five curves owing to the 5-fold cross-validation method. The thick black line is the mean of five curves, and the gray area represents the standard deviation.
Figure 3Model performances of the LR and LightGBM models with various sample sizes. The figure on the left shows the Logloss values as functions of the sample size, and the figure on the right shows the ECE values. The error bars are the standard deviation of 100 trials.
Figure 4Reliability diagrams for the LR (upper figures) and LightGBM (lower figures) methods with various sample sizes. The samples of , , and are shown in the left, middle, and right figures, respectively. There are 100 reliability curves in each figure.
Figure 5AUC values with various sample sizes for the LR and LightGBM methods.