G-Prompt: Graphon-based Prompt Tuning for graph classification
Yutai Duan,Jie Liu,Shaowei Chen,Liyi Chen,Jianhua Wu
DOI: https://doi.org/10.1016/j.ipm.2023.103639
IF: 7.466
2024-01-09
Information Processing & Management
Abstract:Prompt tuning has been demonstrated to be effective in exploiting pre-trained models (PTMs) to perform downstream tasks. Motivated by this trend, researchers explored prompt tuning methods to leverage graph PTMs in downstream tasks related to graph classification. However, the inherent non-Euclidean and abstract characteristics of graph data present a set of challenging issues, which encompass the generation of graph-level prompts with accurate task-related knowledge, the enhancement of prompt adaptability to downstream tasks, and the prediction based on graph-level prompts. To address these issues, we propose Graphon-based Prompt Tuning (G-Prompt) as a systemic solution. G-Prompt consists of two elaborate modules. Specifically, we present a graph-level prompt generation (GP) module that describes the knowledge of downstream tasks by estimating graphons and generates graph-level prompts based on the knowledge. Subsequently, GP module performs prompt ensembling steps and optimization based on downstream tasks to ensure the knowledge distribution of the prompts and their adaptability to downstream tasks. Furthermore, we propose a graph answer (GA) module to make accurate predictions based on multiple graph-level prompts. GA module models the conditional probability that the input graph belongs to the class corresponding to the prompt based on different prompts, respectively. It then selects the class with the highest probability as the prediction. Extensive experiments with 6 real-world datasets show that G-Prompt achieves state-of-the-art performance, outperforming compared methods by an average of 5%.
computer science, information systems,information science & library science