Does Graph Distillation See Like Vision Dataset Counterpart?

Beining Yang, Kai Wang, Qingyun Sun, Cheng Ji, Xingcheng Fu +3 more
10/1/2023

Official code: RingBDStack/SGDD (26 stars, Apache-2.0, last commit 2023-10-19).

Abstract

Training on large-scale graphs has achieved remarkable results in graph representation learning, but its cost and storage have attracted increasing concerns. Existing graph condensation methods primarily focus on optimizing the feature matrices of condensed graphs while overlooking the impact of the structure information from the original graphs. To investigate the impact of the structure information, we conduct analysis from the spectral domain and empirically identify substantial Laplacian Energy Distribution (LED) shifts in previous works. Such shifts lead to poor performance in cross-architecture generalization and specific tasks, including anomaly detection and link prediction. In this paper, we propose a novel Structure-broadcasting Graph Dataset Distillation (SGDD) scheme for broadcasting the original structure information to the generation of the synthetic one, which explicitly prevents overlooking the original structure information. Theoretically, the synthetic graphs by SGDD are expected to have smaller LED shifts than previous works, leading to superior performance in both cross-architecture settings and specific tasks. We validate the proposed SGDD across 9 datasets and achieve state-of-the-art results on all of them: for example, on the YelpChi dataset, our approach maintains 98.6% test accuracy of training on the original graph dataset with 1,000 times saving on the scale of the graph. Moreover, we empirically evaluate there exist 17.6% ~ 31.4% reductions in LED shift crossing 9 datasets. Extensive experiments and analysis verify the effectiveness and necessity of the proposed designs. The code is available in the GitHub repository: https://github.com/RingBDStack/SGDD.

View on arXivView PDF

Code Implementations(1)

RingBDStack/SGDDOfficial100%
Apache-2.0

Code for SGDD

262PythonMay 24, 20233 years agoApache-2.0

Similar Papers

Cite this paper

@article{yang2023does,
  title  = {Does Graph Distillation See Like Vision Dataset Counterpart?},
  author = {Beining Yang and Kai Wang and Qingyun Sun and Cheng Ji and Xingcheng Fu and Hao Tang and Yang You and Jianxin Li},
  year   = {2023},
  eprint = {2310.09192},
  archivePrefix = {arXiv},
  url    = {https://arxiv.org/pdf/2310.09192},
  journal = {NEURIPS 2023 2023}
}

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/2310.09192.svg)](https://codeofpaper.com/paper/2310.09192)

Discussion