J Regularization Improves Imbalanced Multiclass Segmentation

10/22/2019
by   Fidel A. Guerrero Peña, et al.
0

We propose a new loss formulation to further advance the multiclass segmentation of cluttered cells under weakly supervised conditions. We improve the separation of touching and immediate cells, obtaining sharp segmentation boundaries with high adequacy, when we add Youden's J statistic regularization term to the cross entropy loss. This regularization intrinsically supports class imbalance thus eliminating the necessity of explicitly using weights to balance training. Simulations demonstrate this capability and show how the regularization leads to better results by helping advancing the optimization when cross entropy stalls. We build upon our previous work on multiclass segmentation by adding yet another training class representing gaps between adjacent cells. This addition helps the classifier identify narrow gaps as background and no longer as touching regions. We present results of our methods for 2D and 3D images, from bright field to confocal stacks containing different types of cells, and we show that they accurately segment individual cells after training with a limited number of annotated images, some of which are poorly annotated.

READ FULL TEXT

page 1

page 3

page 4

research
02/21/2018

Multiclass Weighted Loss for Instance Segmentation of Cluttered Cells

We propose a new multiclass weighted loss function for instance segmenta...
research
06/28/2021

Striking the Right Balance: Recall Loss for Semantic Segmentation

Class imbalance is a fundamental problem in computer vision applications...
research
04/04/2018

Normalized Cut Loss for Weakly-supervised CNN Segmentation

Most recent semantic segmentation methods train deep convolutional neura...
research
04/21/2021

Sparse-Shot Learning for Extremely Many Localisations

Object localisation is typically considered in the context of regular im...
research
08/26/2019

A Weakly Supervised Method for Instance Segmentation of Biological Cells

We present a weakly supervised deep learning method to perform instance ...
research
08/10/2022

Imbalance Trouble: Revisiting Neural-Collapse Geometry

Neural Collapse refers to the remarkable structural properties character...

Please sign up or login with your details

Forgot password? Click here to reset