当前位置:
X-MOL 学术
›
arXiv.cs.LG
›
论文详情
Our official English website, www.x-mol.net, welcomes your feedback! (Note: you will need to create a separate account there.)
Dynamic Fusion based Federated Learning for COVID-19 Detection
arXiv - CS - Machine Learning Pub Date : 2020-09-22 , DOI: arxiv-2009.10401 Weishan Zhang, Tao Zhou, Qinghua Lu, Xiao Wang, Chunsheng Zhu, Haoyun Sun, Zhipeng Wang, Sin Kit Lo, Fei-Yue Wang
arXiv - CS - Machine Learning Pub Date : 2020-09-22 , DOI: arxiv-2009.10401 Weishan Zhang, Tao Zhou, Qinghua Lu, Xiao Wang, Chunsheng Zhu, Haoyun Sun, Zhipeng Wang, Sin Kit Lo, Fei-Yue Wang
Medical diagnostic image analysis (e.g., CT scan or X-Ray) using machine
learning is an efficient and accurate way to detect COVID-19 infections.
However, sharing diagnostic images across medical institutions is usually not
allowed due to the concern of patients' privacy. This causes the issue of
insufficient datasets for training the image classification model. Federated
learning is an emerging privacy-preserving machine learning paradigm that
produces an unbiased global model based on the received updates of local models
trained by clients without exchanging clients' local data. Nevertheless, the
default setting of federated learning introduces huge communication cost of
transferring model updates and can hardly ensure model performance when data
heterogeneity of clients heavily exists. To improve communication efficiency
and model performance, in this paper, we propose a novel dynamic fusion-based
federated learning approach for medical diagnostic image analysis to detect
COVID-19 infections. First, we design an architecture for dynamic fusion-based
federated learning systems to analyse medical diagnostic images. Further, we
present a dynamic fusion method to dynamically decide the participating clients
according to their local model performance and schedule the model fusion-based
on participating clients' training time. In addition, we summarise a category
of medical diagnostic image datasets for COVID-19 detection, which can be used
by the machine learning community for image analysis. The evaluation results
show that the proposed approach is feasible and performs better than the
default setting of federated learning in terms of model performance,
communication efficiency and fault tolerance.
中文翻译:
用于 COVID-19 检测的基于动态融合的联邦学习
使用机器学习的医学诊断图像分析(例如 CT 扫描或 X 射线)是检测 COVID-19 感染的有效且准确的方法。但是,出于对患者隐私的考虑,通常不允许跨医疗机构共享诊断图像。这会导致用于训练图像分类模型的数据集不足的问题。联邦学习是一种新兴的隐私保护机器学习范式,它根据接收到的客户端训练的本地模型的更新来生成无偏的全局模型,而无需交换客户端的本地数据。然而,联邦学习的默认设置引入了传输模型更新的巨大通信成本,并且在客户端数据异构性严重存在的情况下难以保证模型性能。为了提高通信效率和模型性能,在本文中,我们提出了一种新的基于动态融合的联邦学习方法,用于医学诊断图像分析以检测 COVID-19 感染。首先,我们为基于动态融合的联邦学习系统设计了一种架构来分析医学诊断图像。此外,我们提出了一种动态融合方法,根据参与客户的本地模型性能动态决定参与客户,并根据参与客户的训练时间安排模型融合。此外,我们总结了一类用于 COVID-19 检测的医学诊断图像数据集,可供机器学习社区用于图像分析。
更新日期:2020-10-27
中文翻译:
用于 COVID-19 检测的基于动态融合的联邦学习
使用机器学习的医学诊断图像分析(例如 CT 扫描或 X 射线)是检测 COVID-19 感染的有效且准确的方法。但是,出于对患者隐私的考虑,通常不允许跨医疗机构共享诊断图像。这会导致用于训练图像分类模型的数据集不足的问题。联邦学习是一种新兴的隐私保护机器学习范式,它根据接收到的客户端训练的本地模型的更新来生成无偏的全局模型,而无需交换客户端的本地数据。然而,联邦学习的默认设置引入了传输模型更新的巨大通信成本,并且在客户端数据异构性严重存在的情况下难以保证模型性能。为了提高通信效率和模型性能,在本文中,我们提出了一种新的基于动态融合的联邦学习方法,用于医学诊断图像分析以检测 COVID-19 感染。首先,我们为基于动态融合的联邦学习系统设计了一种架构来分析医学诊断图像。此外,我们提出了一种动态融合方法,根据参与客户的本地模型性能动态决定参与客户,并根据参与客户的训练时间安排模型融合。此外,我们总结了一类用于 COVID-19 检测的医学诊断图像数据集,可供机器学习社区用于图像分析。