arXiv:2501.11388cs.LGcs.DC2025-01被引 6

解决医院间数据不重叠患者的预测难题,实现跨域医疗知识迁移。

UniTrans: A Unified Vertical Federated Knowledge Transfer Framework for Enhancing Cross-Hospital Collaboration

  • 构建统一框架,联合学习重叠患者特征并迁移至非重叠患者。
  • 在真实医疗数据集上提升跨域与同域患者的预测准确率。
  • 适合关注隐私保护下跨机构医疗协作的研究者与从业者。

跨医院协作有望缓解区域间医疗资源不均问题,但严格隐私法规禁止直接共享敏感患者信息。垂直联邦学习(VFL)提供了一种保护隐私的机器学习范式,可最大化多机构数据价值。然而,传统VFL方法主要惠及数据重叠患者,对非重叠患者无法保证预测性能提升。虽有知识迁移技术可改善非重叠患者表现,却难以应对重叠与非重叠患者分属不同领域时带来的特征异构与标签异构问题。为此,我们提出统一的垂直联邦知识迁移框架UniTrans。该框架包含三步:首先,通过有效的垂直联邦表示学习方法在线建模多方联合特征,提取重叠患者联邦表示;其次,各医院离线训练本地知识迁移模块,将重叠患者联邦表示以领域自适应方式迁移至本地非重叠患者,生成增强表示;最后,利用这些增强表示提升各类下游医疗预测任务性能。在真实医疗数据集上的实验验证了该框架在同域与跨域知识迁移中的双重有效性。代码已开源。

原文摘要 · Abstract (English)

Cross-hospital collaboration has the potential to address disparities in medical resources across different regions. However, strict privacy regulations prohibit the direct sharing of sensitive patient information between hospitals. Vertical federated learning (VFL) offers a novel privacy-preserving machine learning paradigm that maximizes data utility across multiple hospitals. Traditional VFL methods, however, primarily benefit patients with overlapping data, leaving vulnerable non-overlapping patients without guaranteed improvements in medical prediction services. While some knowledge transfer techniques can enhance the prediction performance for non-overlapping patients, they fall short in addressing scenarios where overlapping and non-overlapping patients belong to different domains, resulting in challenges such as feature heterogeneity and label heterogeneity. To address these issues, we propose a novel unified vertical federated knowledge transfer framework (Unitrans). Our framework consists of three key steps. First, we extract the federated representation of overlapping patients by employing an effective vertical federated representation learning method to model multi-party joint features online. Next, each hospital learns a local knowledge transfer module offline, enabling the transfer of knowledge from the federated representation of overlapping patients to the enriched representation of local non-overlapping patients in a domain-adaptive manner. Finally, hospitals utilize these enriched local representations to enhance performance across various downstream medical prediction tasks. Experiments on real-world medical datasets validate the framework's dual effectiveness in both intra-domain and cross-domain knowledge transfer. The code of \method is available at \url{https://github.com/Chung-ju/Unitrans}.

联邦学习医疗AI知识迁移

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