Does Graph Distillation See Like Vision Dataset Counterpart?

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.

Paper

References (100)

Scroll for more · 38 remaining

Similar papers

Peer review

Reviewer FmUF4/10 · confidence 3/52023-07-02

Summary

This paper mainly focuses on the Laplacian Energy Distribution (LED) shift problem of graph dataset condensation.

Strengths

The pipeline figure is clear and straightforward. The studied problem, how to condense a graph dataset, is relatively important.

Weaknesses

Compared to existing studies that widely investigated the gradient matching, it seems that the major contribution of this paper is considering the LED shift. Why LED and how can LED benefit? The authors need to provide more analyses and explanation about their motivation. In the Introduction, the authors mention “We empirically find a positive correlation between LED shift and the performance in cross-architecture settings”. This can only be considered as a finding. The motivation for introducing LED is ambiguous, especially given the correlation does not mean causality. The authors simply compare their method with Gcond and consider it as the SOTA. However, there are many works that have been proposed since Gcond. The authors should consider compare with them as well for a comprehensive evaluation of their method. For example, [1, 2, 3, 4, 5] [1] Jin, Wei, et al. "Condensing graphs via one-step gradient matching." Proceedings of the 28th ACM SIGKDD Conference on Knowledge Discovery and Data Mining. 2022. [2] Yu, Ruonan, Songhua Liu, and Xinchao Wang. "Dataset distillation: A comprehensive review." arXiv preprint arXiv:2301.07014 (2023). [3] Liu, Chuang, et al. "Comprehensive graph gradual pruning for sparse training in graph neural networks." IEEE Transactions on Neural Networks and Learning Systems (2023). [4] Zheng, Xin, et al. "Structure-free Graph Condensation: From Large-scale Graphs to Condensed Graph-free Data." arXiv preprint arXiv:2306.02664 (2023). [5] Yang, Shuo, et al. "Dataset pruning: Reducing training data by examining generalization influence." arXiv preprint arXiv:2205.09329 (2022).

Questions

see above.

Rating

4: Borderline reject: Technically solid paper where reasons to reject, e.g., limited evaluation, outweigh reasons to accept, e.g., good evaluation. Please use sparingly.

Confidence

3: You are fairly confident in your assessment. It is possible that you did not understand some parts of the submission or that you are unfamiliar with some pieces of related work. Math/other details were not carefully checked.

Soundness

2 fair

Presentation

3 good

Contribution

2 fair

Limitations

see above.

Reviewer UGAQ7/10 · confidence 4/52023-07-06

Summary

This paper proposes a novel method called SGDD for condensing large-scale graph datasets while preserving the original structure information. The proposed method uses a graphon approximation method to broadcast the original structure as supervision for generating the condensed graph structure and optimizes it using an optimal transport method. The proposed SGDD achieves state-of-the-art results on various datasets and tasks.

Strengths

(1) This paper uncovers the issue of Laplacian energy distribution shift during the condensation of graph datasets and shows that previous graph condensation methods that overlook the original structure information can lead to poor performance in cross-architecture generalization and specific tasks. This paper introduces a metric SC to quantify the Laplacian energy distribution issue. (2) A new condensing paradigm in graph condensation. This paper leverages the graphon theory to guarantee the structure information consistency between the original and generated graph, which firstly involves the generated learning fashion in graph condensation. Consequently, the generated structure may capture the native properties of the original structure in comparison to current methods, which benefits the downstream tasks. (3) Extensive experimental results demonstrate the effectiveness of the proposed method. This work includes extensive experiments on three node-level tasks as well as multiple ablation studies. All of the experiments demonstrate superior results. The most appealing result is that this method performs well in a cross-architecture setting, significantly improving the effectiveness of graph condensation. (4) Potential Impact. When the data and model go bigger and bigger, graph condensation may have more applications and research meanings. As the applications and challenges related to graph condensation remain unidentified, this paper may set the stage for future exploration of this problem and inspire the communities to further investigate and may benefit the areas like pruning, LLM with graphs, and NAS.

Weaknesses

(1) The motivation for using JS divergence in the calculation of SC is unclear. There are serval ways of calculating the distance of two distributions. (2) In Figure.2, the author uses the BWGNN without specifying the difference between BWGNN and other backbones. (3) There is another related method called DosCond[1] which is mentioned in the introduction but not involved in the baselines. The reason why not choosing this method for comparison should be specified. [1] Jin, Wei, et al. “Condensing Graphs via One-Step Gradient Matching.” Proceedings of the 28th ACM SIGKDD Conference on Knowledge Discovery and Data Mining, 2022.

Questions

(1) Please detailed the motivation in weakness.1 and weakness.2. (2) Consider adding the baseline mentioned in weakness (3) or specify the reason why do not compare with this method. (3) There is a high variance in the results of SGDD in Table 1. Would changing the backbone potentially improve the overall performance as shown in Table 1?

Rating

7: Accept: Technically solid paper, with high impact on at least one sub-area, or moderate-to-high impact on more than one areas, with good-to-excellent evaluation, resources, reproducibility, and no unaddressed ethical considerations.

Confidence

4: You are confident in your assessment, but not absolutely certain. It is unlikely, but not impossible, that you did not understand some parts of the submission or that you are unfamiliar with some pieces of related work.

Soundness

3 good

Presentation

3 good

Contribution

3 good

Limitations

(1) In the real world, graphs may contain multiple types of edges, resulting in heterogeneity. However, the current learning methods may fail to capture this heterogeneity, leading to information loss. (2) Lack of discussion on different backbones.

Reviewer 9hTZ4/10 · confidence 4/52023-07-07

Summary

The paper proposes a novel approach for graph dataset distillation called Structure-broadcasting Graph Dataset Distillation (SGDD). The authors explicitly consider the impact of the original structure information on graph condensation and demonstrate that their approach achieves state-of-the-art results on 9 datasets, showing superior performance in cross-architecture settings and specific tasks. Overall, the paper presents a significant contribution to the field of graph dataset distillation.

Strengths

- The paper proposes a novel approach for graph dataset distillation that explicitly considers the impact of the original structure information on graph condensation. This is a significant departure from existing methods that primarily focus on optimizing the feature matrices of condensed graphs while overlooking the structure information. - The authors introduce the concept of Substantial Laplacian Energy Distribution (LED) shifts and demonstrate that previous works suffer from such shifts, leading to poor performance in cross-architecture generalization and specific tasks. The authors propose SGDD as a solution to this problem, which is a novel and original contribution. - The authors provide a thorough analysis of the proposed approach, including theoretical analysis and empirical evaluation on 9 datasets. The results demonstrate that SGDD consistently outperforms existing state-of-the-art methods, indicating the high quality of the proposed approach.

Weaknesses

- [Actual Computation Savings] The authors distill the graph dataset. However, the authors only provide the distilled dataset's size ratio to original dataset size. It would be helpful to provide the actual computation savings in terms of time and memory usage. - [The Cost during Graph Distillation] The cost of graph distillation for condensed small graph dataset is recommended to be provided. Since the used benchmark graph datasets are less computationally expensive than datasets in computer vision and natural language processing, the cost of graph distillation period will reflect whether the dataset is necessary for practical use. - [Comparison with Other Methods] The paper provides a comparison of the proposed method with other methods in terms of performance and size ratio (e.g., Table 1). However, it would be beneficial to also compare the cost of graph distillation for the proposed method with other methods. This information would provide a more comprehensive understanding of the practicality of the proposed approach compared to other methods.

Questions

Questions: Please see weaknesses. I would like to update my evaluation after the discussion.

Rating

4: Borderline reject: Technically solid paper where reasons to reject, e.g., limited evaluation, outweigh reasons to accept, e.g., good evaluation. Please use sparingly.

Confidence

4: You are confident in your assessment, but not absolutely certain. It is unlikely, but not impossible, that you did not understand some parts of the submission or that you are unfamiliar with some pieces of related work.

Soundness

3 good

Presentation

3 good

Contribution

3 good

Limitations

N.A.

Reviewer ZZ8V6/10 · confidence 2/52023-07-11

Summary

The paper investigates the effects of structural information in graph condensation methods. The authors claim that by maintaining the original structure during condensation using a newly formulated method called Structure-broadcasting Graph Dataset Distillation (SGDD), they are able to achieve more refined results on 9 data sets thereby significantly reducing Laplacian Energy Distribution (LED) shift.

Strengths

This paper addresses an important question regarding the impact of structural information. The authors conduct a comprehensive analysis from the spectral domain and empirically identify significant shifts in Laplacian Energy Distribution (LED), which ultimately result in poor performance in cross-architecture generalization and specific tasks. The proposed method demonstrates effectiveness in the experiments, and empirical analysis confirms the efficacy and necessity of the proposed designs. Overall, the presentation of the paper is also commendable

Weaknesses

large datasets in the Open Graph Benchmark (OGB) is more effective in demonstrating the efficacy of these methods.

Questions

I don't have question regarding this paper.

Rating

6: Weak Accept: Technically solid, moderate-to-high impact paper, with no major concerns with respect to evaluation, resources, reproducibility, ethical considerations.

Confidence

2: You are willing to defend your assessment, but it is quite likely that you did not understand the central parts of the submission or that you are unfamiliar with some pieces of related work. Math/other details were not carefully checked.

Soundness

3 good

Presentation

4 excellent

Contribution

3 good

Limitations

Need experiment for large datasets

Reviewer 5ivm5/10 · confidence 3/52023-07-12

Summary

This paper proposes a new graph condensation framework called SGDD. The authors argue that existing methods overlook the structure information of the original graph during the condensation. And thus they propose to 1) use the Laplacian Energy Distribution (LED) shift to indicate the generalization performance of the condensed graph, and 2) minimize the LED shift (they claim it’s equivalent to the OT distance between the original and the condensed graphs) to preserve the structure information. Experiments on different datasets over node classification, anomaly detection and link prediction tasks show the effectiveness of the proposed method.

Strengths

- The motivation of preserving the structure information is clear and reasonable. - The experiments seem relatively sufficient and the results are good.

Weaknesses

My major concern is the soundness of this paper: - Some claims seem to be problematic. For example, in line 176-179, the authors claim that ‘minimizing the LED shift is equivalent to minimizing the distance of Laplacian pseudo-inverse matrices [65]’ and ‘Following the previous work [78], minimizing such distance can be further approximated to optimizing a free parameter $P$ in Eq(6)’. However, the equivalence in [65] only applies to the objective using a specific GW distance and the approximation in [78] only applies to another OT distance based objective. So the two claims cannot hold simultaneously. What’s worse, the authors do not describe which OT distance is used in their original objective function. Then how could we conclude that ‘minimizing the upper bound of Eq. (7) is equal to optimizing the $L_{structure}$ on Eq. (5)’ in line 189? - The writing is not very easy to follow. The mixed use of ‘LED shift’, ‘LED shift coefficient’, ‘SC’ is very confusing (e.g., in Def2 and the caption of Fig2). And it’s not clear what’s the role of the proposed LED shift coefficient. This coefficient is neither used to derive the objective function, nor adopted as a performance metric in the tables for comparison. ============================= Post-rebuttal: The author responses have basically solved my concerns about the soundness.

Questions

Please see the above part.

Rating

5: Borderline accept: Technically solid paper where reasons to accept outweigh reasons to reject, e.g., limited evaluation. Please use sparingly.

Confidence

3: You are fairly confident in your assessment. It is possible that you did not understand some parts of the submission or that you are unfamiliar with some pieces of related work. Math/other details were not carefully checked.

Soundness

2 fair

Presentation

2 fair

Contribution

2 fair

Limitations

None.

Authorsrebuttal2023-08-17

Further Discussions with Reviewer 5ivm

Dear reviewer 5ivm: Thanks again for pointing out our key benefits of preserving the structure information. According to your constructive comments, we made the necessary adjustments on introducing the optimal transport and conducted time-consuming experiments on $SC$. Additionally, we have revised our writing of relevant concepts to enhance the accessibility for readers. As the rebuttal period is about to close, may I know if our rebuttal addresses your concerns? Thank you for taking the time to review our work and provide your insightful comments.

Authorsrebuttal2023-08-20

Looking forward to your reply!

Dear reviewer 5ivm, As we are nearing the rebuttal deadline, may I know if you have any other concerns regarding our work? Thanks! Best, Author of Submission 1279

Reviewer 5ivm2023-08-21

Thank you for the response

Thank the authors for the effort. My major concerns are basically solved, so I will raise the score to 5. However, the authors are recommended improving the writing/presentation carefully if accepted.

Authorsrebuttal2023-08-17

Further Discussions with Reviewer FmUF

Dear reviewer FmUF: Thanks for taking the time to review our paper. We detailed our motivation in the response and provide a more comprehensive study with additional 5 works. Hope our rebuttal has addressed your concerns. As the discussion period is nearing its end, please feel free to let us know if you have any other concerns. Thanks!

Authorsrebuttal2023-08-20

Looking forward to your reply!

Dear Reviewer FmUF, As we draw closer to the rebuttal deadline, I would like to inquire if you have any additional questions or concerns about our work. We greatly value your feedback. Thank you! Best, Authors from submission 1279

Authorsrebuttal2023-08-21

Dear Reviewer FmUF, Considering the limited time available, and in order to save the reviewer's time, we have concisely summarized our responses here. 1. ****More analyses and explanations of motivation:**** Question: The authors need to provide more analyses and explanations about their motivation. Response: **The core question in our paper is: “Does graph distillation see like vision dataset counterpart?“.** We start from the point of researching on “structure”, and choose the spectral perspective. The step-by-step stroyline is listed in our rebuttal. Question: The motivation for introducing LED is ambiguous, especially given the correlation does not mean causality. Response: **The correlation inspires us to design the SGDD method,** which effectively broadcast the structural information from the original graph dataset to a condensed version. Our SGDD achieves SOTA in most cases, including new SOTA results on YelpChi and Amazon with improvements of 9.6% and 7.7%, respectively. **Empirical results support the consistency between LED shift and generalization performance (see Fig. 1(b, e), Fig. 5(b, e), and response to Q4 of reviewer 5ivm)**. Compared to previous methods, our SGDD significantly improves performance while reducing $SC$. 2. ****Consider comparing with more baselines**** Question: The authors should consider compare with them as well for a comprehensive evaluation of their method. For example, [1, 2, 3, 4, 5]. Response: **We compared with 5 new baselines on 7 datasets,** including the dataset pruning methods and the latest graph condensation work SFGC (released later than NeurIPS submission deadline). **The results further proved the effectiveness of the graph dataset distillation as well as our method SGDD. We would like to add the experiment to our revisions.** Conclusion(s): - *We are endeavored to answer the question “Does graph distillation see like vision dataset counterpart?” through the spectral perspective.* - *Compared to other dataset pruning methods[2, A, 3, 5] and the latest graph condensation methods[1, 4], **our proposed SGDD still outperform them at least 2.9% in average, reflecting the importance of broadcasting the structural information in condensing datasets.*** Since the discussion period is about to close in several hours, may I know if our responses addressed your concerns? Please feel free to let us know if you have any other concerns. Thanks! Best, Authors from submission 1279

Reviewer UGAQ2023-08-18

Thanks the authors for their clear rebuttals and efforts, which have addressed my concerns and problems. Especially the extensive experiments on various datasets validate their effectiveness and superiority compared to baselines. In summary, this paper proposes a novel method called SGDD for condensing large-scale graph datasets while preserving the original structure information, and achieves state-of-the-art results. It is worth noting that data condensation has not been extensively explored in the field of graph learning, and this work addresses the problem from a novel perspective of graph structure, which will play a leading role in the field of work. I suggest this paper to be accepted.

Authorsrebuttal2023-08-19

Dear Reviewer UGAQ, We would like to express our sincere gratitude to the reviewer UGAQ for endorsing our work and providing constructive suggestions. Yes, we find that graph dataset distillation is largely different from vision dataset distillation, so we design our method. Thanks again for the time and effort in reviewing our work.

Authorsrebuttal2023-08-19

Looking forward to the updated evaluation

Dear Reviewer 9hTZ, Thanks for the response. Based on your reviews, we are looking forward to your updated evaluations of our work.

Authorsrebuttal2023-08-20

Dear Reviewer 9hTZ, As we are approaching the rebuttal deadline in less than two days, may I know if you have other concerns regarding our work? We truly value your insights and looking forward to your feedback! Best, Authors from Submission 1279

Authorsrebuttal2023-08-21

Dear Reviewer 9hTZ, Considering the limited time available, and in order to save the reviewer's time, we summarized our responses here. 1. **[Actual Computation Savings]** Question: It would be helpful to provide the actual computation savings in terms of time and memory usage. Response: We conduct the experiments on 5 datasets, and the results show that **our method achieves a speedup of at least 23.0X and a compression ratio of at least 3.9X.** 2. **[The Cost during Graph Distillation]** Question: The cost of graph distillation for condensed small graph dataset is recommended to be provided. Response: We conduct experiments to show the actual cost of the distillation process. **The results show that our SGDD require 1.6X-2.8X time to the naive training when condensing.** 3. **[Comparison with Other Methods]** Question: It would be beneficial to also compare the cost of graph distillation for the proposed method with other methods. Response: We conduct experiments on 6 other baselines on three datasets. From the table in the rebuttal, **our method only introduces a slightly higher computational cost than GDC and GCond (2.1X vs 1.9X and 2.0X), but achieves faster convergence (1.7X vs 2.6X and 3.0X).** Conclusion(s): - *While our method's cost is **1.6X-2.8X** higher, its value is evident when considering the acceleration rate of **23-51X** and compression rate of **3.9-15.9X** upon completion of the condensation process. For instance, training Ogbn-arxiv requires **17.4 GB** of memory at the beginning, but this can be reduced to just **1.2 GB** after condensation. **This makes our method highly beneficial for users with limited resources as well as boosting scenarios like neural architecture search and incremental learning.*** The author-reviewer discussion period will be closed less than hours, may I know if there are any other concerns? We truly value your insightful comment!

Authorsrebuttal2023-08-19

Further experiments on Ogbn-products

Dear reviewer ZZ8V: Thanks for your patience, we finished our experiments on Ogbn-products, and show the table as follows. We will add these results in the revision. **Results on Ogbn-products:** Note: we report the results of SGDD and the other 3 baselines on 7 different condensation ratios($r$). We use the $\Delta$ to denote the improvements for our proposed SGDD to K-Center. | Condensation Ratio ($r$) | Random | K-Center | GCond [B] | SGDD | $\Delta$ (%) | | --- | --- | --- | --- | --- | --- | | 0.0001% | 18.8 | 32.6 | 36.5 | 36.8 | $\uparrow$4.1 | | 0.0002% | 18.4 | 34.7 | 36.4 | 36.8 | $\uparrow$2.0 | | 0.0003% | 21.5 | 35.8 | 36.4 | 38.8 | $\uparrow$3.0 | | 0.0004% | 23.8 | 35.6 | 36.9 | 38.6 | $\uparrow$3.0 | | 0.0005% | 25.4 | 35.7 | 37.6 | 38.8 | $\uparrow$3.1 | | 0.001% | 34.8 | 35.4 | 38.2 | 40.1 | $\uparrow$4.7 | | 0.002% | 35.2 | 36.2 | 38.8 | 40.3 | $\uparrow$4.0 | Conclusion(s): - **Our proposed SGDD consistently outperforms other baselines across all condensation ratios with no meticulous tuning. Notably, it improves accuracy by 4.7% over K-Center at the 0.001% ratio, and by 1.8% and 5.3% over GCond [B] and Random, respectively.** We eagerly await your feedback to know if our experiments have adequately addressed your concerns. Please feel free if you have other questions. Thanks for your constructive suggestion again! [B] Wei Jin, et.al. Graph condensation for graph neural networks. ICLR 2022.

Reviewer ZZ8V2023-08-20

I would like to thank the authors for the detailed response. It helps clarify the work. Regrading the scores, it remains valid to reflect the quality of the work.

Authorsrebuttal2023-08-20

Dear Reviewer ZZ8V, we are truly grateful for your acknowledgment regarding our additional evaluation. We would like to express our sincere thanks once again for the reviewer’s work on reviewing our work!

Program Chairsdecision2023-09-21

Decision

Accept (poster)

© 2026 NYSGPT2525 LLC