An Embarrassingly Simple Baseline to One-shot Learning

被引:5
作者
Liu, Chen [1 ]
Xu, Chengming [1 ]
Wang, Yikai [1 ,4 ]
Zhang, Li [2 ]
Fu, Yanwei [1 ,3 ,4 ]
机构
[1] Fudan Univ, Sch Data Sci, Shanghai, Peoples R China
[2] Univ Oxford, Dept Engn Sci, Oxford, England
[3] Fudan Univ, MOE Frontiers Ctr Brain Sci, Shanghai, Peoples R China
[4] Fudan Univ, Shanghai Key Lab Intelligent Informat Proc, Shanghai, Peoples R China
来源
2020 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION WORKSHOPS (CVPRW 2020) | 2020年
关键词
D O I
10.1109/CVPRW50498.2020.00469
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
In this paper, we propose an embarrassingly simple approach for one-shot learning. Our insight is that the one-shot tasks have domain gap to the network pretrained tasks and thus some features from the pretrained network are not relevant, or harmful to the specific one-shot task. Therefore, we propose to directly prune the features from the pretrained network for a specific one-shot task rather than update it via an optimized scheme with complex network structure. Without bells and whistles, our simple yet effective method achieves leading performances on miniImageNet (60.63%) and tieredImageNet (69.02%) for 5-way one-shot setting. The best trial can hit to 66.83% on miniImageNet and 74.04% on tieredImageNet, establishing a new state-of-the-art. We strongly advocate that our method can serve as a strong baseline for one-shot learning. The codes and trained models will be released at http://github.com/corwinliu9669/embarrassingly-simple-baseline.
引用
收藏
页码:4005 / 4009
页数:5
相关论文
共 34 条
  • [1] [Anonymous], 2018, NEURIPS
  • [2] Large-Scale Machine Learning with Stochastic Gradient Descent
    Bottou, Leon
    [J]. COMPSTAT'2010: 19TH INTERNATIONAL CONFERENCE ON COMPUTATIONAL STATISTICS, 2010, : 177 - 186
  • [3] FlyMap: Interacting with Maps Projected from a Drone
    Brock, Anke M.
    Chatain, Julia
    Park, Michelle
    Fang, Tommy
    Hachet, Martin
    Landay, James A.
    Cauchard, Jessica R.
    [J]. PROCEEDINGS PERVASIVE DISPLAYS 2018: THE 7TH ACM INTERNATIONAL SYMPOSIUM ON PERVASIVE DISPLAYS, 2018,
  • [4] Chen Da, 2019, ARXIV191208395
  • [5] Finn C, 2017, PR MACH LEARN RES, V70
  • [6] Goodfellow I.J., 2014, ADV NEUR IN, p1406.2661, DOI DOI 10.1145/3422622
  • [7] He K., 2016, PROC IEEE COMPUT SOC, P1026, DOI [10.1109/cvpr.2016.90., DOI 10.1109/CVPR.2016.90]
  • [8] Strip Pooling: Rethinking Spatial Pooling for Scene Parsing
    Hou, Qibin
    Zhang, Li
    Cheng, Ming-Ming
    Feng, Jiashi
    [J]. 2020 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2020, : 4002 - 4011
  • [9] Hou RB, 2019, ADV NEUR IN, V32
  • [10] Ioffe S, 2015, PR MACH LEARN RES, V37, P448