DeepAI AI Chat
Log In Sign Up

FU-net: Multi-class Image Segmentation Using Feedback Weighted U-net

by   Mina Jafari, et al.

In this paper, we present a generic deep convolutional neural network (DCNN) for multi-class image segmentation. It is based on a well-established supervised end-to-end DCNN model, known as U-net. U-net is firstly modified by adding widely used batch normalization and residual block (named as BRU-net) to improve the efficiency of model training. Based on BRU-net, we further introduce a dynamically weighted cross-entropy loss function. The weighting scheme is calculated based on the pixel-wise prediction accuracy during the training process. Assigning higher weights to pixels with lower segmentation accuracies enables the network to learn more from poorly predicted image regions. Our method is named as feedback weighted U-net (FU-net). We have evaluated our method based on T1- weighted brain MRI for the segmentation of midbrain and substantia nigra, where the number of pixels in each class is extremely unbalanced to each other. Based on the dice coefficient measurement, our proposed FU-net has outperformed BRU-net and U-net with statistical significance, especially when only a small number of training examples are available. The code is publicly available in GitHub (GitHub link:


PARSE challenge 2022: Pulmonary Arteries Segmentation using Swin U-Net Transformer(Swin UNETR) and U-Net

In this work, we present our proposed method to segment the pulmonary ar...

Feedback U-net for Cell Image Segmentation

Human brain is a layered structure, and performs not only a feedforward ...

A Unified Mammogram Analysis Method via Hybrid Deep Supervision

Automatic mammogram classification and mass segmentation play a critical...

A Few-Shot Attention Recurrent Residual U-Net for Crack Segmentation

Recent studies indicate that deep learning plays a crucial role in the a...

Attention U-Net: Learning Where to Look for the Pancreas

We propose a novel attention gate (AG) model for medical imaging that au...

Enhancing Organ at Risk Segmentation with Improved Deep Neural Networks

Organ at risk (OAR) segmentation is a crucial step for treatment plannin...