English

Horospherical Decision Boundaries for Large Margin Classification in Hyperbolic Space

Machine Learning 2023-09-29 v3 Machine Learning

Abstract

Hyperbolic spaces have been quite popular in the recent past for representing hierarchically organized data. Further, several classification algorithms for data in these spaces have been proposed in the literature. These algorithms mainly use either hyperplanes or geodesics for decision boundaries in a large margin classifiers setting leading to a non-convex optimization problem. In this paper, we propose a novel large margin classifier based on horospherical decision boundaries that leads to a geodesically convex optimization problem that can be optimized using any Riemannian gradient descent technique guaranteeing a globally optimal solution. We present several experiments depicting the competitive performance of our classifier in comparison to SOTA.

Keywords

Cite

@article{arxiv.2302.06807,
  title  = {Horospherical Decision Boundaries for Large Margin Classification in Hyperbolic Space},
  author = {Xiran Fan and Chun-Hao Yang and Baba C. Vemuri},
  journal= {arXiv preprint arXiv:2302.06807},
  year   = {2023}
}

Comments

To appear at Neural Information Processing Systems (NeurIPS) 2023

R2 v1 2026-06-28T08:39:28.343Z