arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2410.20102cs.CV

解剖学3D风格迁移实现极低通信成本的高效联邦学习

Anatomical 3D Style Transfer Enabling Efficient Federated Learning with Extremely Low Communication Costs

Preferred Networks

Yuto Shibata, Yasunori Kudo, Yohei Sugawara

AI总结:

针对多器官分割联邦学习中的数据异质性与高通信成本问题,提出A3DFDG,利用器官位置聚类并混合3D风格,在仅1.25%通信成本下提升全局Dice得分4.3%。

摘要:

在本研究中,我们提出了一种新颖的联邦学习(FL)方法,该方法利用3D风格迁移来完成多器官分割任务。通过整合多个数据集获得的多器官数据集具有很高的可扩展性,并且能够随着数据量的增加提升泛化性能。然而,由于不同客户端具有多样的成像条件和目标器官,数据的异质性可能导致局部模型严重过拟合。为了对齐在不同局部数据集上过拟合的模型,现有方法需要与中央服务器频繁通信,从而导致更高的通信成本和隐私泄露风险。为了实现高效且安全的FL,我们提出了一种用于FL的解剖学3D频域泛化(A3DFDG)方法。A3DFDG利用人体器官的结构信息,并根据器官位置对3D风格进行聚类。通过基于这些聚类混合风格,该方法保留了解剖学信息,并引导模型学习器官内多样性,同时对齐每个局部模型的优化。实验表明,即使在通信成本受到高度限制(仅为原始成本的1.25%)的情况下,我们的方法仍能保持其准确率,并且与基线相比取得了显著差异,全局Dice相似系数得分提高4.3%。尽管该方法简单且计算开销极小,但这些结果表明,在需要低通信成本和简单流程的真实场景中,我们的方法具有很高的实用性。本项目中使用的代码将公开发布。

英文摘要:

In this study, we propose a novel federated learning (FL) approach that utilizes 3D style transfer for the multi-organ segmentation task. The multi-organ dataset, obtained by integrating multiple datasets, has high scalability and can improve generalization performance as the data volume increases. However, the heterogeneity of data owing to different clients with diverse imaging conditions and target organs can lead to severe overfitting of local models. To align models that overfit to different local datasets, existing methods require frequent communication with the central server, resulting in higher communication costs and risk of privacy leakage. To achieve an efficient and safe FL, we propose an Anatomical 3D Frequency Domain Generalization (A3DFDG) method for FL. A3DFDG utilizes structural information of human organs and clusters the 3D styles based on the location of organs. By mixing styles based on these clusters, it preserves the anatomical information and leads models to learn intra-organ diversity, while aligning the optimization of each local model. Experiments indicate that our method can maintain its accuracy even in cases where the communication cost is highly limited (=1.25% of the original cost) while achieving a significant difference compared to baselines, with a higher global dice similarity coefficient score of 4.3%. Despite its simplicity and minimal computational overhead, these results demonstrate that our method has high practicality in real-world scenarios where low communication costs and a simple pipeline are required. The code used in this project will be publicly available.

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

↑