Hierarchical Proxy-based Loss for Deep Metric Learning

by   Zhibo Yang, et al.

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.



There are no comments yet.


page 8


Proxy Anchor Loss for Deep Metric Learning

Existing metric learning losses can be categorized into two classes: pai...

Proxy Synthesis: Learning with Synthetic Classes for Deep Metric Learning

One of the main purposes of deep metric learning is to construct an embe...

Smooth Proxy-Anchor Loss for Noisy Metric Learning

Many industrial applications use Metric Learning as a way to circumvent ...

Multi-Head Deep Metric Learning Using Global and Local Representations

Deep Metric Learning (DML) models often require strong local and global ...

Triplet-Center Loss for Multi-View 3D Object Retrieval

Most existing 3D object recognition algorithms focus on leveraging the s...

Mask Proxy Loss for Text-Independent Speaker Recognition

Open-set speaker recognition can be regarded as a metric learning proble...

VeriMedi: Pill Identification using Proxy-based Deep Metric Learning and Exact Solution

We present the system that we have developed for the identification and ...
This week in AI

Get the week's most popular data science and artificial intelligence research sent straight to your inbox every Saturday.