在联邦学习中实现跨扫描仪的隐私保护泛化,提升医学影像模型的通用性。
FedDAPL: Toward Client-Private Generalization in Federated Learning
- 将域对抗网络嵌入联邦学习流程,解决设备差异导致的泛化问题。
- 在15个训练站点、19个未见站点上验证,性能优于传统方法。
- 适合关注医疗数据隐私与跨中心模型泛化的研究者。
联邦学习(FL)通过在各研究中心本地训练模型并仅聚合模型更新,契合医学影像领域禁止共享原始数据的隐私要求。主要挑战来自扫描仪引起的域偏移:硬件或采集协议的非生物差异会导致模型在外部站点失效。现有校正方法需跨站点直接比较数据,违背了联邦学习的隐私约束。域泛化(DG)虽能无须共享原始数据学习站点无关表征,但传统方法仍依赖集中式多站点数据访问,同样违反联邦学习原则。本文通过将域对抗神经网络(DANN)直接融入联邦学习过程应对上述难题。发现朴素联邦DANN难以收敛后,提出一种近端正则化方法以稳定客户端间的对抗训练。在OpenBHB数据集上的实验表明,基于15个站点的训练数据,在19个未见站点上进行脑龄预测(训练人群年龄6-64岁,均值22±6岁;男性占比45%;验证人群年龄6-79岁,均值19±13岁;男性占比55%),该方法相比FedAvg和ERM实现了更优的跨站点泛化性能,同时保障数据隐私。
原文摘要 · Abstract (English)
Federated Learning (FL) trains models locally at each research center or clinic and aggregates only model updates, making it a natural fit for medical imaging, where strict privacy laws forbid raw data sharing. A major obstacle is scanner-induced domain shift: non-biological variations in hardware or acquisition protocols can cause models to fail on external sites. Most harmonization methods correct this shift by directly comparing data across sites, conflicting with FL's privacy constraints. Domain Generalization (DG) offers a privacy-friendly alternative - learning site-invariant representations without sharing raw data - but standard DG pipelines still assume centralized access to multi-site data, again violating FL's guarantees. This paper meets these difficulties with a straightforward integration of a Domain-Adversarial Neural Network (DANN) within the FL process. After demonstrating that a naive federated DANN fails to converge, we propose a proximal regularization method that stabilizes adversarial training among clients. Experiments on T1-weighted 3-D brain MRIs from the OpenBHB dataset, performing brain-age prediction on participants aged 6-64 y (mean 22+/-6 y; 45 percent male) in training and 6-79 y (mean 19+/-13 y; 55 percent male) in validation, show that training on 15 sites and testing on 19 unseen sites yields superior cross-site generalization over FedAvg and ERM while preserving data privacy.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。