ICML2021
Connecting Sphere Manifolds Hierarchically for Regularization
Damien Scieur, Youngsung Kim
3 citations
Abstract
This paper considers classification problems with hierarchically organized classes. We force the classifier (hyperplane) of each class to belong to a sphere manifold, whose center is the classifier of its super-class. Then, individual sphere manifolds are connected based on their hierarchical relations. Our technique replaces the last layer of a neural network by combining a spherical fully-connected layer with a hierarchical layer. This regularization is shown to improve the performance of widely used deep neural network architectures (ResNet and DenseNet) on publicly available datasets (CI-FAR100, CUB200, Stanford dogs, Stanford cars, and Tiny-ImageNet).