DivergentNets: Medical Image Segmentation by Network Ensemble

2021-07-01 · Vajira Thambawita, Steven A. Hicks, Pål Halvorsen, Michael A. Riegler

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.

📄 PDF Abstract BibTeX arXiv:2107.00283

Code (1)

vlbthambawita/divergent-nets 공식 구현 pytorch

Tasks

Image Segmentation Medical Image Segmentation object-detection Object Detection Segmentation Semantic Segmentation

Methods 이 논문이 사용한 방법론

Dice Loss \begin{equation} DiceLoss\left( y, \overline{p} \right) = 1 - \dfrac{\left( 2y\overline{p} + 1 \right)} {\left( y+\overline{p } + 1 \right)} \end{equation}

Spatial Pyramid Pooling Spatial Pyramid Pooling (SPP) is a pooling layer that removes the fixed-size constraint of the network, i.e. a CNN does not require a fixed-size input image. Specifically, we…

Convolution A convolution is a type of matrix operation, consisting of a kernel, a small matrix of weights, that slides over input data performing element-wise multiplication with the…

Batch Normalization 설명 없음

ASPP Atrous Spatial Pyramid Pooling (ASPP) is a semantic segmentation module for resampling a given feature layer at multiple rates prior to…

Dilated Convolution 설명 없음

1x1 Convolution A 1 x 1 Convolution is a convolution with some special properties in that it can be used for dimensionality reduction,…

UNet++ UNet++ is an architecture for semantic segmentation based on the U-Net. Through the use of densely connected nested decoder…

Diversity-Promoting Ensemble for Medical Image Segmentation

2022-10-22 · Mariana-Iuliana Georgescu, Radu Tudor Ionescu, Andreea-Iuliana Miron

Medical image segmentation is an actively studied task in medical imaging, where the precision of the annotations is of utter importance towards accurate diagnosis and treatment. In recent years, the task has been approa…

Diversity Image Segmentation Medical Image Segmentation Segmentation+1

Self-Adaptive 2D-3D Ensemble of Fully Convolutional Networks for Medical Image Segmentation

2019-07-26 · Maria G. Baldeon Calisto, Susana K. Lai-Yuen

Segmentation is a critical step in medical image analysis. Fully Convolutional Networks (FCNs) have emerged as powerful segmentation models achieving state-of-the-art results in various medical image datasets. Network ar…

Image Segmentation Medical Image Analysis Medical Image Segmentation Segmentation+1

Two layer Ensemble of Deep Learning Models for Medical Image Segmentation

2021-04-10 · Truong Dang, Tien Thanh Nguyen, John McCall, Eyad Elyan 외

In recent years, deep learning has rapidly become a method of choice for the segmentation of medical images. Deep Neural Network (DNN) architectures such as UNet have achieved state-of-the-art results on many medical dat…

Deep Learning Image Segmentation Medical Image Segmentation Segmentation+2

Uncertainty-Aware Retinal Vessel Segmentation via Ensemble Distillation

2025-09-15 · Jeremiah Fadugba, Petru Manescu, Bolanle Oladejo, Delmiro Fernandez-Reyes 외 arxiv

Uncertainty estimation is critical for reliable medical image segmentation, particularly in retinal vessel analysis, where accurate predictions are essential for diagnostic applications. Deep Ensembles, where multiple ne…

Retinal Vessel Segmentation Medical Image Segmentation

BUSU-Net: An Ensemble U-Net Framework for Medical Image Segmentation

2020-03-03 · Wei Hao Khoong

In recent years, convolutional neural networks (CNNs) have revolutionized medical image analysis. One of the most well-known CNN architectures in semantic segmentation is the U-net, which has achieved much success in sev…

AutoML Image Segmentation Medical Image Analysis Medical Image Segmentation+3