Few-Shot Classification Based on Feature Enhancement Network

Shasha Hu, Han Su, Ruixuan Gao

Abstract: Few-shot image classification stands as a pivotal task within the realm of computer vision. However, obtaining accurate class prototypes from limited annotated samples is a challenging problem. In recent years, many methods based on prototype networks have shown excellent performance. Nevertheless, existing methods overlook the discriminative semantic information lost due to sample scarcity and the hidden category information in the query set, failing to address the issue of unreliable prototypes generated from limited annotated samples. In this paper, we propose a feature enhancement network for few-shot classification. To improve the accuracy and robustness of few-shot classification models, we first enhance the support set through learning a weight matrix and then align the enhanced support set prototypes with textual semantics. To avoid being influenced by introduced prior noise, we fuse between semantically aligned prototypes and mean prototypes and ultimately utilize query prototypes for dynamic updating to obtain more accurate class prototypes. Extensive experiments demonstrate that our method achieves competitive performance on miniImageNet and tieredImageNet datasets. Furthermore, it exhibits excellent results in cross-domain few-shot classification.