Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Deep Graph Reprogramming

About

In this paper, we explore a novel model reusing task tailored for graph neural networks (GNNs), termed as "deep graph reprogramming". We strive to reprogram a pre-trained GNN, without amending raw node features nor model parameters, to handle a bunch of cross-level downstream tasks in various domains. To this end, we propose an innovative Data Reprogramming paradigm alongside a Model Reprogramming paradigm. The former one aims to address the challenge of diversified graph feature dimensions for various tasks on the input side, while the latter alleviates the dilemma of fixed per-task-per-model behavior on the model side. For data reprogramming, we specifically devise an elaborated Meta-FeatPadding method to deal with heterogeneous input dimensions, and also develop a transductive Edge-Slimming as well as an inductive Meta-GraPadding approach for diverse homogenous samples. Meanwhile, for model reprogramming, we propose a novel task-adaptive Reprogrammable-Aggregator, to endow the frozen model with larger expressive capacities in handling cross-domain tasks. Experiments on fourteen datasets across node/graph classification/regression, 3D object recognition, and distributed action recognition, demonstrate that the proposed methods yield gratifying results, on par with those by re-training from scratch.

Yongcheng Jing, Chongbin Yuan, Li Ju, Yiding Yang, Xinchao Wang, Dacheng Tao• 2023

Related benchmarks

TaskDatasetResultRank
3D Object ClassificationModelNet40--
62
Regressionmolesol OGB
RMSE2.05
26
3D Object RecognitionShapeNet
Accuracy60.9
20
ClassificationProtein (test)
Accuracy60.71
11
Molecule RegressionQM7b (test)
MAE0.8889
5
Node Classificationogbn-arxiv (s2)
Accuracy60.34
5
Molecular Classificationogbg-molbbbp downstream
ROC-AUC0.6691
3
Node Classificationogbn-arxiv (s1)
Accuracy78.84
3
Distributed Action RecognitionWARD Downstream Action Categories
ReSt0.7337
1
Distributed Action RecognitionWARD Action Categories (Pre-trained)--
1
Showing 10 of 11 rows

Other info

Follow for update