Pseudo-Riemannian Graph Convolutional Networks
File version
Version of Record (VoR)
Author(s)
Zhu, S
Potyka, N
Pan, S
Zhou, C
Staab, S
Griffith University Author(s)
Primary Supervisor
Other Supervisors
Editor(s)
Date
Size
File type(s)
Location
New Orleans, USA
License
Abstract
Graph convolutional networks (GCNs) are powerful frameworks for learning embeddings of graph-structured data. GCNs are traditionally studied through the lens of Euclidean geometry. Recent works find that non-Euclidean Riemannian manifolds provide specific inductive biases for embedding hierarchical or spherical data. However, they cannot align well with data of mixed graph topologies. We consider a larger class of pseudo-Riemannian manifolds that generalize hyperboloid and sphere. We develop new geodesic tools that allow for extending neural network operations into geodesically disconnected pseudo-Riemannian manifolds. As a consequence, we derive a pseudo-Riemannian GCN that models data in pseudo-Riemannian manifolds of constant nonzero curvature in the context of graph neural networks. Our method provides a geometric inductive bias that is sufficiently flexible to model mixed heterogeneous topologies like hierarchical graphs with cycles. We demonstrate the representational capabilities of this method by applying it to the tasks of graph reconstruction, node classification and link prediction on a series of standard graphs with mixed topologies. Empirical results demonstrate that our method outperforms Riemannian counterparts when embedding graphs of complex topologies.
Journal Title
Conference Title
Advances in Neural Information Processing Systems
Book Title
Edition
Volume
35
Issue
Thesis Type
Degree Program
School
DOI
Patent number
Funder(s)
Grant identifier(s)
Rights Statement
Rights Statement
© The Author(s) 2022. The attached file is reproduced here in accordance with the copyright policy of the publisher. For information about this conference please refer to the conference’s website or contact the author(s).
Item Access Status
Note
Access the data
Related item(s)
Subject
Machine learning
Persistent link to this record
Citation
Xiong, B; Zhu, S; Potyka, N; Pan, S; Zhou, C; Staab, S, Pseudo-Riemannian Graph Convolutional Networks, Advances in Neural Information Processing Systems, 2022, 35