arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2005.06047cs.CVcs.MM
收藏

基于基元发现和增强的组合Few-Shot识别

Compositional Few-Shot Recognition with Primitive Discovery and Enhancing

Peking University(北京大学);University of California, Berkeley(加州大学伯克利分校);South China University of Technology(华南理工大学);PengCheng Laboratory(鹏程实验室);Carnegie Mellon University(卡内基梅隆大学)

Yixiong Zou, Shanghang Zhang, Ke Chen, Yonghong Tian, Yaowei Wang, José M. F. Moura

摘要:

少样本学习(FSL)的目标是在只有少量训练样本的情况下识别新类,这对深度学习来说仍然是一个巨大的挑战。然而,人类可以很容易地识别新的类,只有很少的样本。这种能力的一个关键组成部分是人类可以执行的成分识别,这在认知科学中得到了很好的研究,但在FSL中没有得到很好的探索。受人类这种能力的启发,为了模仿人类学习视觉基元和组成基元以识别新类的能力,我们提出了一种FSL方法,用于学习由重要基元组成的特征表示,该特征表示由基元发现和基元增强两部分联合训练。在基元发现中,我们专注于通过图像分割顺序的自我监督来学习与对象部分相关的基元,避免额外的费力注释并减轻语义间隙的影响。在原始增强中,受当前关于深度网络可解释性的研究的启发,我们为FSL基线模型提供了我们的组成视图。为了修改这个模型的有效组成,启发数学推导和生物学研究(赫布学习规则和赢家通吃机制),我们提出了一个软组合机制,通过扩大激活的重要原语,而减少其他的,以提高重要原语的影响力,更好地利用这些原语组成新的类。在公共基准上进行了大量的实验,在Few-Shot图像分类和视频识别任务。我们的方法在所有这些数据集上都达到了最先进的性能,并显示出更好的可解释性。

英文摘要:

Few-shot learning (FSL) aims at recognizing novel classes given only few training samples, which still remains a great challenge for deep learning. However, humans can easily recognize novel classes with only few samples. A key component of such ability is the compositional recognition that human can perform, which has been well studied in cognitive science but is not well explored in FSL. Inspired by such capability of humans, to imitate humans' ability of learning visual primitives and composing primitives to recognize novel classes, we propose an approach to FSL to learn a feature representation composed of important primitives, which is jointly trained with two parts, i.e. primitive discovery and primitive enhancing. In primitive discovery, we focus on learning primitives related to object parts by self-supervision from the order of image splits, avoiding extra laborious annotations and alleviating the effect of semantic gaps. In primitive enhancing, inspired by current studies on the interpretability of deep networks, we provide our composition view for the FSL baseline model. To modify this model for effective composition, inspired by both mathematical deduction and biological studies (the Hebbian Learning rule and the Winner-Take-All mechanism), we propose a soft composition mechanism by enlarging the activation of important primitives while reducing that of others, so as to enhance the influence of important primitives and better utilize these primitives to compose novel classes. Extensive experiments on public benchmarks are conducted on both the few-shot image classification and video recognition tasks. Our method achieves the state-of-the-art performance on all these datasets and shows better interpretability.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑