MONAI icon indicating copy to clipboard operation
MONAI copied to clipboard

Add Hybrid Segmentation Loss

Open yashika-git opened this issue 2 years ago • 3 comments

I was trying to implement UNet++ (issue #896 ). In the UNet++ paper, authors suggest using hybrid segmentation loss:


We then define a hybrid segmentation loss consisting of pixelwise cross-entropy loss and soft dice-coefficient loss for each
semantic scale. The hybrid loss may take advantages of what both loss functions have to offer: smooth gradient and handling
of class imbalance.

Equation 2 in this paper, describes the Mathematical formula to calculate the loss.

Can this loss be implemented in MONAI?

Additional context image

yashika-git avatar Aug 29 '22 12:08 yashika-git

@Nic-Ma, Shall this loss be added, will it be helpful? Thank you

yashika-git avatar Aug 30 '22 16:08 yashika-git

Hi @yashika-git ,

I think it would be a nice enhancement for MONAI, welcome your contribution. @wyli Do you have any concern?

Thanks.

Nic-Ma avatar Aug 30 '22 17:08 Nic-Ma

Thank you, I'll try adding this.

yashika-git avatar Aug 30 '22 17:08 yashika-git

is this hybrid loss the same as Dice + CrossEntropy loss? https://github.com/Project-MONAI/MONAI/blob/dev/monai/losses/dice.py#L609

myron avatar Oct 24 '22 01:10 myron

@myron, Yes, it seems to be the same. I'll close the issue.

yashika-git avatar Oct 24 '22 06:10 yashika-git