Hyperbolic Neural Networks++
Authors: Ryohei Shimizu, YUSUKE Mukuta, Tatsuya Harada
Venue: ICLR 2021 Poster
Keywords: Hyperbolic Geometry, Poincaré Ball Model, Parameter-Reduced MLR, Geodesic-Aware FC Layer, Convolutional Layer, Attention Mechanism
Abstract: 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é 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.