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

Virtual Node Tuning for Few-shot Node Classification

About

Few-shot Node Classification (FSNC) is a challenge in graph representation learning where only a few labeled nodes per class are available for training. To tackle this issue, meta-learning has been proposed to transfer structural knowledge from base classes with abundant labels to target novel classes. However, existing solutions become ineffective or inapplicable when base classes have no or limited labeled nodes. To address this challenge, we propose an innovative method dubbed Virtual Node Tuning (VNT). Our approach utilizes a pretrained graph transformer as the encoder and injects virtual nodes as soft prompts in the embedding space, which can be optimized with few-shot labels in novel classes to modulate node embeddings for each specific FSNC task. A unique feature of VNT is that, by incorporating a Graph-based Pseudo Prompt Evolution (GPPE) module, VNT-GPPE can handle scenarios with sparse labels in base classes. Experimental results on four datasets demonstrate the superiority of the proposed approach in addressing FSNC with unlabeled or sparsely labeled base classes, outperforming existing state-of-the-art methods and even fully supervised baselines.

Zhen Tan, Ruocheng Guo, Kaize Ding, Huan Liu• 2023

Related benchmarks

TaskDatasetResultRank
Node ClassificationCora
Accuracy79.39
583
Node ClassificationCiteseer
Accuracy73.9
503
Node ClassificationCora-ML
Accuracy89.26
326
few-shot node classificationCoraFull
Accuracy70.16
68
few-shot node classificationCoauther CS
Accuracy89.95
68
Node ClassificationCoraFull 5-way 3-shot (test)
Accuracy55.19
36
Node ClassificationCora 2 way 3 shot
Accuracy (%)75.32
20
Node ClassificationCoauthor-CS 5 way 3 shot
Accuracy80.16
20
Node ClassificationCoauthor-CS 5 way 5 shot
Accuracy82.92
20
Node ClassificationCoraFull 5 way 5 shot
Accuracy70.16
20
Showing 10 of 16 rows

Other info

Follow for update