Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Graph Few-shot Learning with Task-specific Structures

About

Graph few-shot learning is of great importance among various graph learning tasks. Under the few-shot scenario, models are often required to conduct classification given limited labeled samples. Existing graph few-shot learning methods typically leverage Graph Neural Networks (GNNs) and perform classification across a series of meta-tasks. Nevertheless, these methods generally rely on the original graph (i.e., the graph that the meta-task is sampled from) to learn node representations. Consequently, the graph structure used in each meta-task is identical. Since the class sets are different across meta-tasks, node representations should be learned in a task-specific manner to promote classification performance. Therefore, to adaptively learn node representations across meta-tasks, we propose a novel framework that learns a task-specific structure for each meta-task. To handle the variety of nodes across meta-tasks, we extract relevant nodes and learn task-specific structures based on node influence and mutual information. In this way, we can learn node representations with the task-specific structure tailored for each meta-task. We further conduct extensive experiments on five node classification datasets under both single- and multiple-graph settings to validate the superiority of our framework over the state-of-the-art baselines. Our code is provided at https://github.com/SongW-SW/GLITTER.

Song Wang, Chen Chen, Jundong Li• 2022

Related benchmarks

TaskDatasetResultRank
Node ClassificationCora Full
Accuracy66.2
88
Node ClassificationDBLP
Accuracy79.3
67
few-shot node classificationTissue-PPI multiple-graph, Shared Graph, Shared Label (test)
Accuracy69.7
5
few-shot node classificationTissue-PPI multiple-graph Disjoint Graph Shared Label (test)
Accuracy73.1
5
few-shot node classificationTissue-PPI multiple-graph Disjoint Graph Disjoint Label (test)
Accuracy60.3
5
few-shot node classificationFold-PPI multiple-graph, Shared Graph, Shared Label (test)
Accuracy53.3
5
few-shot node classificationFold-PPI multiple-graph, Disjoint Graph, Shared Label (test)
Accuracy61.3
5
few-shot node classificationFold-PPI multiple-graph, Disjoint Graph, Disjoint Label (test)
Accuracy54.7
5
Showing 8 of 8 rows

Other info

Code

Follow for update