Image Segmentation Mehtod Based on Federal Style Transfer

被引:0
|
作者
Ma J. [1 ,2 ,3 ,4 ]
Yin X. [1 ,2 ,3 ]
Hu C. [1 ,2 ,5 ]
Ma B. [1 ,2 ,5 ,6 ]
Ban X. [1 ,2 ,5 ,6 ,7 ]
机构
[1] Beijing Advanced Innovation Center for Materials Genome Engineering, University of Science and Technology Beijing, Beijing
[2] School of Intelligence Science and Technology, University of Science and Technology Beijing, Beijing
[3] Collaborative Innovation Center of Steel Generic Technology, Beijing University of Science and Technology, Beijing
[4] HBIS Group Company Limited, Shijiazhuang
[5] Shunde Innovation School, University of Science and Technology Beijing, Foshan
[6] Key Laboratory of Intelligent Bionic Unmanned Systems, Ministry of Education, University of Science and Technology Beijing, Beijing
[7] Institute for Materials Intelligent Technology, Liaoning Academy of Materials, Shenyang
关键词
deep learning; federated learning; image segmentation; style transfer;
D O I
10.13190/j.jbupt.2022-257
中图分类号
学科分类号
摘要
In this work, we propose an image segmentation method based on federated style transfer to slove the non-independent and identically distributed problem in federated learning. By sharing style information that is not sensitive to user privacy, this method generates synthetic data for data expansion, and reduces data differences between different users while ensuring that important structural information of data is not disclosed. Experiments results show that this method effectively alleviates the influence of nonindependent and identically distributed problem among nodes on the performance of the federated model in the liver image segmentation task. Therefore, the proposed method can further improve the performance of federal model, which provides the possibility to break the data island and establish a general model in medical field. © 2023 Beijing University of Posts and Telecommunications. All rights reserved.
引用
收藏
页码:27 / 32
页数:5
相关论文
共 21 条
  • [1] HESAMIAN M H, JIA W, HE X, Et al., Deep learning techniques for medical image segmentation: achievements and challenges, Journal of Digital Imaging, 32, 4, pp. 582-596, (2019)
  • [2] TAHA A A, HANBURY A., Metrics for evaluating 3D medical image segmentation: analysis, selection, and tool, BMC Medical Imaging, 15, 1, pp. 1-28, (2015)
  • [3] ZHOU T, RUAN S, CANU S., A review: deep learning for medical image segmentation using multi-modality fusion, Array, 3, (2019)
  • [4] KONG L J, WANG Q W, BAO Y C, Et al., A survey on medical image segmentation based on deep learning, Radio Communications Technology, 47, 2, pp. 121-130, (2021)
  • [5] WU Y J, FANG L, XU Y., Predicting interfacial thermal resistance by machine learning, Npj Computational Materials, 5, 1, pp. 1-8, (2019)
  • [6] KAIROUZ P, MCMAHAN H B, AVENT B, Et al., Advances and open problems in federated learning [ J ], Foundations and Trends in Machine Learning, 14, 1, pp. 1-210, (2021)
  • [7] Huang Y, Chu L, Zhou Z, Et al., Personalized cross-silo federated learning on Non-IID data [ C], 椅 Proceedings of the AAAI Conference on Artificial Intelligence, pp. 7865-7873, (2021)
  • [8] ZHAO Y, LI M, LAI L, Et al., Federated learning with non-IID data [ EB / OL ], (2022)
  • [9] KANG X D., Medical image processing [ M], pp. 25-27, (2009)
  • [10] HESAMIAN M H, JIA W, HE X, Et al., Deep learning techniques for medical image segmentation: achievements and challenges, Journal of Digital Imaging, 32, 4, pp. 582-596, (2019)