用视觉大模型融合多模态数据,实现医疗联邦学习一次通信高精度诊断。
Multi-Modal One-Shot Federated Ensemble Learning for Medical Data with Vision Large Language Model
- 利用视觉大模型生成图像文本报告,融合图文特征提升诊断能力。
- 单轮通信下在四大数据集上性能超现有方法17.5%以上。
- 适合医疗隐私保护场景,尤其适用于非独立同分布数据。
联邦学习因能保障数据隐私,在医疗领域备受关注。但传统方法需多轮通信,导致带宽受限环境下延迟高。一次联邦学习通过单轮通信完成训练与聚合,显著降低通信开销。其中,一次联邦集成学习通过投票等集成策略融合客户端独立训练模型,提升非独立同分布(non-IID)数据下的性能。然而,现有医疗方法多依赖单一模态数据(如图像或文本),限制了诊断准确性。为此,本文提出FedMME框架,一种基于视觉大语言模型的一次多模态联邦集成学习方法。该方法利用视觉大模型从医学图像生成文本报告,再使用BERT提取文本特征,与视觉特征融合以增强诊断能力。实验表明,该方法在四个不同数据分布的医疗数据集上均优于现有一次联邦学习方法,例如在RSNA数据集上,使用Dirichlet分布(α = 0.3)时,准确率提升超过17.5%。
原文摘要 · Abstract (English)
Federated learning (FL) has attracted considerable interest in the medical domain due to its capacity to facilitate collaborative model training while maintaining data privacy. However, conventional FL methods typically necessitate multiple communication rounds, leading to significant communication overhead and delays, especially in environments with limited bandwidth. One-shot federated learning addresses these issues by conducting model training and aggregation in a single communication round, thereby reducing communication costs while preserving privacy. Among these, one-shot federated ensemble learning combines independently trained client models using ensemble techniques such as voting, further boosting performance in non-IID data scenarios. On the other hand, existing machine learning methods in healthcare predominantly use unimodal data (e.g., medical images or textual reports), which restricts their diagnostic accuracy and comprehensiveness. Therefore, the integration of multi-modal data is proposed to address these shortcomings. In this paper, we introduce FedMME, an innovative one-shot multi-modal federated ensemble learning framework that utilizes multi-modal data for medical image analysis. Specifically, FedMME capitalizes on vision large language models to produce textual reports from medical images, employs a BERT model to extract textual features from these reports, and amalgamates these features with visual features to improve diagnostic accuracy. Experimental results show that our method demonstrated superior performance compared to existing one-shot federated learning methods in healthcare scenarios across four datasets with various data distributions. For instance, it surpasses existing one-shot federated learning approaches by more than 17.5% in accuracy on the RSNA dataset when applying a Dirichlet distribution with ($α$ = 0.3).
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。