1. LM-Metric: Learned pair weighting and contextual memory for deep metric learning.
- Author
-
Yan, Shiyang, Xu, Lin, Shu, Xinyao, Lu, Zhenyu, and Shen, Jialie
- Subjects
- *
COMPUTER vision , *DEEP learning , *IMAGE retrieval , *CONTEXTUAL learning , *MACHINE learning - Abstract
Deep Metric Learning (DML) is a crucial machine learning method in computer vision, and constructing an effective loss function. Average Precision (AP) is a well-known evaluation metric for image retrieval tasks. However, AP's non-differentiable and non-decomposable nature constrains its potential for adoption as an optimization goal. We propose a deep metric learning method with learned parametric metric learning loss and a contextual memory block (LM-Metric) for large-scale image retrieval tasks, which overcome AP's drawbacks and integrate AP within DML loss. We first introduce a parametric pairwise weighting scheme via policy gradient optimization and model the batch-wise inter-sample relationship via a Gated Recurrent Unit (GRU). Furthermore, a conditional Normalizing Flow-based contextual memory feature block to learn a compact single embedding for each image containing the contextual information during retrieval. We perform experiments on retrieval benchmark datasets and improve performance over the state-of-the-arts. • We propose a policy gradient-based meta-learned parametric metric learning scheme (LM-Metric). • We consider the inter-sample relationship during training to solve the non-decomposability problem of AP optimization. • We validate the effectiveness of our method, by applying it to variouis (DML) loss, e.g., Triplet, Margin, and Proxy-anchor losses. [ABSTRACT FROM AUTHOR]
- Published
- 2024
- Full Text
- View/download PDF