ProG: A Graph Prompt Learning Benchmark

Chenyi Zi, Haihong Zhao, Xiangguo Sun, Yiqing Lin, Hong Cheng +1 more
6/8/2024
cs.LG

Official code: sheldonresearch/ProG (589 stars, MIT, last commit 2025-05-15).

Abstract

Artificial general intelligence on graphs has shown significant advancements across various applications, yet the traditional 'Pre-train & Fine-tune' paradigm faces inefficiencies and negative transfer issues, particularly in complex and few-shot settings. Graph prompt learning emerges as a promising alternative, leveraging lightweight prompts to manipulate data and fill the task gap by reformulating downstream tasks to the pretext. However, several critical challenges still remain: how to unify diverse graph prompt models, how to evaluate the quality of graph prompts, and to improve their usability for practical comparisons and selection. In response to these challenges, we introduce the first comprehensive benchmark for graph prompt learning. Our benchmark integrates SIX pre-training methods and FIVE state-of-the-art graph prompt techniques, evaluated across FIFTEEN diverse datasets to assess performance, flexibility, and efficiency. We also present 'ProG', an easy-to-use open-source library that streamlines the execution of various graph prompt models, facilitating objective evaluations. Additionally, we propose a unified framework that categorizes existing graph prompt methods into two main approaches: prompts as graphs and prompts as tokens. This framework enhances the applicability and comparison of graph prompt techniques. The code is available at: https://github.com/sheldonresearch/ProG.

View on arXivView PDF

Code Implementations(1)

MIT

A Unified Python Library for Graph Prompting

58974Python, Jupyter NotebookJul 13, 20231 years agoMIT
graph-neural-networksgraph-promptprompt-learning

Similar Papers

Cite this paper

@article{zi2024prog,
  title  = {ProG: A Graph Prompt Learning Benchmark},
  author = {Chenyi Zi and Haihong Zhao and Xiangguo Sun and Yiqing Lin and Hong Cheng and Jia Li},
  year   = {2024},
  eprint = {2406.05346},
  archivePrefix = {arXiv},
  doi    = {10.48550/arXiv.2406.05346},
  url    = {http://arxiv.org/abs/2406.05346v2},
  journal = {NEURIPS 2024 2024}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2406.05346.svg)](https://codeofpaper.com/paper/2406.05346)

Discussion