Category Relevance Redirection Network for Few-Shot Classification

Xiangtao Tian,Zhengli Zhao,Peng Jiang,Jianxin Wu,Guoqiang Zhong
DOI: https://doi.org/10.1109/ACCESS.2022.3199003
IF: 3.9
2022-01-01
IEEE Access
Abstract:The metric learning method has been successfully developed in few-shot learning by learning the deeply embedded metric space. However, many methods ignore the relationship between the support set and the query set. In this paper, we propose the category relevance redirection network (CRRNet), which is a novel metric learning-based few-shot learning classification network. The learning of CRRNet includes two stages. In the category relevance pre-training stage, the convolutional block attention module (CBAM) and the classifier are jointly trained using the center loss and cross entropy loss. In the category redirection training stage, the meta fusion block and the weight generator are used to represent the weighted query set features, which are then input into the spatial attention generator to obtain the discriminative features for classification. CRRNet can not only learn the relevant category characteristics of the support set samples, but emphasizes the relevant category characteristics of the current task's query samples. Experiments on three datasets show that CRRNet can greatly improve the embedding learning performance and classification accuracy of metric-based few-shot classification. For the 5-way 1-shot and 5-way 5-shot scenarios, CRRNet may produce 9.68 percent and 12.42 percent increases over the traditional methods on the miniImageNet dataset, respectively.
What problem does this paper attempt to address?