Saved in:
Bibliographic Details
Main Authors: Yan, Yeyu, Zheng, Shuai, Hui, Wenjun, Zhu, Xiangkai, Chen, Dong, Zhu, Zhenfeng, Zhao, Yao, He, Kunlun
Format: Preprint
Published: 2025
Subjects:
Online Access:https://arxiv.org/abs/2509.14722
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866912591570272256
author Yan, Yeyu
Zheng, Shuai
Hui, Wenjun
Zhu, Xiangkai
Chen, Dong
Zhu, Zhenfeng
Zhao, Yao
He, Kunlun
author_facet Yan, Yeyu
Zheng, Shuai
Hui, Wenjun
Zhu, Xiangkai
Chen, Dong
Zhu, Zhenfeng
Zhao, Yao
He, Kunlun
contents Graph condensation (GC) aims to distill the original graph into a small-scale graph, mitigating redundancy and accelerating GNN training. However, conventional GC approaches heavily rely on rigid GNNs and task-specific supervision. Such a dependency severely restricts their reusability and generalization across various tasks and architectures. In this work, we revisit the goal of ideal GC from the perspective of GNN optimization consistency, and then a generalized GC optimization objective is derived, by which those traditional GC methods can be viewed nicely as special cases of this optimization paradigm. Based on this, Pre-trained Graph Condensation (PreGC) via optimal transport is proposed to transcend the limitations of task- and architecture-dependent GC methods. Specifically, a hybrid-interval graph diffusion augmentation is presented to suppress the weak generalization ability of the condensed graph on particular architectures by enhancing the uncertainty of node states. Meanwhile, the matching between optimal graph transport plan and representation transport plan is tactfully established to maintain semantic consistencies across source graph and condensed graph spaces, thereby freeing graph condensation from task dependencies. To further facilitate the adaptation of condensed graphs to various downstream tasks, a traceable semantic harmonizer from source nodes to condensed nodes is proposed to bridge semantic associations through the optimized representation transport plan in pre-training. Extensive experiments verify the superiority and versatility of PreGC, demonstrating its task-independent nature and seamless compatibility with arbitrary GNNs.
format Preprint
id arxiv_https___arxiv_org_abs_2509_14722
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Towards Pre-trained Graph Condensation via Optimal Transport
Yan, Yeyu
Zheng, Shuai
Hui, Wenjun
Zhu, Xiangkai
Chen, Dong
Zhu, Zhenfeng
Zhao, Yao
He, Kunlun
Machine Learning
Graph condensation (GC) aims to distill the original graph into a small-scale graph, mitigating redundancy and accelerating GNN training. However, conventional GC approaches heavily rely on rigid GNNs and task-specific supervision. Such a dependency severely restricts their reusability and generalization across various tasks and architectures. In this work, we revisit the goal of ideal GC from the perspective of GNN optimization consistency, and then a generalized GC optimization objective is derived, by which those traditional GC methods can be viewed nicely as special cases of this optimization paradigm. Based on this, Pre-trained Graph Condensation (PreGC) via optimal transport is proposed to transcend the limitations of task- and architecture-dependent GC methods. Specifically, a hybrid-interval graph diffusion augmentation is presented to suppress the weak generalization ability of the condensed graph on particular architectures by enhancing the uncertainty of node states. Meanwhile, the matching between optimal graph transport plan and representation transport plan is tactfully established to maintain semantic consistencies across source graph and condensed graph spaces, thereby freeing graph condensation from task dependencies. To further facilitate the adaptation of condensed graphs to various downstream tasks, a traceable semantic harmonizer from source nodes to condensed nodes is proposed to bridge semantic associations through the optimized representation transport plan in pre-training. Extensive experiments verify the superiority and versatility of PreGC, demonstrating its task-independent nature and seamless compatibility with arbitrary GNNs.
title Towards Pre-trained Graph Condensation via Optimal Transport
topic Machine Learning
url https://arxiv.org/abs/2509.14722