IEEE J Biomed Health Inform. 2023 Dec;27(12):5946-5957. doi: 10.1109/JBHI.2023.3317632. Epub 2023 Dec 5.
U-shaped networks have become prevalent in various medical image tasks such as segmentation, and restoration. However, most existing U-shaped networks rely on centralized learning which raises privacy concerns. To address these issues, federated learning (FL) and split learning (SL) have been proposed. However, achieving a balance between the local computational cost, model privacy, and parallel training remains a challenge. In this articler, we propose a novel hybrid learning paradigm called Dynamic Corrected Split Federated Learning (DC-SFL) for U-shaped medical image networks. To preserve data privacy, including the input, model parameters, label and output simultaneously, we propose to split the network into three parts hosted by different parties. We propose a Dynamic Weight Correction Strategy (DWCS) to stabilize the training process and avoid the model drift problem due to data heterogeneity. To further enhance privacy protection and establish a trustworthy distributed learning paradigm, we propose to introduce additively homomorphic encryption into the aggregation process of client-side model, which helps prevent potential collusion between parties and provides a better privacy guarantee for our proposed method. The proposed DC-SFL is evaluated on various medical image tasks, and the experimental results demonstrate its effectiveness. In comparison with state-of-the-art distributed learning methods, our method achieves competitive performance.
U 形网络在分割、恢复等各种医学图像任务中已经变得流行。然而,大多数现有的 U 形网络依赖于集中式学习,这引发了隐私问题。为了解决这些问题,已经提出了联邦学习 (FL) 和分裂学习 (SL)。然而,在本地计算成本、模型隐私和并行训练之间取得平衡仍然是一个挑战。在这篇文章中,我们提出了一种名为动态校正分裂联邦学习 (DC-SFL) 的新型混合学习范例,用于 U 形医学图像网络。为了保护数据隐私,包括输入、模型参数、标签和输出,我们将网络分为三部分,由不同的方托管。我们提出了一种动态权重校正策略 (DWCS) 来稳定训练过程,并避免由于数据异质性导致的模型漂移问题。为了进一步增强隐私保护并建立值得信赖的分布式学习范例,我们建议将加性同态加密引入客户端模型的聚合过程中,这有助于防止各方之间的潜在勾结,并为我们提出的方法提供更好的隐私保护。所提出的 DC-SFL 在各种医学图像任务上进行了评估,实验结果证明了其有效性。与最先进的分布式学习方法相比,我们的方法具有竞争力的性能。