Multi-level Attention Feature Network for Few-shot Learning

被引:2
|
作者
Wang Ronggui [1 ]
Han Mengya [1 ]
Yang Juan [1 ]
Xue Lixia [1 ]
Hu Min [1 ]
机构
[1] Hefei Univ Technol, Sch Comp & Informat, Hefei 230009, Peoples R China
基金
中国国家自然科学基金;
关键词
Image processing; Multi-scale images; Few-shot learning; Multi-level attention feature; Similarity metric;
D O I
10.11999/JEIT190242
中图分类号
TM [电工技术]; TN [电子技术、通信技术];
学科分类号
0808 ; 0809 ;
摘要
Existing few-shot methods have problems that feature extraction scale is single, the learned class representations are inaccurate, the similarity calculation still relies on standard metrics. In order to solve the above problems, multi-level attention feature network is proposed. Firstly, the multiple scale images are obtained by scale processing, the features of multiple scale images are extracted and the image-level attention features are obtained by the image-level attention mechanism to fusion them. Then, class-level attention features are learned by using the class- level attention mechanism. Finally, the classification is performed by using the network to compute the similarity scores between features. The proposed method is evaluated on the Omniglot dataset and the MiniImagenet dataset. The experimental results show that multi-level attention feature network can further improve the classification accuracy under small sample conditions compared to the single-scale image features and average prototypes.
引用
收藏
页码:772 / 778
页数:7
相关论文
共 16 条
  • [1] [Anonymous], DYNAMIC INPUT STRUCT
  • [2] [Anonymous], OPTIMIZATION MODEL F
  • [3] AGA : Attribute-Guided Augmentation
    Dixit, Mandar
    Kwitt, Roland
    Niethammer, Marc
    Vasconcelos, Nuno
    [J]. 30TH IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2017), 2017, : 3328 - 3336
  • [4] Finn C, 2017, PR MACH LEARN RES, V70
  • [5] Fast R-CNN
    Girshick, Ross
    [J]. 2015 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV), 2015, : 1440 - 1448
  • [6] Low-shot Visual Recognition by Shrinking and Hallucinating Features
    Hariharan, Bharath
    Girshick, Ross
    [J]. 2017 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV), 2017, : 3037 - 3046
  • [7] He D, 2016, ADV NEUR IN, V29
  • [8] Densely Connected Convolutional Networks
    Huang, Gao
    Liu, Zhuang
    van der Maaten, Laurens
    Weinberger, Kilian Q.
    [J]. 30TH IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2017), 2017, : 2261 - 2269
  • [9] Koch G, 2015, P DEEP LEARN WORKSH, V2
  • [10] One-shot learning of object categories
    Li, FF
    Fergus, R
    Perona, P
    [J]. IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2006, 28 (04) : 594 - 611