arXiv · 2307.07620
Generalizable Embeddings with Cross-batch Metric Learning
Abstract
Global average pooling (GAP) is a popular component in deep metric learning (DML) for aggregating features. Its effectiveness is often attributed to treating each feature vector as a distinct semantic entity and GAP as a combination of them. Albeit substantiated, such an explanation's algorithmic implications to learn generalizable entities to represent unseen classes, a crucial DML goal, remain unclear. To address this, we formulate GAP as a convex combination of learnable prototypes. We then show that the prototype learning can be expressed as a recursive process fitting a linear predictor to a batch of samples. Building on that perspective, we consider two batches of disjoint classes at each iteration and regularize the learning by expressing the samples of a batch with the prototypes that are fitted to the other batch. We validate our approach on 4 popular DML benchmarks.
Explore related subjects
Keep this discovery
Yeti Z. Gurbuz, A. Aydin Alatan. 2023-07-14. Generalizable Embeddings with Cross-batch Metric Learning. https://arxiv.org/abs/2307.07620
Cite the original work for its findings. Save a collection to share your selection of sources.