Unbalanced Optimal Transport: A Unified Framework for Object Detection

CVPR 2023


Henri De Plaen1*, Pierre-François De Plaen2*, Johan A. K. Suykens1, Marc Proesmans2, Tinne Tuytelaars2, Luc Van Gool2,3
* These authors contributed equally to this work.

Abstract

TL;DR: We introduce a much more versatile new class of matching strategies unifying many existing ones, as well as being well suited for GPUs.

During training, supervised object detection tries to correctly match the predicted bounding boxes and associated classification scores to the ground truth. This is essential to determine which predictions are to be pushed towards which solutions, or to be discarded. Popular matching strategies include matching to the closest ground truth box (mostly used in combination with anchors), or matching via the Hungarian algorithm (mostly used in anchor-free methods). Each of these strategies comes with its own properties, underlying losses, and heuristics. We show how Unbalanced Optimal Transport unifies these different approaches and opens a whole continuum of methods in between. This allows for a finer selection of the desired properties. Experimentally, we show that training an object detection model with Unbalanced Optimal Transport is able to reach the state-of-the-art both in terms of Average Precision and Average Recall as well as to provide a faster initial convergence. The approach is well suited for GPU implementation, which proves to be an advantage for large-scale models.


Interactive Tool

Image 24919 from the MS COCO validation dataset.
1
2
3
4
5
A
B
COCO validation set - image 24919
Cost matrix
1 1 1 1
2 1 1 1
3 1 1 1
4 1 1 1
5 1 1 1
A B \(\varnothing\)
Set parameters
Match \(\hat{\boldsymbol{P}}\)
1 1 1 1
2 1 1 1
3 1 1 1
4 1 1 1
5 1 1 1
A B \(\varnothing\)

Presentation

You can also directly visualize the interactive presentation (small screen compatibility not guaranteed) or the slides in PDF.

Reference

Henri De Plaen, Pierre-François De Plaen, Johan A. K. Suykens, Marc Proesmans, Tinne Tuytelaars, and Luc Van Gool. 2023. “Unbalanced Optimal Transport: A Unified Framework for Object Detection.” In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR)
You can also download the BibTex or copy it from the following:

@InProceedings{De_Plaen_2023_CVPR,
    author    = {De Plaen, Henri and De Plaen, Pierre-Fran\c{c}ois and Suykens, Johan A. K. and Proesmans, Marc and Tuytelaars, Tinne and Van Gool, Luc},
    title     = {Unbalanced Optimal Transport: A Unified Framework for Object Detection},
    booktitle = {Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR)},
    month     = {June},
    year      = {2023},
    pages     = {3198-3207}
}
        

Acknowlegdements

EU: The research leading to these results has received funding from the European Research Council under the European Union’s Horizon 2020 research and innovation program / ERC Advanced Grant E-DUALITY (787960). This paper reflects only the authors’ views and the Union is not liable for any use that may be made of the contained information. Research Council KUL: Optimization frameworks for deep kernel machines C14/18/068. Flemish Government: FWO: projects: GOA4917N (Deep Restricted Kernel Machines: Methods and Foundations), PhD/Postdoc grant; This research received funding from the Flemish Government (AI Research Program). All the authors are also affiliated to Leuven.AI - KU Leuven institute for AI, B-3000, Leuven, Belgium.

European Union
European Research Council
KU Leuven
Fonds voor Wetenschappelijk Onderzoek
Flanders AI