用不确定性引导联邦域适应,提升多中心阿尔茨海默病检测精度
UG-FedDA: Uncertainty-Guided Federated Domain Adaptation for Multi-Center Alzheimer's Disease Detection
- 结合不确定性量化与联邦域适应,自适应对齐跨中心脑影像特征
- 在三个数据集上实现最高90.54%准确率,显著优于传统方法
- 适合需要隐私保护的多中心医疗模型部署
阿尔茨海默病(AD)是一种不可逆的神经退行性疾病,早期诊断对及时干预至关重要。然而,现有分类框架在多中心研究中常忽视站点间异质性且缺乏不确定性量化机制,限制了其鲁棒性与临床应用。为此,本文提出不确定性引导的联邦域适应(UG-FedDA)框架,融合不确定性量化(UQ)与联邦域适应,在隐私约束下处理跨站点结构磁共振成像(MRI)异质性。该方法采用自注意力变换器提取多模板感兴趣区域(RoI)特征,捕捉区域表征及其交互关系;通过不确定性量化引导特征对齐,降低不确定样本权重以缓解源-目标分布偏移。在阿尔茨海默病神经影像计划(ADNI)、澳大利亚影像生物标志物与生活方式研究(AIBL)和开放获取影像研究系列(OASIS)三个公开数据集上进行实验。在三类分类任务中,UG-FedDA均实现跨域性能提升:在NC vs. AD任务中,准确率分别为90.54%(ADNI)、89.04%(AIBL)、77.78%(OASIS);MCI vs. AD任务中为80.20%(ADNI)、71.91%(AIBL)、79.73%(OASIS);NC vs. MCI任务中为76.87%(ADNI)、73.91%(AIBL)、83.73%(OASIS)。结果表明,该框架不仅能高效适应多中心数据,还严格保障隐私。
原文摘要 · Abstract (English)
Alzheimer's disease (AD) is an irreversible neurodegenerative disorder, and early diagnosis is critical for timely intervention. However, most existing classification frameworks face challenges in multicenter studies, as they often neglect inter-site heterogeneity and lack mechanisms to quantify uncertainty, which limits their robustness and clinical applicability. To address these issues, we proposed Uncertainty-Guided Federated Domain Adaptation (UG-FedDA), a novel multicenter AD classification framework that integrates uncertainty quantification (UQ) with federated domain adaptation to handle cross-site structure magnetic resonance imaging (MRI) heterogeneity under privacy constraints. Our approach extracts multi-template region-of-interest (RoI) features using a self-attention transformer, capturing both regional representations and their interactions. UQ is integrated to guide feature alignment, mitigating source-target distribution shifts by down-weighting uncertain samples. Experiments are conducted on three public datasets: the Alzheimer's Disease Neuroimaging Initiative (ADNI), the Australian Imaging, Biomarkers and Lifestyle study (AIBL), and the Open Access Series of Imaging Studies (OASIS). UG-FedDA achieved consistent cross-domain improvements in accuracy, sensitivity, and area under the ROC curve across three classification tasks: AD vs. normal controls (NC), mild cognitive impairment (MCI) vs. AD, and NC vs. MCI. For NC vs. AD, UG-FedDA achieves accuracies of 90.54%, 89.04%, and 77.78% on ADNI, AIBL and OASIS datasets, respectively. For MCI vs. AD, accuracies are 80.20% (ADNI), 71.91% (AIBL), and 79.73% (OASIS). For NC vs. MCI, results are 76.87% (ADNI), 73.91% (AIBL), and 83.73% (OASIS). These results demonstrate that the proposed framework not only adapts efficiently across multiple sites but also preserves strict privacy.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。