On the Robustness of Metric Learning: An Adversarial Perspective

被引:9
作者
Huai, Mengdi [1 ]
Zheng, Tianhang [2 ]
Miao, Chenglin [3 ]
Yao, Liuyi [4 ]
Zhang, Aidong [1 ]
机构
[1] Univ Virginia, Dept Comp Sci, 85 Engineers Way, Charlottesville, VA 22904 USA
[2] Univ Toronto, Dept Elect & Comp Engn, 10 Kings Coll Rd, Toronto, ON M5S 3G8, Canada
[3] Univ Georgia, Dept Comp Sci, Boyd Grad Studies Res Ctr, DW Brooks Dr, Athens, GA 30602 USA
[4] Alibaba Grp, 969 West Wen Yi Rd, Hangzhou 311121, Zhejiang, Peoples R China
基金
美国国家科学基金会;
关键词
Metric learning; robustness; adversarial perturbations;
D O I
10.1145/3502726
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Metric learning aims at automatically learning a distance metric from data so that the precise similarity between data instances can be faithfully reflected, and its importance has long been recognized in many fields. An implicit assumption in existing metric learning works is that the learned models are performed in a reliable and secure environment. However, the increasingly critical role of metric learning makes it susceptible to a risk of being malicious attacked. To well understand the performance of metric learning models in adversarial environments, in this article, we study the robustness of metric learning to adversarial perturbations, which are also known as the imperceptible changes to the input data that are crafted by an attacker to fool a well-learned model. However, different from traditional classification models, metric learning models take instance pairs rather than individual instances as input, and the perturbation on one instance may not necessarily affect the prediction result for an instance pair, which makes it more difficult to study the robustness of metric learning. To address this challenge, in this article, we first provide a definition of pairwise robustness for metric learning, and then propose a novel projected gradient descent-based attack method (called AckMetric) to evaluate the robustness of metric learning models. To further explore the capability of the attacker to change the prediction results, we also propose a theoretical framework to derive the upper bound of the pairwise adversarial loss. Finally, we incorporate the derived bound into the training process of metric learning and design a novel defense method to make the learned models more robust. Extensive experiments on real-world datasets demonstrate the effectiveness of the proposed methods.
引用
收藏
页数:25
相关论文
共 73 条
[21]   Learning Distance Metrics from Probabilistic Information [J].
Huai, Mengdi ;
Miao, Chenglin ;
Li, Yaliang ;
Suo, Qiuling ;
Su, Lu ;
Zhang, Aidong .
ACM TRANSACTIONS ON KNOWLEDGE DISCOVERY FROM DATA, 2020, 14 (05)
[22]  
Huai Mengdi, 2018, P 24 ACM SIGKDD INT
[23]  
Huang Chen, 2016, ADV NEUR IN, V29
[24]   Transfer metric learning for unsupervised domain adaptation [J].
Huang, Junchu ;
Zhou, Zhiheng .
IET IMAGE PROCESSING, 2019, 13 (05) :804-810
[25]   A Free Energy Based Approach for Distance Metric Learning [J].
Inaba, Sho ;
Fakhry, Carl T. ;
Kulkarni, Rahul V. ;
Zarringhalam, Kourosh .
KDD'19: PROCEEDINGS OF THE 25TH ACM SIGKDD INTERNATIONAL CONFERENCCE ON KNOWLEDGE DISCOVERY AND DATA MINING, 2019, :5-13
[26]  
Insua David Rios, 2020, ARXIV PREPRINT ARXIV
[27]   Deep Metric Learning with Angular Loss [J].
Wang, Jian ;
Zhou, Feng ;
Wen, Shilei ;
Liu, Xiao ;
Lin, Yuanqing .
2017 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV), 2017, :2612-2620
[28]   Deep Metric Learning Beyond Binary Supervision [J].
Kim, Sungyeon ;
Seo, Minkyo ;
Laptev, Ivan ;
Cho, Minsu ;
Kwak, Suha .
2019 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2019), 2019, :2283-2292
[29]   Attention-Based Ensemble for Deep Metric Learning [J].
Kim, Wonsik ;
Goyal, Bhavya ;
Chawla, Kunal ;
Lee, Jungmin ;
Kwon, Keunjoo .
COMPUTER VISION - ECCV 2018, PT I, 2018, 11205 :760-777
[30]   Transfer Metric Learning for Unseen Domains [J].
Kumagai, Atsutoshi ;
Iwata, Tomoharu ;
Fujiwara, Yasuhiro .
DATA SCIENCE AND ENGINEERING, 2020, 5 (02) :140-151