| Title: | Development and validation of a two-stage machine learning model for personalised type 2 diabetes screening in the All of Us Research Program and UK Biobank |
| Journal: | BMJ Open |
| Published: | 27 Jan 2026 |
| Pubmed: | https://pubmed.ncbi.nlm.nih.gov/41592831/ |
| DOI: | https://doi.org/10.1136/bmjopen-2025-103171 |
| URL: | https://bmjopen.bmj.com/content/bmjopen/16/1/e103171.full.pdf |
Publication 18453
WARNING: the interactive features of this website use CSS3, which your browser does not support. To use the full features of this website, please update your browser.
Abstract
OBJECTIVE: To develop and externally validate a two-stage machine learning framework that integrates polygenic risk and clinical variables for early identification of individuals at risk of developing type 2 diabetes.</p>
METHODS: We conducted a prospective prediction study using data from the All of Us Research Program for model development and the UK Biobank for external validation. Two models were constructed. Stage 1 used gradient boosted decision trees (XGBoost) with cross validation, automated hyperparameter optimisation and class weighting to predict 5-year incident type 2 diabetes using demographic, clinical and polygenic predictors. Stage 2 incorporated glycated haemoglobin or fasting glucose measurements to refine risk estimates. Model interpretation used SHapley Additive exPlanations values and permutation importance, and logistic regression and random forest models served as comparators. Discrimination of all models was compared using the DeLong test.</p>
RESULTS: The Stage 1 model achieved an area under the receiver operating characteristic curve (AUROC) of 0.81 in All of Us and 0.82 in UK Biobank, performing significantly better than the phenotype-only model in UK Biobank (DeLong p=1.05×10⁻⁷⁶). Higher polygenic risk quartiles were associated with increased incidence of type 2 diabetes in both cohorts (global χ2 p<0.001). The Stage 2 model achieved AUROC values of 0.78 in All of Us and 0.77 in UK Biobank. Subgroup performance was consistent across sex and ancestry groups, with CIs reported. Cost analysis suggested potential net savings compared with the American Diabetes Association test.</p>
CONCLUSION: A two-stage machine learning framework that integrates genetic and clinical information can support personalised screening for type 2 diabetes across diverse populations. The approach demonstrated robust performance across cohorts and offers a practical structure for early risk identification.</p>
19 Keywords
- Aged
- Biological Specimen Banks
- Blood Glucose
- Diabetes Mellitus, Type 2
- Female
- Glycated Hemoglobin
- Humans
- Logistic Models
- Machine Learning
- Male
- Mass Screening
- Middle Aged
- Prospective Studies
- ROC Curve
- Risk Assessment
- Risk Factors
- UK Biobank
- United Kingdom
- United States
5 Authors
- Ahmed Khattab
- Shang-Fu Chen
- Hossein Javedani Sadaei
- Nathan E. Wineinger
- Ali Torkamani
1 Application
| Application ID | Title |
|---|---|
| 41999 | Identification of Genomic Risk Factors and Prediction of Cardiovascular Disease Risk through Deep Learning |
Enabling scientific discoveries that improve human health