Timezone: »
Intersection over union (IoU) score, also named Jaccard Index, is one of the most fundamental evaluation methods in machine learning. The original IoU computation cannot provide non-zero gradients and thus cannot be directly optimized by nowadays deep learning methods. Several recent works generalized IoU for bounding box regression, but they are not straightforward to adapt for pixelwise prediction. In particular, the original IoU fails to provide effective gradients for the non-overlapping and location-deviation cases, which results in performance plateau. In this paper, we propose PixIoU, a generalized IoU for pixelwise prediction that is sensitive to the distance for non-overlapping cases and the locations in prediction. We provide proofs that PixIoU holds many nice properties as the original IoU. To optimize the PixIoU, we also propose a loss function that is proved to be submodular, hence we can apply the Lov\'asz functions, the efficient surrogates for submodular functions for learning this loss. Experimental results show consistent performance improvements by learning PixIoU over the original IoU for several different pixelwise prediction tasks on Pascal VOC, VOT-2020 and Cityscapes.
Author Information
Jiaqian Yu (Samsung Research Institute China – Beijing)
Jingtao Xu (Samsung Research China-Beijing (SRC-B))
Yiwei Chen (Samsung Research Institute China – Beijing)
Weiming Li (Samsung Research China – Beijing (SRC-B))
Qiang Wang (Samsung Research China, Beijing)
ByungIn Yoo (Samsung Advanced Institute of Technology)
Jae-Joon Han (Samsung Advanced Institute of Technology)
Related Events (a corresponding poster, oral, or spotlight)
-
2021 Spotlight: Learning Generalized Intersection Over Union for Dense Pixelwise Prediction »
Thu. Jul 22nd 02:45 -- 02:50 AM Room
More from the Same Authors
-
2020 Poster: Meta Variance Transfer: Learning to Augment from the Others »
Seong-Jin Park · Seungju Han · Ji-won Baek · Insoo Kim · Juhwan Song · Hae Beom Lee · Jae-Joon Han · Sung Ju Hwang