Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

A unifying mutual information view of metric learning: cross-entropy vs. pairwise losses

About

Recently, substantial research efforts in Deep Metric Learning (DML) focused on designing complex pairwise-distance losses, which require convoluted schemes to ease optimization, such as sample mining or pair weighting. The standard cross-entropy loss for classification has been largely overlooked in DML. On the surface, the cross-entropy may seem unrelated and irrelevant to metric learning as it does not explicitly involve pairwise distances. However, we provide a theoretical analysis that links the cross-entropy to several well-known and recent pairwise losses. Our connections are drawn from two different perspectives: one based on an explicit optimization insight; the other on discriminative and generative views of the mutual information between the labels and the learned features. First, we explicitly demonstrate that the cross-entropy is an upper bound on a new pairwise loss, which has a structure similar to various pairwise losses: it minimizes intra-class distances while maximizing inter-class distances. As a result, minimizing the cross-entropy can be seen as an approximate bound-optimization (or Majorize-Minimize) algorithm for minimizing this pairwise loss. Second, we show that, more generally, minimizing the cross-entropy is actually equivalent to maximizing the mutual information, to which we connect several well-known pairwise losses. Furthermore, we show that various standard pairwise losses can be explicitly related to one another via bound relationships. Our findings indicate that the cross-entropy represents a proxy for maximizing the mutual information -- as pairwise losses do -- without the need for convoluted sample-mining heuristics. Our experiments over four standard DML benchmarks strongly support our findings. We obtain state-of-the-art results, outperforming recent and complex DML methods.

Malik Boudiaf, J\'er\^ome Rony, Imtiaz Masud Ziko, Eric Granger, Marco Pedersoli, Pablo Piantanida, Ismail Ben Ayed• 2020

Related benchmarks

TaskDatasetResultRank
Image RetrievalCUB-200-2011 (test)
Recall@169.2
251
Image RetrievalStanford Online Products (test)
Recall@181.1
220
Image RetrievalCARS196 (test)
Recall@189.3
134
In-shop clothes retrievalin-shop clothes retrieval dataset (test)
Recall@190.6
78
Image RetrievalSOP (test)
Recall@181.1
42
Image RetrievalStanford Online Products (SOP) standard (test)
Recall@181.1
27
Image RetrievalCars196 standard (test)
Recall@189.3
23
Object instance retrievalStanford Online Products (SOP) (test)
R@181.1
13
Image RetrievalCaltech-UCSD Birds-200 2011 (test)
Recall@169.2
12
Showing 9 of 9 rows

Other info

Code

Follow for update