Deep Neural Networks with Intersection over Union Loss for Binary Image Segmentation

Floris van Beers, Arvid Lindström, Emmanuel Okafor, Marco Wiering

2019

Abstract

In semantic segmentation tasks the Jaccard Index, or Intersection over Union (IoU), is often used as a measure of success. While this measure is more representative than per-pixel accuracy, state-of-the-art deep neural networks are still trained on accuracy by using Binary Cross Entropy loss. In this research, an alternative is used where deep neural networks are trained for a segmentation task of human faces by optimizing directly an approximation of IoU. When using this approximation, IoU becomes differentiable and can be used as a loss function. The comparison between IoU loss and Binary Cross Entropy loss is made by testing two deep neural network models on multiple datasets and data splits. The results show that training directly on IoU significantly increases performance for both models compared to training on conventional Binary Cross Entropy loss.

Paper Citation

in Harvard Style

van Beers F., Lindström A., Okafor E. and Wiering M. (2019). Deep Neural Networks with Intersection over Union Loss for Binary Image Segmentation.In Proceedings of the 8th International Conference on Pattern Recognition Applications and Methods - Volume 1: ICPRAM, ISBN 978-989-758-351-3, pages 438-445. DOI: 10.5220/0007347504380445

in Bibtex Style

@conference{icpram19,
author={Floris van Beers and Arvid Lindström and Emmanuel Okafor and Marco Wiering},
title={Deep Neural Networks with Intersection over Union Loss for Binary Image Segmentation},
booktitle={Proceedings of the 8th International Conference on Pattern Recognition Applications and Methods - Volume 1: ICPRAM,},
year={2019},
pages={438-445},
publisher={SciTePress},
organization={INSTICC},
doi={10.5220/0007347504380445},
isbn={978-989-758-351-3},
}

in EndNote Style

TY - CONF

JO - Proceedings of the 8th International Conference on Pattern Recognition Applications and Methods - Volume 1: ICPRAM,
TI - Deep Neural Networks with Intersection over Union Loss for Binary Image Segmentation
SN - 978-989-758-351-3
AU - van Beers F.
AU - Lindström A.
AU - Okafor E.
AU - Wiering M.
PY - 2019
SP - 438
EP - 445
DO - 10.5220/0007347504380445