October 13, 2023
Conference Paper

A Unification Framework for Euclidean and Hyperbolic Graph Neural Networks

Abstract

Hyperbolic neural networks have recently gained significant attention due to their promising results on several graph problems including node classification and link prediction. The primary reason for this success is the effectiveness of hyperbolic space in capturing the inherent hierarchy of graph datasets. However, they are limited in terms of generalization, scalability, and have inferior performance when applied to non-hierarchical datasets. In this paper, we take a completely different perspective for modeling hyperbolic networks and answer the following question: is an Euclidean model able to approximate a function or behavior in the hyperbolic space? Extending the universal approximation theory developed for Euclidean models, We draw an analogy from the hyperbolic components to the Euclidean counterparts and conclude that, in order to capture hierarchical features, it is possible to generalize hyperbolic models to be a special case of Euclidean models with the proposed Pseudo-Poincaré technique. We applied our non-linear hyperbolic normalization to the current state-of-the-art homogeneous and multi-relational graph networks and demonstrate significant improvements in performance compared to both Euclidean and hyperbolic counterparts. The primary impact of this work lies in its ability to capture hierarchical features in the Euclidean space, and thus, can replace hyperbolic networks without any loss in performance metrics while simultaneously leveraging the power of Euclidean networks such as interpretability and efficient execution of various model components.

Published: October 13, 2023

Citation

Khatir M., N. Choudhary, S. Choudhury, K. Agarwal, and C. Reddy. 2023. A Unification Framework for Euclidean and Hyperbolic Graph Neural Networks. In Proceedings of the Thirty-Second International Joint Conference on Artificial Intelligence, August 19-25, 2023, Macao, edited by E. Elkind, 2023, 3875-3883. Red Hook, New York:International Joint Conferences on Artifical Intelligence (IJCAI); Curran Associates, Inc. PNNL-SA-173369. doi:10.24963/ijcai.2023/431