FAITH: Few-Shot Graph Classification with Hierarchical Task Graphs

TL;DR

FAITH introduces a hierarchical task graph approach for few-shot graph classification, effectively capturing task correlations to improve classification accuracy with limited labeled data.

Contribution

The paper proposes a novel hierarchical task graph framework and a loss-based sampling strategy to leverage task correlations in few-shot graph classification.

Findings

Outperforms state-of-the-art methods on four datasets.

Effectively captures task correlations for better generalization.

Demonstrates robustness with limited labeled graphs.

Abstract

Few-shot graph classification aims at predicting classes for graphs, given limited labeled graphs for each class. To tackle the bottleneck of label scarcity, recent works propose to incorporate few-shot learning frameworks for fast adaptations to graph classes with limited labeled graphs. Specifically, these works propose to accumulate meta-knowledge across diverse meta-training tasks, and then generalize such meta-knowledge to the target task with a disjoint label set. However, existing methods generally ignore task correlations among meta-training tasks while treating them independently. Nevertheless, such task correlations can advance the model generalization to the target task for better classification performance. On the other hand, it remains non-trivial to utilize task correlations due to the complex components in a large number of meta-training tasks. To deal with this, we propose a novel few-shot learning framework FAITH that captures task correlations via constructing a hierarchical task graph at different granularities. Then we further design a loss-based sampling strategy to select tasks with more correlated classes. Moreover, a task-specific classifier is proposed to utilize the learned task correlations for few-shot classification. Extensive experiments on four prevalent few-shot graph classification datasets demonstrate the superiority of FAITH over other state-of-the-art baselines.