Improving Generalizability of Hip Fracture Risk Prediction via Domain Adaptation Across Multiple Cohorts

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Sun, Shuo, Zhou, Meiling, Zhao, Chen, Keyak, Joyce H., Lane, Nancy E., Deng, Jeffrey D., Su, Kuan-Jui, Shen, Hui, Deng, Hong-Wen, Zhang, Kui, Zhou, Weihua
Natura: Preprint
Pubblicazione: 2026
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866912915245760512
author Sun, Shuo
Zhou, Meiling
Zhao, Chen
Keyak, Joyce H.
Lane, Nancy E.
Deng, Jeffrey D.
Su, Kuan-Jui
Shen, Hui
Deng, Hong-Wen
Zhang, Kui
Zhou, Weihua
author_facet Sun, Shuo
Zhou, Meiling
Zhao, Chen
Keyak, Joyce H.
Lane, Nancy E.
Deng, Jeffrey D.
Su, Kuan-Jui
Shen, Hui
Deng, Hong-Wen
Zhang, Kui
Zhou, Weihua
contents Clinical risk prediction models often fail to be generalized across cohorts because underlying data distributions differ by clinical site, region, demographics, and measurement protocols. This limitation is particularly pronounced in hip fracture risk prediction, where the performance of models trained on one cohort (the source cohort) can degrade substantially when deployed in other cohorts (target cohorts). We used a shared set of clinical and DXA-derived features across three large cohorts - the Study of Osteoporotic Fractures (SOF), the Osteoporotic Fractures in Men Study (MrOS), and the UK Biobank (UKB), to systematically evaluate the performance of three domain adaptation methods - Maximum Mean Discrepancy (MMD), Correlation Alignment (CORAL), and Domain - Adversarial Neural Networks (DANN) and their combinations. For a source cohort with males only and a source cohort with females only, domain-adaptation methods consistently showed improved performance than the no-adaptation baseline (source-only training), and the use of combinations of multiple domain adaptation methods delivered the largest and most stable gains. The method that combines MMD, CORAL, and DANN achieved the highest discrimination with the area under curve (AUC) of 0.88 for a source cohort with males only and 0.95 for a source cohort with females only), demonstrating that integrating multiple domain adaptation methods could produce feature representations that are less sensitive to dataset differences. Unlike existing methods that rely heavily on supervised tuning or assume known outcomes of samples in target cohorts, our outcome-free approaches enable the model selection under realistic deployment conditions and improve generalization of models in hip fracture risk prediction.
format Preprint
id arxiv_https___arxiv_org_abs_2602_17962
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Improving Generalizability of Hip Fracture Risk Prediction via Domain Adaptation Across Multiple Cohorts
Sun, Shuo
Zhou, Meiling
Zhao, Chen
Keyak, Joyce H.
Lane, Nancy E.
Deng, Jeffrey D.
Su, Kuan-Jui
Shen, Hui
Deng, Hong-Wen
Zhang, Kui
Zhou, Weihua
Machine Learning
Clinical risk prediction models often fail to be generalized across cohorts because underlying data distributions differ by clinical site, region, demographics, and measurement protocols. This limitation is particularly pronounced in hip fracture risk prediction, where the performance of models trained on one cohort (the source cohort) can degrade substantially when deployed in other cohorts (target cohorts). We used a shared set of clinical and DXA-derived features across three large cohorts - the Study of Osteoporotic Fractures (SOF), the Osteoporotic Fractures in Men Study (MrOS), and the UK Biobank (UKB), to systematically evaluate the performance of three domain adaptation methods - Maximum Mean Discrepancy (MMD), Correlation Alignment (CORAL), and Domain - Adversarial Neural Networks (DANN) and their combinations. For a source cohort with males only and a source cohort with females only, domain-adaptation methods consistently showed improved performance than the no-adaptation baseline (source-only training), and the use of combinations of multiple domain adaptation methods delivered the largest and most stable gains. The method that combines MMD, CORAL, and DANN achieved the highest discrimination with the area under curve (AUC) of 0.88 for a source cohort with males only and 0.95 for a source cohort with females only), demonstrating that integrating multiple domain adaptation methods could produce feature representations that are less sensitive to dataset differences. Unlike existing methods that rely heavily on supervised tuning or assume known outcomes of samples in target cohorts, our outcome-free approaches enable the model selection under realistic deployment conditions and improve generalization of models in hip fracture risk prediction.
title Improving Generalizability of Hip Fracture Risk Prediction via Domain Adaptation Across Multiple Cohorts
topic Machine Learning
url https://arxiv.org/abs/2602.17962