Connecting Sphere Manifolds Hierarchically for Regularization
Damien Scieur, Youngsung Kim
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).
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Cited by top-tier papers2
- Inducing Neural Collapse to a Fixed Hierarchy-Aware Frame for Reducing Mistake SeverityTong Liang, Jim DavisICCV 2023 · 14 citations
- Strongly Convex Sets in Riemannian ManifoldsDamien Scieur, David Martínez-Rubio, Thomas Kerdreux, Alexandre d'Aspremont et al.ICLR 2026
Builds on4
- Hyperbolic Neural Networks++Ryohei Shimizu, Yusuke Mukuta, Tatsuya HaradaICLR 2021 · 791 citations
- Constant Curvature Graph Convolutional NetworksGregor Bachmann, Gary Bécigneul, Octavian GaneaICML 2020 · 169 citations
- Mixed-curvature Variational AutoencodersOndrej Skopek, Octavian-Eugen Ganea, Gary BécigneulICLR 2020 · 122 citations
- Hyperbolic Image EmbeddingsValentin Khrulkov, Leyla Mirvakhabova, Evgeniya Ustinova, Ivan V. Oseledets et al.CVPR 2020
Related papers
- Coherent Hierarchical Multi-Label Classification NetworksEleonora Giunchiglia, Thomas LukasiewiczNeurIPS 2020 · 142 citations
- Rethinking the compositionality of point clouds through regularization in the hyperbolic spaceAntonio Montanaro, Diego Valsesia, Enrico MagliNeurIPS 2022 · 46 citations
- Neural Collapse with Normalized Features: A Geometric Analysis over the Riemannian ManifoldCan Yaras, Peng Wang, Zhihui Zhu, Laura Balzano et al.NeurIPS 2022 · 60 citations
- Prototype Based Classification from Hierarchy to FairnessMycal Tucker, Julie A. ShahICML 2022 · 7 citations
- Towards Robustness of Deep Neural Networks via RegularizationYao Li, Martin Renqiang Min, Thomas C. M. Lee, Wenchao Yu et al.ICCV 2021 · 8 citations
