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

区分度强制损失改进表示学习

Discriminability-enforcing loss to improve representation learning

University of Bucharest(布加勒斯特大学)

Florinel-Alin Croitoru, Diana-Nicoleta Grigore, Radu Tudor Ionescu

摘要:

在训练过程中,深度神经网络隐式地学习通过特征层次来表示输入数据样本,其中层次的大小由层数确定。在本文中,我们专注于加强高级表示的区分能力,这些表示通常由更深层(更接近输出)学习。为此,我们引入了一个新的损失项的灵感来自基尼杂质,其目的是最大限度地减少熵(增加区分能力)的个别高级别的功能相对于类标签。虽然我们的基尼损失导致了高区分度的特征,但它并不能确保高级特征的分布与类的分布相匹配。因此,我们引入另一个损失项来最小化两个分布之间的Kullback-Leibler散度。我们对两个图像分类数据集(CIFAR-100和Caltech 101)进行了实验,考虑了从卷积网络(ResNet-17,ResNet-18,ResNet-50)到Transformers(CvT)的多种神经架构。我们的实证结果表明,将我们的新损失项集成到训练目标中始终优于单独使用交叉熵训练的模型,而根本不会增加推理时间。

英文摘要:

During the training process, deep neural networks implicitly learn to represent the input data samples through a hierarchy of features, where the size of the hierarchy is determined by the number of layers. In this paper, we focus on enforcing the discriminative power of the high-level representations, that are typically learned by the deeper layers (closer to the output). To this end, we introduce a new loss term inspired by the Gini impurity, which is aimed at minimizing the entropy (increasing the discriminative power) of individual high-level features with respect to the class labels. Although our Gini loss induces highly-discriminative features, it does not ensure that the distribution of the high-level features matches the distribution of the classes. As such, we introduce another loss term to minimize the Kullback-Leibler divergence between the two distributions. We conduct experiments on two image classification data sets (CIFAR-100 and Caltech 101), considering multiple neural architectures ranging from convolutional networks (ResNet-17, ResNet-18, ResNet-50) to transformers (CvT). Our empirical results show that integrating our novel loss terms into the training objective consistently outperforms the models trained with cross-entropy alone, without increasing the inference time at all.

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

↑