arXiv:2603.21656cs.LGcs.CY2026-03

提出可信赖的医疗联邦学习框架,解决数据隐私下的不确定性量化难题。

TrustFed: Enabling Trustworthy Medical AI under Data Privacy Constraints

  • 通过感知表示的客户端分配机制,提升跨机构模型校准效果。
  • 在超43万张医学影像上验证,覆盖六种成像模态,实现稳定置信区间覆盖。
  • 适合需要高可靠性预测的临床场景,如疾病筛查与诊断辅助系统。

保护患者隐私是医疗领域大规模应用机器学习的核心障碍,因伦理、法律及监管限制,原始数据难以集中。联邦学习虽可在不共享数据的前提下实现多机构协作训练,但现实部署中面临数据异质性、机构特异性偏差和类别不平衡等问题,导致预测可靠性下降,现有不确定性量化方法失效。本文提出 TrustFed,一种无需中央访问即可提供分布无关、有限样本覆盖保证的联邦不确定性量化框架。该框架引入基于内部模型表示的客户端分配机制,实现跨机构有效校准;并采用软最近阈值聚合策略,在降低分配不确定性的同时生成紧凑可靠的预测集。基于涵盖六种临床差异成像模态、超过43万张医学图像的广泛评估,实验表明 TrustFed 在各类别基数与不平衡场景下均具备稳健的覆盖性能。本研究将不确定性感知联邦学习从概念验证推进至可临床部署的模态无关范式,确立统计保证的不确定性为下一代医疗AI系统的核心要求。

原文摘要 · Abstract (English)

Protecting patient privacy remains a fundamental barrier to scaling machine learning across healthcare institutions, where centralizing sensitive data is often infeasible due to ethical, legal, and regulatory constraints. Federated learning offers a promising alternative by enabling privacy-preserving, multi-institutional training without sharing raw patient data; however, real-world deployments face severe challenges from data heterogeneity, site-specific biases, and class imbalance, which degrade predictive reliability and render existing uncertainty quantification methods ineffective. Here, we present TrustFed, a federated uncertainty quantification framework that provides distribution-free, finite-sample coverage guarantees under heterogeneous and imbalanced healthcare data, without requiring centralized access. TrustFed introduces a representation-aware client assignment mechanism that leverages internal model representations to enable effective calibration across institutions, along with a soft-nearest threshold aggregation strategy that mitigates assignment uncertainty while producing compact and reliable prediction sets. Using over 430,000 medical images across six clinically distinct imaging modalities, we conduct one of the most comprehensive evaluations of uncertainty-aware federated learning in medical imaging, demonstrating robust coverage guarantees across datasets with diverse class cardinalities and imbalance regimes. By validating TrustFed at this scale and breadth, our study advances uncertainty-aware federated learning from proof-of-concept toward clinically meaningful, modality-agnostic deployment, positioning statistically guaranteed uncertainty as a core requirement for next-generation healthcare AI systems.

联邦学习医疗AI不确定性量化隐私保护

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。