College of Sciences, China Jiliang University, Hangzhou 310018, Zhejiang, China.
Neural Netw. 2021 Dec;144:755-765. doi: 10.1016/j.neunet.2021.09.029. Epub 2021 Oct 7.
Deep learning has shown its great potential in the field of image classification due to its powerful feature extraction ability, which heavily depends on the number of available training samples. However, it is still a huge challenge on how to obtain an effective feature representation and further learn a promising classifier by deep networks when faced with few-shot classification tasks. This paper proposes a multi-features adaptive aggregation meta-learning method with an information enhancer for few-shot classification tasks, referred to as MFAML. It contains three main modules, including a feature extraction module, an information enhancer, and a multi-features adaptive aggregation classifier (MFAAC). During the meta-training stage, the information enhancer comprised of some deconvolutional layers is designed to promote the effective utilization of samples and thereby capturing more valuable information in the process of feature extraction. Simultaneously, the MFAAC module integrates the features from several convolutional layers of the feature extraction module. The obtained features then feed into the similarity module so that implementing the adaptive adjustment of the predicted label. The information enhancer and MFAAC are connected by a hybrid loss, providing an excellent feature representation. During the meta-test stage, the information enhancer is removed and we keep the remaining architecture for fast adaption on the final target task. The whole MFAML framework is solved by the optimization strategy of model-agnostic meta-learner (MAML) and can effectively improve generalization performance. Experimental results on several benchmark datasets demonstrate the superiority of the proposed method over other representative few-shot classification methods.
深度学习在图像分类领域由于其强大的特征提取能力而显示出巨大的潜力,这主要依赖于可用的训练样本数量。然而,当面临少样本分类任务时,如何通过深度网络获得有效的特征表示并进一步学习有前途的分类器仍然是一个巨大的挑战。本文提出了一种用于少样本分类任务的多特征自适应聚合元学习方法,称为 MFAML。它包含三个主要模块,包括特征提取模块、信息增强器和多特征自适应聚合分类器(MFAAC)。在元训练阶段,包含一些去卷积层的信息增强器被设计用于促进样本的有效利用,从而在特征提取过程中捕获更多有价值的信息。同时,MFAAC 模块集成了特征提取模块中几个卷积层的特征。获得的特征然后输入到相似度模块中,以便实现预测标签的自适应调整。信息增强器和 MFAAC 通过混合损失连接,提供了优秀的特征表示。在元测试阶段,会移除信息增强器,并保留其余的架构用于在最终目标任务上快速适应。整个 MFAML 框架通过模型不可知元学习器(MAML)的优化策略来解决,可以有效地提高泛化性能。在几个基准数据集上的实验结果表明,该方法优于其他有代表性的少样本分类方法。