论文标题
零光学习的增强多样性的泛化特殊化平衡
Diversity-boosted Generalization-Specialization Balancing for Zero-shot Learning
论文作者
论文摘要
零射门学习(ZSL)旨在将分类能力从可见的类别转移到看不见的类别。最近的方法证明,概括和专业化是在ZSL中实现良好性能的两种基本能力。但是,仅关注其中一种能力可能会导致模型过于笼统,具有降级分类能力,或者太专业而无法概括以至于看不见的类别。在本文中,我们提出了一个端到端网络,称为BGSNET,该网络在实例和数据集级别上为概括和专业化能力提供了平衡。具体而言,BGSNET由两个分支组成:概括网络(GNET),该分支将情节性元学习用于学习通用知识,以及平衡的专业化网络(BSNET),该网络(BSNET)采用多个专注的提取器来提取歧视性特征并实现实例级别的平衡。一种新颖的自调整多样性损失旨在优化冗余降低和多样性的BSNET。我们进一步提出了一个可区分的数据集级别平衡,并在线性退火计划中更新权重,以模拟网络修剪,从而获得BSNET的最佳结构,并实现了数据集级别的平衡。四个基准数据集的实验证明了我们的模型的有效性。足够的组件消融证明了整合和平衡概括和专业化能力的必要性。
Zero-Shot Learning (ZSL) aims to transfer classification capability from seen to unseen classes. Recent methods have proved that generalization and specialization are two essential abilities to achieve good performance in ZSL. However, focusing on only one of the abilities may result in models that are either too general with degraded classification ability or too specialized to generalize to unseen classes. In this paper, we propose an end-to-end network, termed as BGSNet, which equips and balances generalization and specialization abilities at the instance and dataset level. Specifically, BGSNet consists of two branches: the Generalization Network (GNet), which applies episodic meta-learning to learn generalized knowledge, and the Balanced Specialization Network (BSNet), which adopts multiple attentive extractors to extract discriminative features and achieve instance-level balance. A novel self-adjusted diversity loss is designed to optimize BSNet with redundancy reduced and diversity boosted. We further propose a differentiable dataset-level balance and update the weights in a linear annealing schedule to simulate network pruning and thus obtain the optimal structure for BSNet with dataset-level balance achieved. Experiments on four benchmark datasets demonstrate our model's effectiveness. Sufficient component ablations prove the necessity of integrating and balancing generalization and specialization abilities.