高级检索
郭璐. 基于困难样本对激励的小样本图像分类方法[J]. 计算机辅助设计与图形学学报.
引用本文: 郭璐. 基于困难样本对激励的小样本图像分类方法[J]. 计算机辅助设计与图形学学报.
GUO. A Few-Shot Image Classification Method by Hard Pairwise-Based Excitation[J]. Journal of Computer-Aided Design & Computer Graphics.
Citation: GUO. A Few-Shot Image Classification Method by Hard Pairwise-Based Excitation[J]. Journal of Computer-Aided Design & Computer Graphics.

基于困难样本对激励的小样本图像分类方法

A Few-Shot Image Classification Method by Hard Pairwise-Based Excitation

  • 摘要: 使用少量标签样本训练得到的传统模型往往预测精度低、泛化能力弱, 很难应用到实际生产中. 针对小样本图像提出一种基于困难样本对激励分类方法, 包括预训练阶段和元学习阶段. 预训练阶段在基类数据集上训练编码器, 并作为元学习阶段的初始特征编码器; 元学习阶段将进一步优化此编码器, 元训练过程使用本质特征法来降低异常样本对质心的影响;结合度量学习与元学习设计了困难样本对激励损失函数,从样本对角度出发, 在训练过程中引导模型扩大正负样本间距离, 使同类样本更加紧凑. 在公开数据集mini-ImageNet, tiered-ImageNet上进行实验, 分类精度分别为64.12%, 70.15%, 验证了所提方法的有效性和可行性.

     

    Abstract: Traditional models trained with a small number of label samples often have low prediction accuracy and weak generalization ability and are difficult to be applied to practical production. A classification method named hard pairwise-based excitation is proposed for the few-shot image classification, including pre-training stage and meta learning stage. Pre-training stage trains the encoder on the base class dataset and used as the initial feature en-coder in the meta learning stage; In the meta learning stage, the encoder will be further optimized, and the meta training process uses the essential feature method to reduce the impact of abnormal samples on the centroid; Combining measurement learning and meta learning, a loss function named hard sample-pairs excitation is de-signed. From the perspective of sample pairs, the model is guided to expand the distance between positive and negative samples during the training process, making similar samples more compact. Experiments are conducted on public datasets mini-ImageNet and tiered-ImageNet, and the classification accuracy is 64.12% and 70.15%, respectively, verifying the effectiveness and feasibility of the proposed method.

     

/

返回文章
返回