Slow Learning and Fast Inference: Efficient Graph Similarity Computation via Knowledge Distillation

被引:0
|
作者
Qin, Can [1 ]
Zhao, Handong [3 ]
Wang, Lichen [1 ]
Wang, Huan [1 ]
Zhang, Yulun [1 ]
Fu, Yun [1 ,2 ]
机构
[1] Northeastern Univ, Dept Elect & Comp Engn, Boston, MA 02115 USA
[2] Northeastern Univ, Khoury Coll Comp Sci, Boston, MA 02115 USA
[3] Adobe Res, San Jose, CA USA
关键词
D O I
暂无
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
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.
引用
收藏
页数:12
相关论文
共 50 条
  • [1] Accelerating Graph Similarity Search via Efficient GED Computation
    Chang, Lijun
    Feng, Xing
    Yao, Kai
    Qin, Lu
    Zhang, Wenjie
    [J]. IEEE TRANSACTIONS ON KNOWLEDGE AND DATA ENGINEERING, 2023, 35 (05) : 4485 - 4498
  • [2] More Interpretable Graph Similarity Computation via Maximum Common Subgraph Inference
    Lan, Zixun
    Hong, Binjie
    Ma, Ye
    Ma, Fei
    [J]. IEEE Transactions on Knowledge and Data Engineering, 2024, 36 (11) : 6588 - 6599
  • [3] Grand: A Fast and Accurate Graph Retrieval Framework via Knowledge Distillation
    Lan, Lin
    Wang, Pinghui
    Shi, Rui
    Liu, Tingqing
    Zeng, Juxiang
    Sun, Feiyang
    Ren, Yang
    Tao, Jing
    Guan, Xiaohong
    [J]. PROCEEDINGS OF THE 47TH INTERNATIONAL ACM SIGIR CONFERENCE ON RESEARCH AND DEVELOPMENT IN INFORMATION RETRIEVAL, SIGIR 2024, 2024, : 1639 - 1648
  • [4] Learning-Based Efficient Graph Similarity Computation via Multi-Scale Convolutional Set Matching
    Bai, Yunsheng
    Ding, Hao
    Gu, Ken
    Sun, Yizhou
    Wang, Wei
    [J]. THIRTY-FOURTH AAAI CONFERENCE ON ARTIFICIAL INTELLIGENCE, THE THIRTY-SECOND INNOVATIVE APPLICATIONS OF ARTIFICIAL INTELLIGENCE CONFERENCE AND THE TENTH AAAI SYMPOSIUM ON EDUCATIONAL ADVANCES IN ARTIFICIAL INTELLIGENCE, 2020, 34 : 3219 - 3226
  • [5] Communication-efficient federated learning via knowledge distillation
    Wu, Chuhan
    Wu, Fangzhao
    Lyu, Lingjuan
    Huang, Yongfeng
    Xie, Xing
    [J]. NATURE COMMUNICATIONS, 2022, 13 (01)
  • [6] Computation-Efficient Knowledge Distillation via Uncertainty-Aware Mixup
    Xu, Guodong
    Liu, Ziwei
    Loy, Chen Change
    [J]. PATTERN RECOGNITION, 2023, 138
  • [7] Knowledge Distillation via Constrained Variational Inference
    Saeedi, Ardavan
    Utsumi, Yuria
    Sun, Li
    Batmanghelich, Kayhan
    Lehman, Li-wei H.
    [J]. THIRTY-SIXTH AAAI CONFERENCE ON ARTIFICIAL INTELLIGENCE / THIRTY-FOURTH CONFERENCE ON INNOVATIVE APPLICATIONS OF ARTIFICIAL INTELLIGENCE / TWELVETH SYMPOSIUM ON EDUCATIONAL ADVANCES IN ARTIFICIAL INTELLIGENCE, 2022, : 8132 - 8140
  • [8] Neural Graph Similarity Computation with Contrastive Learning
    Hu, Shengze
    Zeng, Weixin
    Zhang, Pengfei
    Tang, Jiuyang
    [J]. APPLIED SCIENCES-BASEL, 2022, 12 (15):
  • [9] Learning Patient Similarity via Heterogeneous Medical Knowledge Graph Embedding
    Lin, Zhihuang
    Yang, Dan
    Jiang, Hua
    Yin, Hang
    [J]. IAENG International Journal of Computer Science, 2021, 48 (04)
  • [10] Local similarity preserved hashing learning via Markov graph for efficient similarity search
    Liu, Hong
    Jiang, Aiwen
    Wang, Mingwen
    Wan, Jianyi
    [J]. NEUROCOMPUTING, 2015, 159 : 144 - 150