arXiv:2412.19654cs.LGcs.DC2024-12KDD被引 3

用知识蒸馏让医疗资源少地区也能提升诊断能力

Asymmetrical Reciprocity-based Federated Learning for Resolving Disparities in Medical Diagnosis

  • 通过调用大模型API,小客户端用少量数据学习
  • 跨区域协作使欠发达地区模型性能显著提升
  • 适合医疗资源不均场景下的公平化诊断研究

地理健康差异是全球性挑战,尤其在低收入和中等收入国家的偏远地区。解决这一问题需借助医疗较发达地区的支持,开展协作提升诊疗质量。联邦学习为此提供可能,但偏远地区数据稀缺、计算资源有限,难以训练高性能模型。同时,发达与欠发达地区间存在不对称合作关系。为此,我们提出新型跨孤岛联邦学习框架FedHelp,旨在缓解地理健康差异,增强欠发达地区的诊断能力。具体而言,FedHelp通过一次性API调用获取基础模型知识,指导欠发达小客户端的学习过程,解决数据不足问题;并引入新颖的非对称双知识蒸馏模块,应对不对称互惠关系,实现发达大客户端与欠发达小客户端间必要知识的有效交换。我们在医学图像分类与分割任务上进行了广泛实验,结果表明,相比现有最优基线,FedHelp显著提升了性能,尤其惠及欠发达地区客户端。

原文摘要 · Abstract (English)

Geographic health disparities pose a pressing global challenge, particularly in underserved regions of low- and middle-income nations. Addressing this issue requires a collaborative approach to enhance healthcare quality, leveraging support from medically more developed areas. Federated learning emerges as a promising tool for this purpose. However, the scarcity of medical data and limited computation resources in underserved regions make collaborative training of powerful machine learning models challenging. Furthermore, there exists an asymmetrical reciprocity between underserved and developed regions. To overcome these challenges, we propose a novel cross-silo federated learning framework, named FedHelp, aimed at alleviating geographic health disparities and fortifying the diagnostic capabilities of underserved regions. Specifically, FedHelp leverages foundational model knowledge via one-time API access to guide the learning process of underserved small clients, addressing the challenge of insufficient data. Additionally, we introduce a novel asymmetric dual knowledge distillation module to manage the issue of asymmetric reciprocity, facilitating the exchange of necessary knowledge between developed large clients and underserved small clients. We validate the effectiveness and utility of FedHelp through extensive experiments on both medical image classification and segmentation tasks. The experimental results demonstrate significant performance improvement compared to state-of-the-art baselines, particularly benefiting clients in underserved regions.

联邦学习医疗诊断知识蒸馏公平性

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