提出轻量联邦CLIP模型,实现医疗图像高效跨院协作分类。
Federated CLIP for Resource-Efficient Heterogeneous Medical Image Classification
- 用掩码特征适配模块降低通信开销,冻结主干减少计算。
- 在ISIC2019上比次优基线高8%,训练速度比FedAVG快120倍。
- 适合资源受限的医院联合建模,兼顾隐私与性能。
尽管深度模型在医学影像中表现优异,但仍需源数据训练,引发隐私担忧。联邦学习(FL)通过多医院协作训练共享模型,提供可行方案。然而,数据异构性与资源开销阻碍了视觉语言模型(VLM)的应用。为此,我们提出基于对比语言-图像预训练(CLIP)的联邦医疗图像分类方法(FedMedCLIP)。设计掩码特征适配模块(FAM)作为通信组件,在冻结CLIP编码器的前提下降低通信负载;引入掩码多层感知机(MLP)作为私有本地分类器以适应客户端任务;提出基于自适应KL散度的蒸馏正则化方法,促进FAM与MLP间的互学习;最后结合模型压缩传输FAM参数,并采用集成预测进行分类。在四个公开医疗数据集上的实验表明,本方法在合理资源成本下实现优越性能(如在ISIC2019上比次优基线高8%),且训练速度比FedAVG快120倍。
原文摘要 · Abstract (English)
Despite the remarkable performance of deep models in medical imaging, they still require source data for training, which limits their potential in light of privacy concerns. Federated learning (FL), as a decentralized learning framework that trains a shared model with multiple hospitals (a.k.a., FL clients), provides a feasible solution. However, data heterogeneity and resource costs hinder the deployment of FL models, especially when using vision language models (VLM). To address these challenges, we propose a novel contrastive language-image pre-training (CLIP) based FL approach for medical image classification (FedMedCLIP). Specifically, we introduce a masked feature adaptation module (FAM) as a communication module to reduce the communication load while freezing the CLIP encoders to reduce the computational overhead. Furthermore, we propose a masked multi-layer perceptron (MLP) as a private local classifier to adapt to the client tasks. Moreover, we design an adaptive Kullback-Leibler (KL) divergence-based distillation regularization method to enable mutual learning between FAM and MLP. Finally, we incorporate model compression to transmit the FAM parameters while using ensemble predictions for classification. Extensive experiments on four publicly available medical datasets demonstrate that our model provides feasible performance (e.g., 8\% higher compared to second best baseline on ISIC2019) with reasonable resource cost (e.g., 120$\times$ faster than FedAVG).
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。