Learning Hierarchical Task Structures for Few-shot Graph Classification

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

Research output: Journal article publicationJournal articleAcademic researchpeer-review

Abstract

The problem of few-shot graph classification targets at assigning class labels for graph samples, where only limited labeled graphs are provided for each class. To solve the problem brought by label scarcity, recent studies have proposed to adopt the prevalent few-shot learning framework to achieve fast adaptations to graph classes with limited labeled graphs. In particular, these studies typically propose to accumulate meta-knowledge across a large number of meta-training tasks, and then generalize such meta-knowledge to meta-test tasks sampled from a disjoint class set. Nevertheless, existing studies generally ignore the crucial task correlations among meta-training tasks and treat them independently. In fact, such task correlations can help promote the model generalization to meta-test tasks and result in better classification performance. On the other hand, it remains challenging to capture and utilize task correlations due to the complex components and interactions in meta-training tasks. To deal with this, we propose a novel few-shot graph classification framework FAITH to capture task correlations via learning a hierarchical task structure at different granularities. We further propose a task-specific classifier to incorporate the learned task correlations into the few-shot graph classification process. Moreover, we derive FAITH+, a variant of FAITH that can improve the sampling process for the hierarchical task structure. The extensive experiments on four prevalent graph datasets further demonstrate the superiority of FAITH and FAITH+ over other state-of-the-art baselines.

Original languageEnglish
Article number67
JournalACM Transactions on Knowledge Discovery from Data
Volume18
Issue number3
DOIs
Publication statusPublished - 12 Jan 2024

Keywords

  • Additional Key Words and PhrasesFew-shot learning
  • data mining
  • graph classification
  • graph neural networks

ASJC Scopus subject areas

  • General Computer Science

Fingerprint

Dive into the research topics of 'Learning Hierarchical Task Structures for Few-shot Graph Classification'. Together they form a unique fingerprint.

Cite this