Multi-task Federated Learning for Heterogeneous Pancreas Segmentation

• URL: http://arxiv.org/abs/2108.08537v1
• Date: Thu, 19 Aug 2021 07:24:32 GMT
• Title: Multi-task Federated Learning for Heterogeneous Pancreas Segmentation
• Authors: Chen Shen, Pochuan Wang, Holger R. Roth, Dong Yang, Daguang Xu, Masahiro Oda, Weichung Wang, Chiou-Shann Fuh, Po-Ting Chen, Kao-Lang Liu, Wei-Chih Liao, Kensaku Mori
• Abstract summary: Federated learning for medical image segmentation becomes more challenging in multi-task settings. vanilla federated averaging algorithm makes it possible to obtain more generalizable deep learning-based segmentation models. We show improvements for the automated segmentation of pancreas and pancreatic tumors in abdominal CT images with FL settings.
• Score: 7.9720650899875825
• License: http://creativecommons.org/licenses/by/4.0/
• Abstract: Federated learning (FL) for medical image segmentation becomes more challenging in multi-task settings where clients might have different categories of labels represented in their data. For example, one client might have patient data with "healthy'' pancreases only while datasets from other clients may contain cases with pancreatic tumors. The vanilla federated averaging algorithm makes it possible to obtain more generalizable deep learning-based segmentation models representing the training data from multiple institutions without centralizing datasets. However, it might be sub-optimal for the aforementioned multi-task scenarios. In this paper, we investigate heterogeneous optimization methods that show improvements for the automated segmentation of pancreas and pancreatic tumors in abdominal CT images with FL settings.

Related papers

• AnatoMix: Anatomy-aware Data Augmentation for Multi-organ Segmentation [6.471203541258319]
  We propose a novel data augmentation strategy for increasing the generalizibility of multi-organ segmentation datasets. By object-level matching and manipulation, our method is able to generate new images with correct anatomy. Our augmentation method can lead to mean dice of 76.1, compared with 74.8 of the baseline method.
  arXiv  Detail & Related papers  (2024-03-05T21:07:50Z)
• Learnable Weight Initialization for Volumetric Medical Image Segmentation [66.3030435676252]
  We propose a learnable weight-based hybrid medical image segmentation approach. Our approach is easy to integrate into any hybrid model and requires no external training data. Experiments on multi-organ and lung cancer segmentation tasks demonstrate the effectiveness of our approach.
  arXiv  Detail & Related papers  (2023-06-15T17:55:05Z)
• Explainable Semantic Medical Image Segmentation with Style [7.074258860680265]
  We propose a fully supervised generative framework that can achieve generalisable segmentation with only limited labelled data. The proposed approach creates medical image style paired with a segmentation task driven discriminator incorporating end-to-end adversarial training. Experiments on a fully semantic, publicly available pelvis dataset demonstrated that our method is more generalisable to shifts than other state-of-the-art methods.
  arXiv  Detail & Related papers  (2023-03-10T04:34:51Z)
• PCA: Semi-supervised Segmentation with Patch Confidence Adversarial Training [52.895952593202054]
  We propose a new semi-supervised adversarial method called Patch Confidence Adrial Training (PCA) for medical image segmentation. PCA learns the pixel structure and context information in each patch to get enough gradient feedback, which aids the discriminator in convergent to an optimal state. Our method outperforms the state-of-the-art semi-supervised methods, which demonstrates its effectiveness for medical image segmentation.
  arXiv  Detail & Related papers  (2022-07-24T07:45:47Z)
• Federated Multi-organ Segmentation with Inconsistent Labels [14.096407787470701]
  Federated learning is an emerging paradigm allowing large-scale decentralized learning without sharing data across different data owners. In practice, each clinical site may only annotate certain organs of interest with partial or no overlap with other sites. This work tackles the problem by using a novel federated multi-encoding U-Net (Fed-MENU) method for multi-organ segmentation.
  arXiv  Detail & Related papers  (2022-06-14T20:34:40Z)
• Data variation-aware medical image segmentation [0.0]
  We propose an approach that improves on our previous work in this area. In experiments with a real clinical dataset of CT scans with prostate segmentations, our approach provides an improvement of several percentage points in terms of Dice and surface Dice coefficients.
  arXiv  Detail & Related papers  (2022-02-24T13:35:34Z)
• Learning from Partially Overlapping Labels: Image Segmentation under Annotation Shift [68.6874404805223]
  We propose several strategies for learning from partially overlapping labels in the context of abdominal organ segmentation. We find that combining a semi-supervised approach with an adaptive cross entropy loss can successfully exploit heterogeneously annotated data.
  arXiv  Detail & Related papers  (2021-07-13T09:22:24Z)
• ATSO: Asynchronous Teacher-Student Optimization for Semi-Supervised Medical Image Segmentation [99.90263375737362]
  We propose ATSO, an asynchronous version of teacher-student optimization. ATSO partitions the unlabeled data into two subsets and alternately uses one subset to fine-tune the model and updates the label on the other subset. We evaluate ATSO on two popular medical image segmentation datasets and show its superior performance in various semi-supervised settings.
  arXiv  Detail & Related papers  (2020-06-24T04:05:12Z)
• 3D medical image segmentation with labeled and unlabeled data using autoencoders at the example of liver segmentation in CT images [58.720142291102135]
  This work investigates the potential of autoencoder-extracted features to improve segmentation with a convolutional neural network. A convolutional autoencoder was used to extract features from unlabeled data and a multi-scale, fully convolutional CNN was used to perform the target task of 3D liver segmentation in CT images.
  arXiv  Detail & Related papers  (2020-03-17T20:20:43Z)
• MS-Net: Multi-Site Network for Improving Prostate Segmentation with Heterogeneous MRI Data [75.73881040581767]
  We propose a novel multi-site network (MS-Net) for improving prostate segmentation by learning robust representations. Our MS-Net improves the performance across all datasets consistently, and outperforms state-of-the-art methods for multi-site learning.
  arXiv  Detail & Related papers  (2020-02-09T14:11:50Z)
• Observer variation-aware medical image segmentation by combining deep learning and surrogate-assisted genetic algorithms [0.0]
  We propose an approach capable of mimicking different styles of segmentation. Our approach provides an improvement of up to 23% in terms of Dice and surface Dice coefficients compared to one network trained on all data.
  arXiv  Detail & Related papers  (2020-01-23T14:51:40Z)

This list is automatically generated from the titles and abstracts of the papers in this site.

This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.