Slow Learning and Fast Inference: Efficient Graph Similarity Computation via Knowledge Distillation
Can Qin, Handong Zhao, Lichen Wang, Huan Wang, Yulun Zhang, Yun Fu
Abstract
Graph Similarity Computation (GSC) is essential to wide-ranging graph applications such as retrieval, plagiarism/anomaly detection, etc. The exact computation of graph similarity, e.g., Graph Edit Distance (GED), is an NP-hard problem that cannot be exactly solved within an adequate time given large graphs. Thanks to the strong representation power of graph neural network (GNN), a variety of GNN-based inexact methods emerged. To capture the subtle difference across graphs, the key success is designing the dense interaction with features fusion at the early stage, which, however, is a trade-off between speed and accuracy. For Slow Learning of graph similarity, this paper proposes a novel early-fusion approach by designing a co-attention-based feature fusion network on multilevel GNN features. To further improve the speed without much accuracy drop, we introduce an efficient GSC solution by distilling the knowledge from the slow early-fusion model to the student one for Fast Inference. Such a student model also enables the offline collection of individual graph embeddings, speeding up the inference time in orders. To address the instability through knowledge transfer, we decompose the dynamic joint embedding into the static pseudo individual ones for precise teacher-student alignment. The experimental analysis on the real-world datasets demonstrates the superiority of our approach over the state-of-the-art methods on both accuracy and efficiency. Particularly, we speed up the prior art by more than 10x on the benchmark AIDS data.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Cited by top-tier papers13
- Efficient Graph Similarity Computation with Alignment RegularizationWei Zhuo, Guang TanNeurIPS 2022 · 48 citations
- Graph Edit Distance with General Costs Using Neural Set DivergenceEeshaan Jain, Indradyumna Roy, Saswat Meher, Soumen Chakrabarti et al.NeurIPS 2024 · 26 citations
- Efficient Traffic Prediction Through Spatio-Temporal DistillationQianru Zhang, Xinyi Gao, Haixin Wang, Siu Ming Yiu et al.AAAI 2025 · 22 citations
- Iteratively Refined Early Interaction Alignment for Subgraph Matching based Graph RetrievalAshwin Ramachandran, Vaibhav Raj, Indradyumna Roy, Soumen Chakrabarti et al.NeurIPS 2024 · 7 citations
- Rapid and Precise Topological Comparison with Merge Tree Neural NetworksYu Qin, Brittany Terese Fasy, Carola Wenk, Brian SummaIEEE VIS 2024 · 5 citations
Builds on8
- Contrastive Representation DistillationYonglong Tian, Dilip Krishnan, Phillip IsolaICLR 2020 · 1,305 citations
- Similarity-Preserving Knowledge DistillationFrederick Tung, Greg MoriICCV 2019 · 1,214 citations
- Correlation Congruence for Knowledge DistillationBaoyun Peng, Xiao Jin, Dongsheng Li, Shunfeng Zhou et al.ICCV 2019 · 625 citations
- Inductive Representation Learning in Temporal Networks via Causal Anonymous WalksYanbang Wang, Yen-Yu Chang, Yunyu Liu, Jure Leskovec et al.ICLR 2021 · 326 citations
- Learning-Based Efficient Graph Similarity Computation via Multi-Scale Convolutional Set MatchingYunsheng Bai, Hao Ding, Ken Gu, Yizhou Sun et al.AAAI 2020 · 130 citations
Related papers
- Noah: Neural-optimized A* Search Algorithm for Graph Edit Distance ComputationLei Yang, Lei ZouICDE 2021 · 20 citations
- GHashing: Semantic Graph Hashing for Approximate Similarity Search in Graph DatabasesZongyue Qin, Yunsheng Bai, Yizhou SunKDD 2020 · 29 citations
- GraSP: Simple Yet Effective Graph Similarity PredictionsHaoran Zheng, Jieming Shi, Renchi YangAAAI 2025 · 1 citation
- Boosting Graph Similarity Search through Pre-ComputationJongik KimSIGMOD 2021 · 10 citations
- Computing Graph Edit Distance via Neural Graph MatchingChengzhi Piao, Tingyang Xu, Xiangguo Sun, Yu Rong et al.VLDB 2023 · 49 citations
