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

使用卷积神经网络学习类层次结构

Learn Class Hierarchy using Convolutional Neural Networks

University of Insubria(因苏布里亚大学)

Riccardo La Grassa, Ignazio Gallo, Nicola Landro

摘要:

卷积神经网络的大量研究都集中在多类域的平面分类上。在现实世界中,许多问题自然地表示为分层分类问题,其中要预测的类被组织在类的层次结构中。在本文中,我们提出了一种新的架构,分层分类的图像,引入了一个堆栈的深度线性层与交叉熵损失函数和中心损失相结合。所提出的架构可以扩展任何神经网络模型,并同时优化损失函数,以发现局部层次的类关系和损失函数,以发现全局信息,从整个类层次结构,同时惩罚类层次结构的违规行为。我们的实验表明,我们的分层分类器的优势,传统的分类方法找到应用在计算机视觉任务。

英文摘要:

A large amount of research on Convolutional Neural Networks has focused on flat Classification in the multi-class domain. In the real world, many problems are naturally expressed as problems of hierarchical classification, in which the classes to be predicted are organized in a hierarchy of classes. In this paper, we propose a new architecture for hierarchical classification of images, introducing a stack of deep linear layers with cross-entropy loss functions and center loss combined. The proposed architecture can extend any neural network model and simultaneously optimizes loss functions to discover local hierarchical class relationships and a loss function to discover global information from the whole class hierarchy while penalizing class hierarchy violations. We experimentally show that our hierarchical classifier presents advantages to the traditional classification approaches finding application in computer vision tasks.

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

↑