高级检索
王生生, 路淑贞, 曹斌. 面向隐私保护联邦学习的医学影像目标检测算法[J]. 计算机辅助设计与图形学学报, 2021, 33(10): 1553-1562. DOI: 10.3724/SP.J.1089.2021.18416
引用本文: 王生生, 路淑贞, 曹斌. 面向隐私保护联邦学习的医学影像目标检测算法[J]. 计算机辅助设计与图形学学报, 2021, 33(10): 1553-1562. DOI: 10.3724/SP.J.1089.2021.18416
Wang Shengsheng, Lu Shuzhen, Cao Bin. Medical Image Object Detection Algorithm for Privacy-Preserving Federated Learning[J]. Journal of Computer-Aided Design & Computer Graphics, 2021, 33(10): 1553-1562. DOI: 10.3724/SP.J.1089.2021.18416
Citation: Wang Shengsheng, Lu Shuzhen, Cao Bin. Medical Image Object Detection Algorithm for Privacy-Preserving Federated Learning[J]. Journal of Computer-Aided Design & Computer Graphics, 2021, 33(10): 1553-1562. DOI: 10.3724/SP.J.1089.2021.18416

面向隐私保护联邦学习的医学影像目标检测算法

Medical Image Object Detection Algorithm for Privacy-Preserving Federated Learning

  • 摘要: 联邦学习技术是一种新型多机构协同训练模型范式,广泛应用于多领域,其中模型参数隐私保护是一个关键问题.针对CT影像综合性病灶检测任务,提出隐私保护的联邦学习算法.首先部署松散耦合的客户端-服务器架构;其次在各客户端使用改进的RetinaNet检测器,引入上下文卷积和后向注意力机制;最后完成联邦训练.各客户端使用局部更新策略,采用自适应训练周期,局部目标函数中加入了限制项;服务器使用自适应梯度裁剪策略和高斯噪声差分隐私算法更新全局模型参数.在DeepLesion数据集上的消融分析说明了算法各部分的重要性.实验结果表明,改进的RetinaNet检测器有效地提升了多尺度病灶的检测精度.与集中数据训练模型范式相比,联邦学习所得模型性能略低(mAP分别为75.33%和72.80%),但训练用时缩短近38%,有效地实现了隐私保护、通信效率和模型性能的良好权衡.

     

    Abstract: Federated learning technology is a new type of multi-agency collaborative training model paradigm,which is widely used in many fields,among which model parameter privacy protection is a key issue.Aiming at the task of comprehensive lesion detection in CT images,privacy protected federated learning algorithm is proposed.A loosely coupled client-server architecture is deployed firstly.Then an improved RetinaNet detector is used on each client-side,and contextual convolution and backward attention mechanisms are introduced.Finally,federated training process is finished.Each client uses a local update strategy,adopts an adaptive training epoch,and adds the restriction term to the local objective function.The server uses an adaptive gradi-ent clipping strategy and a Gaussian noise differential privacy algorithm to update the global model parameters.Ablation analysis on the DeepLesion dataset demonstrates the importance of each part of the algorithm.Com-parative experimental results show that the improved RetinaNet detector effectively improves the detection performance of multi-scale lesions.Compared to the centralized data training model paradigm,the perform-ance of the model obtained by federated learning is slightly lower(with mAP of 75.33%and 72.80%respec-tively),but the training time is shortened by nearly 38%.The proposed algorithm effectively achieves a good trade-off between privacy protection,communication efficiency,and model performance.

     

/

返回文章
返回