References & Citations
Computer Science > Computer Vision and Pattern Recognition
Title: Hierarchical Proxy-based Loss for Deep Metric Learning
(Submitted on 25 Mar 2021 (v1), last revised 18 Oct 2021 (this version, v3))
Abstract: Proxy-based metric learning losses are superior to pair-based losses due to their fast convergence and low training complexity. However, existing proxy-based losses focus on learning class-discriminative features while overlooking the commonalities shared across classes which are potentially useful in describing and matching samples. Moreover, they ignore the implicit hierarchy of categories in real-world datasets, where similar subordinate classes can be grouped together. In this paper, we present a framework that leverages this implicit hierarchy by imposing a hierarchical structure on the proxies and can be used with any existing proxy-based loss. This allows our model to capture both class-discriminative features and class-shared characteristics without breaking the implicit data hierarchy. We evaluate our method on five established image retrieval datasets such as In-Shop and SOP. Results demonstrate that our hierarchical proxy-based loss framework improves the performance of existing proxy-based losses, especially on large datasets which exhibit strong hierarchical structure.
Submission history
From: Zhibo Yang [view email][v1] Thu, 25 Mar 2021 00:38:33 GMT (1966kb,D)
[v2] Wed, 18 Aug 2021 21:21:02 GMT (2262kb,D)
[v3] Mon, 18 Oct 2021 02:40:07 GMT (2270kb,D)
Link back to: arXiv, form interface, contact.