Towards Pre-trained Graph Condensation via Optimal Transport
Yeyu Yan, Shuai Zheng, Wenjun Hui, Xiangkai Zhu, Dong Chen, Zhenfeng Zhu, Yao Zhao, Kunlun He
摘要
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.
问问这篇 Paper
智能体会读完全文。
Lune 把这篇 Paper 索引到了每一个公式,引用它的顶会 Paper 也一样。你提问,回答直接引用原文。
引用它的顶会 Paper2
- Subspace-Aware Feature Reshaping for Open-Set Graph Class-Incremental LearningWeichao Zhang, Shuai Zheng, Yeyu Yan, Zhizhe Liu 等ICML 2026
- Stage-Aware Graph Contrastive Learning with Node-oriented Mixture of ExpertsXiangkai Zhu, Yeyu Yan, Saiqin Long, Chao Li 等AAAI 2026
它引用的顶会 Paper33
- Dataset Condensation with Gradient MatchingBo Zhao, Konda Reddy Mopuri, Hakan BilenICLR 2021 · 被引用 684 次
- BernNet: Learning Arbitrary Graph Spectral Filters via Bernstein ApproximationMingguo He, Zhewei Wei, Zengfeng Huang, Hongteng XuNeurIPS 2021 · 被引用 378 次
- Simple Spectral Graph ConvolutionHao Zhu, Piotr KoniuszICLR 2021 · 被引用 352 次
- A Unified Lottery Ticket Hypothesis for Graph Neural NetworksTianlong Chen, Yongduo Sui, Xuxi Chen, Aston Zhang 等ICML 2021 · 被引用 208 次
- Graph Condensation for Graph Neural NetworksWei Jin, Lingxiao Zhao, Shichang Zhang, Yozen Liu 等ICLR 2022 · 被引用 203 次
相关 Paper
- Graph Condensation for Open-World Graph LearningXinyi Gao, Tong Chen, Wentao Zhang, Yayong Li 等KDD 2024 · 被引用 13 次
- ST-GCond: Self-supervised and Transferable Graph Dataset CondensationBeining Yang, Qingyun Sun, Cheng Ji, Xingcheng Fu 等ICLR 2025
- Rethinking and Accelerating Graph Condensation: A Training-Free Approach with Class PartitionXinyi Gao, Guanhua Ye, Tong Chen, Wentao Zhang 等WWW 2025 · 被引用 27 次
- Disentangled Condensation for Large-scale GraphsZhenbang Xiao, Yu Wang, Shunyu Liu, Bingde Hu 等WWW 2025 · 被引用 14 次
- Anchor-guided Hypergraph Condensation with Dual-level DiscriminationFan Li, Xiaoyang Wang, Chen Chen, Wenjie ZhangICML 2026
