Unlocking the Potential of Black-box Pre-trained GNNs for Graph Few-shot Learning
Few-shot learning has emerged as an important problem on graphs to combat label scarcity, which can be approached by current trends in pre-trained graph neural networks (GNNs) and meta-learning. Recent efforts integrate both paradigms in a white-box setting, leaving the more realistic black-box setting largely underexplored, where the parameters and gradients in the pre-trained GNNs are inaccessible. In this paper, we study the critical problem: Leveraging black-box pre-trained GNNs for graph few-shot learning. Despite its appeal, two key issues hinder the unlocking of its potential: the inherent task gap between pre-training and downstream stages, which can introduce irrelevant knowledge and undermine the generalizability of a pre-trained black-box GNN on downstream tasks; and the inaccessibility of parameters and gradients, which limits the model's adaptation to novel tasks. To effectively leverage the black-box pre-trained GNNs and improve generalization, we propose a lightweight graph meta-learner to extract relevant knowledge from a black-box pre-trained GNN, meanwhile harnessing knowledge from related tasks for rapid adaptation on novel tasks. Furthermore, we prune the graph meta-learner to enhance its generalization on novel tasks. Extensive experiments on real-world datasets for few-shot node classification validate the effectiveness of our proposed method in the black-box setting.
Authors
- Xiangliang Zhang (ORCID: https://orcid.org/0000-0002-3574-5665)
- Fang Yuan (ORCID: https://orcid.org/0000-0003-1987-0885)
- Shichao Pei (ORCID: https://orcid.org/0000-0002-0802-1506)
- Qiannan Zhang (ORCID: https://orcid.org/0000-0003-4846-7550)
Institutions
- University of Notre Dame (US)
- Cornell University (US)
- University of Massachusetts Boston (US)
- Singapore Management University (SG)
Publication Details
- Journal
- Proceedings of the AAAI Conference on Artificial Intelligence
- Published
- 2025-04-11
- DOI
- https://doi.org/10.1609/aaai.v39i21.34407
- Citations
- 1
- Primary Topic
- Domain Adaptation and Few-Shot Learning
- Type
- article
- Field-Weighted Citation Impact
- 0.89
Funders
- Ministry of Education - Singapore
- Ministry of Education, India