Deep Metric Learning to Rank

被引:150
|
作者
Cakir, Fatih [1 ]
He, Kun [2 ,3 ]
Xia, Xide [2 ]
Kulis, Brian [2 ]
Sclaroff, Stan [2 ]
机构
[1] FirstFuel, Lexington, MA 02420 USA
[2] Boston Univ, Boston, MA 02215 USA
[3] Facebook Real Labs, Pittsburgh, PA USA
关键词
D O I
10.1109/CVPR.2019.00196
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
We propose a novel deep metric learning method by revisiting the learning to rank approach. Our method, named FastAP, optimizes the rank-based Average Precision measure, using an approximation derived from distance quantization. FastAP has a low complexity compared to existing methods, and is tailored for stochastic gradient descent. To fully exploit the benefits of the ranking formulation, we also propose a new minibatch sampling scheme, as well as a simple heuristic to enable large-batch training. On three few-shot image retrieval datasets, FastAP consistently outperforms competing methods, which often involve complex optimization heuristics or costly model ensembles.
引用
收藏
页码:1861 / 1870
页数:10
相关论文
共 50 条
  • [1] Deep Metric Learning using Similarities from Nonlinear Rank Approximations
    Schall, Konstantin
    Barthel, Kai Uwe
    Hezel, Nico
    Jung, Klaus
    [J]. 2019 IEEE 21ST INTERNATIONAL WORKSHOP ON MULTIMEDIA SIGNAL PROCESSING (MMSP 2019), 2019,
  • [2] METRIC LEARNING WITH RANK AND SPARSITY CONSTRAINTS
    Bah, Bubacarr
    Becker, Stephen
    Cevher, Volkan
    Goezcue, Baran
    [J]. 2014 IEEE INTERNATIONAL CONFERENCE ON ACOUSTICS, SPEECH AND SIGNAL PROCESSING (ICASSP), 2014,
  • [3] Compressed Self-Attention for Deep Metric Learning with Low-Rank Approximation
    Chen, Ziye
    Gong, Mingming
    Ge, Lingjuan
    Du, Bo
    [J]. PROCEEDINGS OF THE TWENTY-NINTH INTERNATIONAL JOINT CONFERENCE ON ARTIFICIAL INTELLIGENCE, 2020, : 2058 - 2064
  • [4] Metric -agnostic Learning -to -Rank via Boosting and Rank Approximation
    Gomez, Camilo
    Wang, Pcngyang
    Fu, Yanjie
    [J]. 23RD IEEE INTERNATIONAL CONFERENCE ON DATA MINING, ICDM 2023, 2023, : 1043 - 1048
  • [5] Deep Metric Learning Based on Rank-sensitive Optimization of Top-k Precision
    Muramoto, Naoki
    Yu, Hai-Tao
    [J]. CIKM '20: PROCEEDINGS OF THE 29TH ACM INTERNATIONAL CONFERENCE ON INFORMATION & KNOWLEDGE MANAGEMENT, 2020, : 2161 - 2164
  • [6] Adaptive Metric Learning with the Low Rank Constraint
    Fang, Yuan
    Yan, Yan
    Wang, Hanzi
    Chen, Si
    Gao, Xinbo
    [J]. 8TH INTERNATIONAL CONFERENCE ON INTERNET MULTIMEDIA COMPUTING AND SERVICE (ICIMCS2016), 2016, : 61 - 65
  • [7] Deep Meta Metric Learning
    Chen, Guangyi
    Zhang, Tianren
    Lu, Jiwen
    Zhou, Jie
    [J]. 2019 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2019), 2019, : 9546 - 9555
  • [8] Deep Transfer Metric Learning
    Hu, Junlin
    Lu, Jiwen
    Tan, Yap-Peng
    Zhou, Jie
    [J]. IEEE TRANSACTIONS ON IMAGE PROCESSING, 2016, 25 (12) : 5576 - 5588
  • [9] Deep Transfer Metric Learning
    Hu, Junlin
    Lu, Jiwen
    Tan, Yap-Peng
    [J]. 2015 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2015, : 325 - 333
  • [10] Deep Compositional Metric Learning
    Zheng, Wenzhao
    Wang, Chengkun
    Lu, Jiwen
    Zhou, Jie
    [J]. 2021 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, CVPR 2021, 2021, : 9316 - 9325