Combo Loss: Handling Input and Output Imbalance in Multi-Organ Segmentation

05/08/2018
by   Saeid Asgari Taghanaki, et al.
2

Simultaneous segmentation of multiple organs from different medical imaging modalities is a crucial task as it can be utilized for computer-aided diagnosis, computer-assisted surgery, and therapy planning. Thanks to the recent advances in deep learning, several deep neural networks for medical image segmentation have been introduced successfully for this purpose. In this paper, we focus on learning a deep multi-organ segmentation network that labels voxels. In particular, we examine the critical choice of a loss function in order to handle the notorious imbalance problem that plagues both the input and output of a learning model. The input imbalance refers to the class-imbalance in the input training samples (i.e. small foreground objects embedded in an abundance of background voxels, as well as organs of varying sizes). The output imbalance refers to the imbalance between the false positives and false negatives of the inference model. We introduce a loss function that integrates a weighted cross-entropy with a Dice similarity coefficient to tackle both types of imbalance during training and inference. We evaluated the proposed loss function on three datasets of whole body PET scans with 5 target organs, MRI prostate scans, and ultrasound echocardigraphy images with a single target organ. We show that a simple network architecture with the proposed integrative loss function can outperform state-of-the-art methods and results of the competing methods can be improved when our proposed loss is used.

READ FULL TEXT

page 4

page 5

research
02/08/2021

A Mixed Focal Loss Function for Handling Class Imbalanced Medical Image Segmentation

Automatic segmentation methods are an important advancement in medical i...
research
07/11/2017

Generalised Dice overlap as a deep learning loss function for highly unbalanced segmentations

Deep-learning has proved in recent years to be a powerful tool for image...
research
08/31/2018

3D Segmentation with Exponential Logarithmic Loss for Highly Unbalanced Object Sizes

With the introduction of fully convolutional neural networks, deep learn...
research
11/26/2020

IFSS-Net: Interactive Few-Shot Siamese Network for Faster Muscles Segmentation and Propagation in 3-D Freehand Ultrasound

We present an accurate, fast and efficient method for segmentation and m...
research
01/05/2022

Tackling the Class Imbalance Problem of Deep Learning Based Head and Neck Organ Segmentation

The segmentation of organs at risk (OAR) is a required precondition for ...
research
11/20/2020

Segmentation overlapping wear particles with few labelled data and imbalance sample

Ferrograph image segmentation is of significance for obtaining features ...
research
03/03/2022

Color Space-based HoVer-Net for Nuclei Instance Segmentation and Classification

Nuclei segmentation and classification is the first and most crucial ste...

Please sign up or login with your details

Forgot password? Click here to reset