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

Deep Relational Metric Learning

About

This paper presents a deep relational metric learning (DRML) framework for image clustering and retrieval. Most existing deep metric learning methods learn an embedding space with a general objective of increasing interclass distances and decreasing intraclass distances. However, the conventional losses of metric learning usually suppress intraclass variations which might be helpful to identify samples of unseen classes. To address this problem, we propose to adaptively learn an ensemble of features that characterizes an image from different aspects to model both interclass and intraclass distributions. We further employ a relational module to capture the correlations among each feature in the ensemble and construct a graph to represent an image. We then perform relational inference on the graph to integrate the ensemble and obtain a relation-aware embedding to measure the similarities. Extensive experiments on the widely-used CUB-200-2011, Cars196, and Stanford Online Products datasets demonstrate that our framework improves existing deep metric learning methods and achieves very competitive results.

Wenzhao Zheng, Borui Zhang, Jiwen Lu, Jie Zhou• 2021

Related benchmarks

TaskDatasetResultRank
Image RetrievalCUB-200-2011 (test)
Recall@168.7
251
Image RetrievalStanford Online Products (test)
Recall@179.9
220
Image RetrievalCUB-200 2011
Recall@168.7
146
Deep Metric LearningCUB200 2011 (test)
Recall@168.7
129
Image RetrievalCARS 196
Recall@186.9
98
Image RetrievalCUB
Recall@168.7
87
Image RetrievalCARS 196 (test)
Recall@186.9
56
Deep Metric LearningCARS196 (test)
R@186.9
56
Image RetrievalSOP
Recall@171.5
32
Image RetrievalCars (test)
Recall@173.3
13
Showing 10 of 10 rows

Other info

Follow for update