MIC-DKFZ / BraTS2017

Apache License 2.0
74 stars 16 forks source link

Why could a multiclass dice loss function solve the class imbalance problem? #29

Closed yianzhongguo closed 5 years ago

yianzhongguo commented 5 years ago

@FabianIsensee Hello, sir. I notice that in the paper of this method the class imbalacnce issue was disposed by formulating a multiclass dice loss function. I cannot figure it out why the loss function could play the role. Could you detailed it, thank you!

FabianIsensee commented 5 years ago

In crossentropy, each pixel has the same weight irrespective of the class. by using a Dice loss, the weight of a pixel is different. If the CE tumor is small for example, then false positives or false negatives will impact the dice loss more and will thus intrinsically be weighted more.

yianzhongguo commented 5 years ago

I have understood it from your answer, thank you very much!,