用域适应提升髋部骨折风险预测模型跨人群泛化能力
Improving Generalizability of Hip Fracture Risk Prediction via Domain Adaptation Across Multiple Cohorts
- 融合MMD、CORAL和DANN的多方法域适应提升模型鲁棒性
- 女性源数据下AUC达0.95,男性源数据下AUC达0.88
- 无需目标集标签,适合真实临床部署场景
临床风险预测模型常因不同医疗中心、地区、人口特征及测量协议导致的数据分布差异而难以跨队列泛化。在髋部骨折风险预测中,单一队列训练的模型在其他队列上性能显著下降。本文基于三大队列(SOF、MrOS、UKB)的共享临床与DXA特征,系统评估了最大均值差异(MMD)、相关性对齐(CORAL)和域对抗神经网络(DANN)及其组合在跨队列泛化中的表现。针对仅含男性的源队列和仅含女性的源队列,域适应方法均优于无适应基线;多种方法组合带来最大且最稳定的性能提升。结合MMD、CORAL与DANN的方法在男性源队列上实现AUC 0.88,女性源队列上达到AUC 0.95,表明多方法融合可生成对数据集差异不敏感的特征表示。相比依赖监督调优或假设目标样本已知结果的现有方法,本研究提出无需目标集结果的策略,更符合实际部署条件,有效提升了髋部骨折风险预测模型的泛化能力。
原文摘要 · Abstract (English)
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.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。