arrow
Return

Graph Few-Shot Learning via Restructuring Task Graph

delete2024-01-01
delete5
PRE
AI
F
Feng Zhao
T
Tiancheng Huang
王东麟 cover
王东麟 (Donglin Wang) *
DOI:10.1109/TNNLS.2022.3178849delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
Existing graph few-shot learning (FSL) methods usually train a model on many task graphs and transfer the learned model to a new task graph. However, the task graphs often contain a great number of isolated nodes, which results in the severe deficiency of learned node embeddings. Furthermore, in the training process, the neglect of task information also constrains the model's expressive ability. In this brief, we propose a novel metric-based graph few-shot learning approach via restructuring task graph (GFL-RTG). To solve the problems above, we innovatively restructure the task graph by adding class nodes and a task node to the original individual task graph. We first add class nodes and determine the connectivity between class nodes and others via their similarity. Then, we utilize a graph pooling network to learn a task embedding, which is regarded as a task node. Finally, the new task graph is restructured by combining class nodes, task node, and original nodes, which is then used as input to the metric-based graph neural network (GNN) to conduct few-shot learning. Our extensive experiments on three graph-structured datasets demonstrate that our proposed method generally outperforms the state-of-the-art baselines in few-shot learning.
Keywords:
Task analysis
Training
Learning systems
Testing
Prototypes
Message passing
Graph neural networks
Aggregation
class node
graph few-shot learning (FSL)
graph neural networks (GNNs)
node classification
task graph
task node

Journal

IEEE Transactions on Neural Networks and Learning Systems cover
IEEE Transactions on Neural Networks and Learning Systems
IF:
8.9
Papers:
7.5K
Citations:
7.2W

Organization

W
westlake university
Scholars:
5.3K
Papers: 3.7K
Citations: 8
Z
zhejiang university
Scholars:
17.5W
Papers: 12.0W
Citations: 152