Graph Prototypical Networks for Few-shot Learning on Attributed Networks
About
Attributed networks nowadays are ubiquitous in a myriad of high-impact applications, such as social network analysis, financial fraud detection, and drug discovery. As a central analytical task on attributed networks, node classification has received much attention in the research community. In real-world attributed networks, a large portion of node classes only contain limited labeled instances, rendering a long-tail node class distribution. Existing node classification algorithms are unequipped to handle the \textit{few-shot} node classes. As a remedy, few-shot learning has attracted a surge of attention in the research community. Yet, few-shot node classification remains a challenging problem as we need to address the following questions: (i) How to extract meta-knowledge from an attributed network for few-shot node classification? (ii) How to identify the informativeness of each labeled instance for building a robust and effective model? To answer these questions, in this paper, we propose a graph meta-learning framework -- Graph Prototypical Networks (GPN). By constructing a pool of semi-supervised node classification tasks to mimic the real test environment, GPN is able to perform \textit{meta-learning} on an attributed network and derive a highly generalizable model for handling the target classification task. Extensive experiments demonstrate the superior capability of GPN in few-shot node classification.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Node Classification | Citeseer | Accuracy19.9 | 931 | |
| Node Classification | Wisconsin | Accuracy40.7 | 627 | |
| Node Classification | Cora Full | Accuracy62.1 | 88 | |
| Node Classification | DBLP | Accuracy75.4 | 67 | |
| Node Classification | Coauthor PHY | Accuracy35.2 | 39 | |
| Node Classification | ogbn-arXiv (train val test) | Accuracy55.35 | 18 | |
| Node Classification | CoraFull (train val test) | Accuracy65.23 | 18 | |
| Node Classification | DBLP (train val test) | Accuracy76.05 | 18 | |
| Node Classification | DBLP | Accuracy51.7 | 12 | |
| Node Classification | Wiki | Accuracy15.8 | 12 |