Catching Image Retrieval Generalization
Machine Learning
2023-06-26 v1 Computer Vision and Pattern Recognition
Abstract
The concepts of overfitting and generalization are vital for evaluating machine learning models. In this work, we show that the popular Recall@K metric depends on the number of classes in the dataset, which limits its ability to estimate generalization. To fix this issue, we propose a new metric, which measures retrieval performance, and, unlike Recall@K, estimates generalization. We apply the proposed metric to popular image retrieval methods and provide new insights about deep metric learning generalization.
Cite
@article{arxiv.2306.13357,
title = {Catching Image Retrieval Generalization},
author = {Maksim Zhdanov and Ivan Karpukhin},
journal= {arXiv preprint arXiv:2306.13357},
year = {2023}
}
Comments
4 pages, 3 figures, 2 tables