Exploring attention mechanism for graph similarity learning

被引:5
作者
Tan, Wenhui [1 ]
Gao, Xin [1 ]
Li, Yiyang [1 ]
Wen, Guangqi [1 ]
Cao, Peng [1 ,2 ]
Yang, Jinzhu [1 ,2 ]
Li, Weiping [3 ]
Zaiane, Osmar R. [4 ]
机构
[1] Northeastern Univ, Comp Sci & Engn, Shenyang, Peoples R China
[2] Northeastern Univ, Key Lab Intelligent Comp Med Image, Minist Educ, Shenyang, Peoples R China
[3] Peking Univ, Sch Software & Microelect, Beijing, Peoples R China
[4] Univ Alberta, Alberta Machine Intelligence Inst, Edmonton, AB, Canada
关键词
Graph similarity; Graph neural network; Multi-head self-attention; Deep learning; Graph embedding learning; COMPUTATION;
D O I
10.1016/j.knosys.2023.110739
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Graph similarity estimation is a challenging task due to the complex graph structure. Though important and well-studied, three key aspects are yet to be fully handled in a unified framework: (i) how to exploit the node embedding by leveraging both local spatial neighborhood information and the global context, (ii) how to effectively learn richer cross graph interactions from a pairwise node perspective and (iii) how to map the similarity matrix into a similarity score by exploiting the inherent structure in the similarity matrix. To solve these issues, we explore multiple attention mechanisms for graph similarity learning in this work. More specifically, we propose a unified graph similarity learning framework involving (i) a hybrid of graph convolution and graph self-attention for node embedding learning, (ii) a cross graph co-attention (GCA) module for graph interaction modeling, (iii) similarity-wise self-attention (SSA) module for graph similarity matrix alignment and (iv) graph similarity matrix learning for predicting the similarity scores. Extensive experimental results on three challenging benchmarks including LINUX, AIDS, and IMDBMulti demonstrate that the proposed NA-GSL performs favorably against state-of-the-art graph similarity estimation methods. The code is available at https://github.com/AlbertTan404/NA-GSL. & COPY; 2023 Elsevier B.V. All rights reserved.
引用
收藏
页数:12
相关论文
共 50 条
  • [41] Graph to sequence learning with attention mechanism for network-wide multi-step-ahead flight delay prediction
    Bao, Jie
    Yang, Zhao
    Zeng, Weili
    [J]. TRANSPORTATION RESEARCH PART C-EMERGING TECHNOLOGIES, 2021, 130
  • [42] T-GAN: A deep learning framework for prediction of temporal complex networks with adaptive graph convolution and attention mechanism
    Huang, Ru
    Ma, Lei
    He, Jianhua
    Chu, Xiaoli
    [J]. DISPLAYS, 2021, 68
  • [43] SGA-Net: A Sparse Graph Attention Network for Two-View Correspondence Learning
    Liao, Tangfei
    Zhang, Xiaoqin
    Xu, Yuewang
    Shi, Ziwei
    Xiao, Guobao
    [J]. IEEE TRANSACTIONS ON CIRCUITS AND SYSTEMS FOR VIDEO TECHNOLOGY, 2023, 33 (12) : 7578 - 7590
  • [44] Robust Clustering Model Based on Attention Mechanism and Graph Convolutional Network
    Xia, Hui
    Shao, Shushu
    Hu, Chunqiang
    Zhang, Rui
    Qiu, Tie
    Xiao, Fu
    [J]. IEEE TRANSACTIONS ON KNOWLEDGE AND DATA ENGINEERING, 2023, 35 (05) : 5203 - 5215
  • [45] A Graph based Calligraphy Similarity Compare Model
    Pan, Guoyang
    Yang, Yi
    Li, Meng
    Hu, Xueyang
    Huang, Weixing
    Wang, Jian
    Wang, Yun
    [J]. 2021 21ST INTERNATIONAL CONFERENCE ON SOFTWARE QUALITY, RELIABILITY AND SECURITY COMPANION (QRS-C 2021), 2021, : 395 - 400
  • [46] Graph Similarity Using Tree Edit Distance
    Dwivedi, Shri Prakash
    Srivastava, Vishal
    Gupta, Umesh
    [J]. STRUCTURAL, SYNTACTIC, AND STATISTICAL PATTERN RECOGNITION, S+SSPR 2022, 2022, 13813 : 233 - 241
  • [47] Rumor detection based on propagation graph neural network with attention mechanism
    Wu, Zhiyuan
    Pi, Dechang
    Chen, Junfu
    Xie, Meng
    Cao, Jianjun
    [J]. EXPERT SYSTEMS WITH APPLICATIONS, 2020, 158
  • [48] Graph Sequence Neural Network with an Attention Mechanism for Traffic Speed Prediction
    Lu, Zhilong
    Lv, Weifeng
    Xie, Zhipu
    Du, Bowen
    Xiong, Guixi
    Sun, Leilei
    Wang, Haiquan
    [J]. ACM TRANSACTIONS ON INTELLIGENT SYSTEMS AND TECHNOLOGY, 2022, 13 (02)
  • [49] DGE-GSIM: A multi-task dual graph embedding learning for graph similarity computation
    Tan, Wenhui
    Cao, Peng
    Jin, Zhiyong
    Luo, Futao
    Wen, Guangqi
    Li, Weiping
    [J]. PROCEEDINGS OF 2022 THE 6TH INTERNATIONAL CONFERENCE ON MACHINE LEARNING AND SOFT COMPUTING, ICMLSC 20222, 2022, : 39 - 47
  • [50] Predicting the risk of mortality in ICU patients based on dynamic graph attention network of patient similarity
    Ma, Manfu
    Sun, Penghui
    Li, Yong
    Huo, Weilong
    [J]. MATHEMATICAL BIOSCIENCES AND ENGINEERING, 2023, 20 (08) : 15326 - 15344