arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2002.08473cs.CV

Revisiting Training Strategies and Generalization Performance in Deep Metric Learning

Mila, Université de Montréal(蒙特利尔大学 Mila 研究院);HCI/IWR, Heidelberg University(海德堡大学 HCI/IWR);University of Toronto(多伦多大学);The Alan Turing Institute, University of Oxford(牛津大学艾伦·图灵研究所)

Karsten Roth, Timo Milbich, Samarth Sinha, Prateek Gupta, Björn Ommer, Joseph Paul Cohen

英文摘要:

Deep Metric Learning (DML) is arguably one of the most influential lines of research for learning visual similarities with many proposed approaches every year. Although the field benefits from the rapid progress, the divergence in training protocols, architectures, and parameter choices make an unbiased comparison difficult. To provide a consistent reference point, we revisit the most widely used DML objective functions and conduct a study of the crucial parameter choices as well as the commonly neglected mini-batch sampling process. Under consistent comparison, DML objectives show much higher saturation than indicated by literature. Further based on our analysis, we uncover a correlation between the embedding space density and compression to the generalization performance of DML models. Exploiting these insights, we propose a simple, yet effective, training regularization to reliably boost the performance of ranking-based DML models on various standard benchmark datasets. Code and a publicly accessible WandB-repo are available at https://github.com/Confusezius/Revisiting_Deep_Metric_Learning_PyTorch.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑