Journal of Xidian University ›› 2022, Vol. 49 ›› Issue (5): 166-174.doi: 10.19665/j.issn1001-2400.2022.05.019

• Computer Science and Technology & Artificial Intelligence • Previous Articles     Next Articles

Algorithm for classification of few-shot images by dynamic subspace

REN Jiaxing1(),CAO Yudong1(),CAO Rui2(),YAN Jia1()   

  1. 1. School of Electronics and Information Engineering,Liaoning University of Technology,Jinzhou 121001,China
    2. School of Automation and Electrical Engineering,Dalian Jiaotong University,Dalian 116028,China
  • Received:2021-09-30 Online:2022-10-20 Published:2022-11-17

Abstract:

The existing few-shot image classification algorithms based on metric learning have a low precision of image categorization and weak generalization performance.A few-shot image classification algorithm by dynamic subspace is proposed in this paper.First,a residual neural network is used to extract few-shot image features.The dynamically orthogonalized projection subspaces representing image categories are generated with decomposed image features of various categorizations so as to enhance the difference of features among categories in orthogonalized projection subspaces.Second,a dynamic subspace classifier based on few-shot learning is constructed by fusing the subspace loss function and the cross-entropy loss function so as to enhance the similarity of samples in the same category.The inter-class distance of the subspace is dynamically updated with the change of sampling amount and sample similarity.Finally,the feature vector of the target image is input into the dynamic subspace classifier,and the Euclidean distance square and the soft max function are used to calculate the category probability of the target feature and predict its category.Performance testing is performed on the few-shot data sets such as mini-ImageNet,CIFAR-100 and Pascal VOC2007.The proposed algorithm is superior to the current mainstream few-shot image classification algorithm,and the average classification precision of the proposed algorithm is 2.3% higher than that of the current DSN with good performance under 5-way 5-shot.Experiments show that the proposed algorithm has a strong generalization performance and an anti-interference ability.

Key words: dynamic subspace, few-shot data, orthogonal projection, image classification, residual network

CLC Number: 

  • TP391