Few-shot image classification based on prototype embedding graph network
DOI:
CSTR:
Author:
Affiliation:

1.School of Data Science and Information Engineering, Guizhou Minzu University,Guiyang 550025, China; 2.Key Laboratory of Pattern Recognition and Intelligent System, Guizhou Minzu University,Guiyang 550025, China

Clc Number:

TP391.4

Fund Project:

  • Article
  • |
  • Figures
  • |
  • Metrics
  • |
  • Reference
  • |
  • Related
  • |
  • Cited by
  • |
  • Materials
  • |
  • Comments
    Abstract:

    Traditional backbone convolutional networks for feature extraction in few-shot image classification suffer from the problems of single context information, limited receptive field, and lack of global edge feature similarity measurement. In this paper, we propose a few-shot image classification algorithm based on prototype embedding graph network. First, the feature obtained by multiplying the weight values generated by CBAM with the features of different scales obtained by ASPP at different sampling rates is used as the node embedding feature of the graph network. Then, the prototype nodes are constructed in the measurement module using the prototype network method, transforming the similarity calculation between paired nodes into the sum of the similarity between a single node and the prototype node, and using the obtained similarity as the edge feature input to the graph neural network. Finally, the label information is propagated from labeled samples to unlabeled samples through the dual graph structure after multiple update iterations. In the classification task using ResNet-12 as the backbone convolutional network, our algorithm achieves classification accuracies of 71.47%, 75.41%, 86.21%, and 79.84% on the miniImageNet, tieredImageNet, CUB-200-2011, and CIFAR-FS datasets, respectively, for the 5way-1shot task. In the backbone convolutional network using Conv-4, our proposed algorithm outperforms existing graph network methods in both 5way-1shot and 5way-5shot tasks.

    Reference
    Related
    Cited by
Get Citation
Share
Article Metrics
  • Abstract:
  • PDF:
  • HTML:
  • Cited by:
History
  • Received:
  • Revised:
  • Adopted:
  • Online: June 05,2024
  • Published:
Article QR Code