FAITH: Few-Shot Graph Classification with Hierarchical Task Graphs

FAITH: Few-Shot Graph Classification with Hierarchical Task Graphs

Song Wang, Yushun Dong, Xiao Huang, Chen Chen, Jundong Li

Proceedings of the Thirty-First International Joint Conference on Artificial Intelligence
Main Track. Pages 2284-2290. https://doi.org/10.24963/ijcai.2022/317

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.
Keywords:
Data Mining: Mining Graphs
Machine Learning: Classification
Machine Learning: Meta-Learning
Machine Learning: Representation learning