Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Hyperbolic Neural Networks++

About

Hyperbolic spaces, which have the capacity to embed tree structures without distortion owing to their exponential volume growth, have recently been applied to machine learning to better capture the hierarchical nature of data. In this study, we generalize the fundamental components of neural networks in a single hyperbolic geometry model, namely, the Poincar\'e ball model. This novel methodology constructs a multinomial logistic regression, fully-connected layers, convolutional layers, and attention mechanisms under a unified mathematical interpretation, without increasing the parameters. Experiments show the superior parameter efficiency of our methods compared to conventional hyperbolic components, and stability and outperformance over their Euclidean counterparts.

Ryohei Shimizu, Yusuke Mukuta, Tatsuya Harada• 2020

Related benchmarks

TaskDatasetResultRank
Node ClassificationCora
Accuracy54.6
885
Node ClassificationPubmed
Accuracy69.8
742
Link PredictionPubmed
AUC94.9
123
Link PredictionCora
AUC0.89
116
Tumor classificationCamelyon17-WILDS (val)
AUC97.95
26
Link PredictionAIRPORT
ROC AUC90.8
26
Tumor classificationCamelyon17-WILDS (test)
AUC0.9807
26
Node ClassificationAIRPORT
Accuracy80.5
20
Node ClassificationAIRPORT δ = 1 (test)
F1 Score (Test)80.5
18
Node ClassificationDISEASE δ = 0 (test)
F1 Score41
18
Showing 10 of 26 rows

Other info

Follow for update