Deep Metric Learning with BIER: Boosting Independent Embeddings Robustly

Michael Opitz, Georg Waltner, Horst Possegger, Horst Bischof

Publikation: Beitrag in einer FachzeitschriftArtikelBegutachtung

Abstract

Learning similarity functions between image pairs with deep neural networks yields highly correlated activations of embeddings. In this work, we show how to improve the robustness of such embeddings by exploiting the independence within ensembles. To this end, we divide the last embedding layer of a deep network into an embedding ensemble and formulate the task of training this ensemble as an online gradient boosting problem. Each learner receives a reweighted training sample from the previous learners. Further, we propose two loss functions which increase the diversity in our ensemble. These loss functions can be applied either for weight initialization or during training. Together, our contributions leverage large embedding sizes more effectively by significantly reducing correlation of the embedding and consequently increase retrieval accuracy of the embedding. Our method works with any differentiable loss function and does not introduce any additional parameters during test time. We evaluate our metric learning method on image retrieval tasks and show that it improves over state-of-the-art methods on the CUB-200-2011, Cars-196, Stanford Online Products, In-Shop Clothes Retrieval and VehicleID datasets. Therefore, our findings suggest that by dividing deep networks at the end into several smaller and diverse networks, we can significantly reduce overfitting.
Originalspracheenglisch
Aufsatznummer8395046
Seiten (von - bis)276 - 290
FachzeitschriftIEEE Transactions on Pattern Analysis and Machine Intelligence
Jahrgang42
Ausgabenummer2
Frühes Online-Datum2018
DOIs
PublikationsstatusVeröffentlicht - 2020

Fingerprint

Untersuchen Sie die Forschungsthemen von „Deep Metric Learning with BIER: Boosting Independent Embeddings Robustly“. Zusammen bilden sie einen einzigartigen Fingerprint.

Dieses zitieren