arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2312.07428cs.CVcs.AI
收藏

集成联邦学习:一种用于协作式肺炎诊断的方法

Ensemble Federated Learning: an approach for collaborative pneumonia diagnosis

Beni-Suef University(贝尼苏韦夫大学);Universidade de Vigo(维戈大学);Zagazig University(扎加齐克大学);Galala University(加拉拉大学);Ajman University(阿治曼大学)

Alhassan Mabrouk, Rebeca P. Díaz Redondo, Mohamed Abd Elaziz, Mohammed Kayed

AI总结:

针对医疗数据隐私保护需求下无法集中训练的问题,提出集成联邦学习方法,通过本地多CNN模型选优集成、中心聚合迭代的方式实现协作肺炎诊断,胸部X射线任务准确率达96.63%,性能优于集中式方法。

摘要:

联邦学习是一种非常适用于以下场景的方法:(i)数据交换存在隐私顾虑,和/或(ii)需要快速响应。在智能医疗系统中,这两方面需求通常同时存在。本文针对第一种场景展开研究,该场景下隐私保护是核心要求,因此无法通过融合不同医疗机构或研究中心(计算节点)的数据集来构建单一的大规模医学图像数据集。我们提出了一种集成联邦学习(EFL)方法,其核心特点如下:第一,每个计算节点使用不同但类型相同的数据集,在本地开展训练,并针对胸部X射线图像采用集成方法,组合8种知名CNN模型(densenet169、mobilenetv2、xception、inceptionv3、vgg16、resnet50、densenet121和resnet152v2)。第二,选取表现最优的两个本地模型构建本地集成模型,并将该模型共享至中心节点。第三,对各本地集成模型进行聚合得到全局模型,再将全局模型下发至各计算节点以开启新一轮迭代。该过程持续进行,直到最优本地模型不再发生变化。我们开展了多组实验,将所提方法与集中式方法(采用或不采用集成策略)进行对比。结果表明,我们的方案在胸部X射线图像任务上表现优于这些方法,准确率达到96.63%,且与现有文献中的其他方案相比具备极具竞争力的性能。

英文摘要:

Federated learning is a very convenient approach for scenarios where (i) the exchange of data implies privacy concerns and/or (ii) a quick reaction is needed. In smart healthcare systems, both aspects are usually required. In this paper, we work on the first scenario, where preserving privacy is key and, consequently, building a unique and massive medical image data set by fusing different data sets from different medical institutions or research centers (computation nodes) is not an option. We propose an ensemble federated learning (EFL) approach that is based on the following characteristics: First, each computation node works with a different data set (but of the same type). They work locally and apply an ensemble approach combining eight well-known CNN models (densenet169, mobilenetv2, xception, inceptionv3, vgg16, resnet50, densenet121, and resnet152v2) on Chest X-ray images. Second, the best two local models are used to create a local ensemble model that is shared with a central node. Third, the ensemble models are aggregated to obtain a global model, which is shared with the computation nodes to continue with a new iteration. This procedure continues until there are no changes in the best local models. We have performed different experiments to compare our approach with centralized ones (with or without an ensemble approach)\color{black}. The results conclude that our proposal outperforms these ones in Chest X-ray images (achieving an accuracy of 96.63\%) and offers very competitive results compared to other proposals in the literature.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑