DivergentNets: Medical Image Segmentation by Network Ensemble
- URL: http://arxiv.org/abs/2107.00283v1
- Date: Thu, 1 Jul 2021 08:15:00 GMT
- Title: DivergentNets: Medical Image Segmentation by Network Ensemble
- Authors: Vajira Thambawita, Steven A. Hicks, P{\aa}l Halvorsen, Michael A.
Riegler
- Abstract summary: Detection of colon polyps has become a trending topic in the fields of machine learning and gastrointestinal endoscopy.
polyp segmentation has the advantage of being more accurate than per-frame classification or object detection as it can show the affected area in greater detail.
We propose a segmentation model named TriUNet composed of three separate UNet models.
We combine TriUNet with an ensemble of well-known segmentation models, namely UNet++, FPN, DeepLabv3, and DeepLabv3+, into a model called DivergentNets to produce more generalizable medical image segmentation masks.
- Score: 0.6372261626436676
- License: http://creativecommons.org/licenses/by/4.0/
- Abstract: Detection of colon polyps has become a trending topic in the intersecting
fields of machine learning and gastrointestinal endoscopy. The focus has mainly
been on per-frame classification. More recently, polyp segmentation has gained
attention in the medical community. Segmentation has the advantage of being
more accurate than per-frame classification or object detection as it can show
the affected area in greater detail. For our contribution to the EndoCV 2021
segmentation challenge, we propose two separate approaches. First, a
segmentation model named TriUNet composed of three separate UNet models.
Second, we combine TriUNet with an ensemble of well-known segmentation models,
namely UNet++, FPN, DeepLabv3, and DeepLabv3+, into a model called
DivergentNets to produce more generalizable medical image segmentation masks.
In addition, we propose a modified Dice loss that calculates loss only for a
single class when performing multiclass segmentation, forcing the model to
focus on what is most important. Overall, the proposed methods achieved the
best average scores for each respective round in the challenge, with TriUNet
being the winning model in Round I and DivergentNets being the winning model in
Round II of the segmentation generalization challenge at EndoCV 2021. The
implementation of our approach is made publicly available on GitHub.
Related papers
- Learning with Explicit Shape Priors for Medical Image Segmentation [17.110893665132423]
We propose a novel shape prior module (SPM) to promote the segmentation performance of UNet-based models.
Explicit shape priors consist of global and local shape priors.
Our proposed model achieves state-of-the-art performance.
arXiv Detail & Related papers (2023-03-31T11:12:35Z) - Self-Supervised Correction Learning for Semi-Supervised Biomedical Image
Segmentation [84.58210297703714]
We propose a self-supervised correction learning paradigm for semi-supervised biomedical image segmentation.
We design a dual-task network, including a shared encoder and two independent decoders for segmentation and lesion region inpainting.
Experiments on three medical image segmentation datasets for different tasks demonstrate the outstanding performance of our method.
arXiv Detail & Related papers (2023-01-12T08:19:46Z) - UNet#: A UNet-like Redesigning Skip Connections for Medical Image
Segmentation [13.767615201220138]
We propose a novel network structure combining dense skip connections and full-scale skip connections, named UNet-sharp (UNet#) for its shape similar to symbol #.
The proposed UNet# can aggregate feature maps of different scales in the decoder sub-network and capture fine-grained details and coarse-grained semantics from the full scale.
arXiv Detail & Related papers (2022-05-24T03:40:48Z) - Two-Stream Graph Convolutional Network for Intra-oral Scanner Image
Segmentation [133.02190910009384]
We propose a two-stream graph convolutional network (i.e., TSGCN) to handle inter-view confusion between different raw attributes.
Our TSGCN significantly outperforms state-of-the-art methods in 3D tooth (surface) segmentation.
arXiv Detail & Related papers (2022-04-19T10:41:09Z) - Multi-organ Segmentation Network with Adversarial Performance Validator [10.775440368500416]
This paper introduces an adversarial performance validation network into a 2D-to-3D segmentation framework.
The proposed network converts the 2D-coarse result to 3D high-quality segmentation masks in a coarse-to-fine manner, allowing joint optimization to improve segmentation accuracy.
Experiments on the NIH pancreas segmentation dataset demonstrate the proposed network achieves state-of-the-art accuracy on small organ segmentation and outperforms the previous best.
arXiv Detail & Related papers (2022-04-16T18:00:29Z) - MFSNet: A Multi Focus Segmentation Network for Skin Lesion Segmentation [28.656853454251426]
This research develops an Artificial Intelligence (AI) framework for supervised skin lesion segmentation.
MFSNet, when evaluated on three publicly available datasets, outperforms state-of-the-art methods.
arXiv Detail & Related papers (2022-03-27T16:10:40Z) - Universal Segmentation of 33 Anatomies [19.194539991903593]
We present an approach for learning a single model that universally segments 33 anatomical structures.
We learn such a model from a union of multiple datasets, with each dataset containing the images that are partially labeled.
We evaluate our model on multiple open-source datasets, proving that our model has a good generalization performance.
arXiv Detail & Related papers (2022-03-04T02:29:54Z) - Deep ensembles based on Stochastic Activation Selection for Polyp
Segmentation [82.61182037130406]
This work deals with medical image segmentation and in particular with accurate polyp detection and segmentation during colonoscopy examinations.
Basic architecture in image segmentation consists of an encoder and a decoder.
We compare some variant of the DeepLab architecture obtained by varying the decoder backbone.
arXiv Detail & Related papers (2021-04-02T02:07:37Z) - DONet: Dual Objective Networks for Skin Lesion Segmentation [77.9806410198298]
We propose a simple yet effective framework, named Dual Objective Networks (DONet), to improve the skin lesion segmentation.
Our DONet adopts two symmetric decoders to produce different predictions for approaching different objectives.
To address the challenge of large variety of lesion scales and shapes in dermoscopic images, we additionally propose a recurrent context encoding module (RCEM)
arXiv Detail & Related papers (2020-08-19T06:02:46Z) - Class-wise Dynamic Graph Convolution for Semantic Segmentation [63.08061813253613]
We propose a class-wise dynamic graph convolution (CDGC) module to adaptively propagate information.
We also introduce the Class-wise Dynamic Graph Convolution Network(CDGCNet), which consists of two main parts including the CDGC module and a basic segmentation network.
We conduct extensive experiments on three popular semantic segmentation benchmarks including Cityscapes, PASCAL VOC 2012 and COCO Stuff.
arXiv Detail & Related papers (2020-07-19T15:26:50Z) - CRNet: Cross-Reference Networks for Few-Shot Segmentation [59.85183776573642]
Few-shot segmentation aims to learn a segmentation model that can be generalized to novel classes with only a few training images.
With a cross-reference mechanism, our network can better find the co-occurrent objects in the two images.
Experiments on the PASCAL VOC 2012 dataset show that our network achieves state-of-the-art performance.
arXiv Detail & Related papers (2020-03-24T04:55:43Z)
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.