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

Hyperbolic Graph Neural Networks at Scale: A Meta Learning Approach

About

The progress in hyperbolic neural networks (HNNs) research is hindered by their absence of inductive bias mechanisms, which are essential for generalizing to new tasks and facilitating scalable learning over large datasets. In this paper, we aim to alleviate these issues by learning generalizable inductive biases from the nodes' local subgraph and transfer them for faster learning over new subgraphs with a disjoint set of nodes, edges, and labels in a few-shot setting. We introduce a novel method, Hyperbolic GRAph Meta Learner (H-GRAM), that, for the tasks of node classification and link prediction, learns transferable information from a set of support local subgraphs in the form of hyperbolic meta gradients and label hyperbolic protonets to enable faster learning over a query set of new tasks dealing with disjoint subgraphs. Furthermore, we show that an extension of our meta-learning framework also mitigates the scalability challenges seen in HNNs faced by existing approaches. Our comparative analysis shows that H-GRAM effectively learns and transfers information in multiple challenging few-shot settings compared to other state-of-the-art baselines. Additionally, we demonstrate that, unlike standard HNNs, our approach is able to scale over large graph datasets and improve performance over its Euclidean counterparts.

Nurendra Choudhary, Nikhil Rao, Chandan K. Reddy• 2023

Related benchmarks

TaskDatasetResultRank
Node ClassificationCora
Accuracy82.7
885
Node ClassificationCiteseer
Accuracy92.4
804
Node ClassificationPubmed
Accuracy71.6
742
Node ClassificationSynth. Cycle (five-fold)
Accuracy88.3
15
Node ClassificationBA synthetic (five-fold)
Accuracy88.8
15
Node ClassificationSyn. Cycle SG, DL
Accuracy88.3
7
Node ClassificationSyn. BA SG, DL
Accuracy87.3
7
Node Classificationogbn-arxiv SG, DL
Accuracy47.2
7
Node Classificationogbn-arxiv official (five-fold)
Accuracy47.2
7
Link PredictionFirstMM-DB MG, SL
Accuracy74.2
5
Showing 10 of 24 rows

Other info

Code

Follow for update