English

Dissecting the impact of different loss functions with gradient surgery

Computer Vision and Pattern Recognition 2022-01-28 v1 Artificial Intelligence Information Retrieval Machine Learning

Abstract

Pair-wise loss is an approach to metric learning that learns a semantic embedding by optimizing a loss function that encourages images from the same semantic class to be mapped closer than images from different classes. The literature reports a large and growing set of variations of the pair-wise loss strategies. Here we decompose the gradient of these loss functions into components that relate to how they push the relative feature positions of the anchor-positive and anchor-negative pairs. This decomposition allows the unification of a large collection of current pair-wise loss functions. Additionally, explicitly constructing pair-wise gradient updates to separate out these effects gives insights into which have the biggest impact, and leads to a simple algorithm that beats the state of the art for image retrieval on the CAR, CUB and Stanford Online products datasets.

Keywords

Cite

@article{arxiv.2201.11307,
  title  = {Dissecting the impact of different loss functions with gradient surgery},
  author = {Hong Xuan and Robert Pless},
  journal= {arXiv preprint arXiv:2201.11307},
  year   = {2022}
}
R2 v1 2026-06-24T09:04:49.957Z