Hyperbolic Random Forests

Doorenbos, Lars; Márquez Neila, Pablo; Sznitman, Raphael; Mettes, Pascal (2024). Hyperbolic Random Forests. Transactions on machine learning research, 2024(05) OpenReview.net

[img]
Preview
Text
2195_hyperbolic_random_forests.pdf - Published Version
Available under License Publisher holds Copyright.

Download (2MB) | Preview

Hyperbolic space is becoming a popular choice for representing data due to the hierarchical structure - whether implicit or explicit - of many real-world datasets. Along with it comes a need for algorithms capable of solving fundamental tasks, such as classification, in hyperbolic space. Recently, multiple papers have investigated hyperbolic alternatives to hyperplane-based classifiers, such as logistic regression and SVMs. While effective, these approaches struggle with more complex hierarchical data. We, therefore, propose to generalize the well-known random forests to hyperbolic space. We do this by redefining the notion of a split using horospheres. Since finding the globally optimal split is computationally intractable, we find candidate horospheres through a large-margin classifier. To make hyperbolic random forests work on multi-class data and imbalanced experiments, we furthermore outline a new method for combining classes based on their lowest common ancestor and a class-balanced version of the large-margin loss. Experiments on standard and new benchmarks show that our approach outperforms both conventional random forest algorithms and recent hyperbolic classifiers.

Item Type:

Journal Article (Original Article)

Division/Institute:

10 Strategic Research Centers > ARTORG Center for Biomedical Engineering Research > ARTORG Center - AI in Medical Imaging Laboratory

UniBE Contributor:

Doorenbos, Lars Jelte, Márquez Neila, Pablo, Sznitman, Raphael

Subjects:

000 Computer science, knowledge & systems

ISSN:

2835-8856

Publisher:

OpenReview.net

Language:

English

Submitter:

Lars Jelte Doorenbos

Date Deposited:

23 May 2024 08:11

Last Modified:

23 May 2024 08:11

BORIS DOI:

10.48350/196977

URI:

https://boris.unibe.ch/id/eprint/196977

Actions (login required)

Edit item Edit item
Provide Feedback