Fine-grained Categorization and Dataset Bootstrapping using Deep Metric Learning with Humans in the Loop

被引:149
作者
Cui, Yin [1 ,2 ]
Zhou, Feng [3 ]
Lin, Yuanqing [3 ]
Belongie, Serge [1 ,2 ]
机构
[1] Cornell Univ, Dept Comp Sci, Ithaca, NY 14853 USA
[2] Cornell Tech, New York, NY 10011 USA
[3] NEC Labs Amer, Princeton, NJ USA
来源
2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR) | 2016年
关键词
D O I
10.1109/CVPR.2016.130
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Existing fine-grained visual categorization methods often suffer from three challenges: lack of training data, large number of fine-grained categories, and high intraclass vs. low inter-class variance. In this work we propose a generic iterative framework for fine-grained categorization and dataset bootstrapping that handles these three challenges. Using deep metric learning with humans in the loop, we learn a low dimensional feature embedding with anchor points on manifolds for each category. These anchor points capture intra-class variances and remain discriminative between classes. In each round, images with high confidence scores from our model are sent to humans for labeling. By comparing with exemplar images, labelers mark each candidate image as either a "true positive" or a "false positive." True positives are added into our current dataset and false positives are regarded as "hard negatives" for our metric learning model. Then the model is retrained with an expanded dataset and hard negatives for the next round. To demonstrate the effectiveness of the proposed framework, we bootstrap a fine-grained flower dataset with 620 categories from Instagram images. The proposed deep metric learning scheme is evaluated on both our dataset and the CUB-200-2001 Birds dataset. Experimental evaluations show significant performance gain using dataset bootstrapping and demonstrate state-of-the-art results achieved by the proposed deep metric learning methods.
引用
收藏
页码:1153 / 1162
页数:10
相关论文
共 49 条
  • [1] Angelova Anelia., 2013, WACV
  • [2] [Anonymous], 2006, CVPR
  • [3] [Anonymous], 2005, CVPR
  • [4] [Anonymous], P ROYAL SOC B
  • [5] [Anonymous], 2002, NIPS
  • [6] [Anonymous], 2009, JMLR
  • [7] [Anonymous], ICCV
  • [8] [Anonymous], 2011, TECHNICAL REPORT
  • [9] [Anonymous], 2008, ICVGIP
  • [10] [Anonymous], 2015, CVPR