Let’s consider a 2-class / binary segmentation problem where c=0 for background (healthy tissues) and c=1 for foreground (diseased tissues).

A trained U-Net model provides pixel-wise prediction either 0 or 1 for the foreground class c=1 only. The prediction for c=0 is obtained by inverting the model's output. Here is an example for a 3×3 image. The ground truth and predicted masks are given for both classes. Several performance matrices i.e., accuracy, precision, recall, dice, and iou are computed.

Example for binary segmentation

What's the standard practice in research community for publishing results in journals/conferences? Should the metrics be reported for c=1 (option 1) or mean of the both class (option 2)?

  1. IoU(c=1) = 0.750 i.e., only diseased tissues are important
  2. mIoU = (IoU(c=0) + IoU(c=1)) / 2 = (0.833 + 0.750) / 2 = 0.792 i.e., both classes are important and we can't afford mislabeling a healthy tissue as a diseased one or vice versa.

I will highly appreciate if you share some resource as reference.


Your Answer

By clicking “Post Your Answer”, you agree to our terms of service, privacy policy and cookie policy

Browse other questions tagged or ask your own question.