arXiv:2503.15390eess.IVcs.CV2025-03

FedSCA提升异构医疗影像分割的联邦微调效果

FedSCA: Federated Tuning with Similarity-guided Collaborative Aggregation for Heterogeneous Medical Image Segmentation

  • 用轻量适配器和相似性聚合,实现高效低通信的联邦微调
  • 在三个医疗图像分割基准上达到新最优性能
  • 适合隐私敏感场景下跨医院协作训练大模型

基于Transformer的医学影像分割基础模型表现优异,但受限于各医院数据规模小且无法集中,难以大规模应用。将联邦学习与基础模型微调结合(FLFM)可解决此问题,实现不共享数据的协同训练。然而,客户端数据分布不均(non-IID)、计算与通信资源有限,仍是关键挑战。本文提出新型联邦微调框架FedSCA,涵盖全流程:(1) 设计参数高效的本地微调策略以提升计算效率;(2) 仅传输部分底层适配器以降低通信开销;(3) 服务器端采用相似性引导的协同聚合(SGCA)应对非独立同分布问题。在三个医疗图像分割联邦基准上的实验表明,该方法显著提升性能,达到当前最优水平。

原文摘要 · Abstract (English)

Transformer-based foundation models (FMs) have recently demonstrated remarkable performance in medical image segmentation. However, scaling these models is challenging due to the limited size of medical image datasets within isolated hospitals, where data centralization is restricted due to privacy concerns. These constraints, combined with the data-intensive nature of FMs, hinder their broader application. Integrating federated learning (FL) with foundation models (FLFM) fine-tuning offers a potential solution to these challenges by enabling collaborative model training without data sharing, thus allowing FMs to take advantage of a diverse pool of sensitive medical image data across hospitals/clients. However, non-independent and identically distributed (non-IID) data among clients, paired with computational and communication constraints in federated environments, presents an additional challenge that limits further performance improvements and remains inadequately addressed in existing studies. In this work, we propose a novel FLFM fine-tuning framework, \underline{\textbf{Fed}}erated tuning with \underline{\textbf{S}}imilarity-guided \underline{\textbf{C}}ollaborative \underline{\textbf{A}}ggregation (FedSCA), encompassing all phases of the FL process. This includes (1) specially designed parameter-efficient fine-tuning (PEFT) for local client training to enhance computational efficiency; (2) partial low-level adapter transmission for communication efficiency; and (3) similarity-guided collaborative aggregation (SGCA) on the server side to address non-IID issues. Extensive experiments on three FL benchmarks for medical image segmentation demonstrate the effectiveness of our proposed FedSCA, establishing new SOTA performance.

联邦学习医学影像模型微调异构数据

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