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 条
  • [1] Exploring Attention and Self-Supervised Learning Mechanism for Graph Similarity Learning
    Wen, Guangqi
    Gao, Xin
    Tan, Wenhui
    Cao, Peng
    Yang, Jinzhu
    Li, Weiping
    Zaiane, Osmar R.
    IEEE TRANSACTIONS ON NEURAL NETWORKS AND LEARNING SYSTEMS, 2024,
  • [2] A Graph Similarity Algorithm Based on Graph Partitioning and Attention Mechanism
    Miao, Fengyu
    Zhou, Xiuzhuang
    Xiao, Shungen
    Zhang, Shiliang
    ELECTRONICS, 2024, 13 (19)
  • [3] Cosine similarity distance pruning algorithm Based on graph attention mechanism
    Yao, Huaxiong
    Huang, Yang
    Hu, Jiabei
    Xie, Wenqi
    2020 IEEE INTERNATIONAL CONFERENCE ON BIG DATA (BIG DATA), 2020, : 3311 - 3318
  • [4] Binary Function Similarity Detection Based on Graph Neural Network with Self-Attention Mechanism
    Wu, Dingjie
    He, Xuanzhang
    Zhang, Yao
    Zhu, Junjie
    Zhang, Xinyuan
    Ye, Minchao
    Gao, Zhigang
    2022 IEEE INTL CONF ON DEPENDABLE, AUTONOMIC AND SECURE COMPUTING, INTL CONF ON PERVASIVE INTELLIGENCE AND COMPUTING, INTL CONF ON CLOUD AND BIG DATA COMPUTING, INTL CONF ON CYBER SCIENCE AND TECHNOLOGY CONGRESS (DASC/PICOM/CBDCOM/CYBERSCITECH), 2022, : 971 - 975
  • [5] Heterogeneous Graph Contrastive Learning with Attention Mechanism for Recommendation
    Li, Ruxing
    Yang, Dan
    Gong, Xi
    ENGINEERING LETTERS, 2024, 32 (10) : 1930 - 1938
  • [6] Multilevel Graph Matching Networks for Deep Graph Similarity Learning
    Ling, Xiang
    Wu, Lingfei
    Wang, Saizhuo
    Ma, Tengfei
    Xu, Fangli
    Liu, Alex X.
    Wu, Chunming
    Ji, Shouling
    IEEE TRANSACTIONS ON NEURAL NETWORKS AND LEARNING SYSTEMS, 2023, 34 (02) : 799 - 813
  • [7] Robust Representation Learning via Sparse Attention Mechanism for Similarity Models
    Ermilova, Alina
    Baramiia, Nikita
    Kornilov, Valerii
    Petrakov, Sergey
    Zaytsev, Alexey
    IEEE ACCESS, 2024, 12 : 97833 - 97850
  • [8] Learning From Crowds Using Graph Neural Networks With Attention Mechanism
    Zhang, Jing
    Wu, Ming
    Sun, Zeyi
    Zhou, Cangqi
    IEEE TRANSACTIONS ON BIG DATA, 2025, 11 (01) : 86 - 98
  • [9] Two-level attention mechanism with contrastive learning for heterogeneous graph representation learning
    Moradi, Mahnaz
    Moradi, Parham
    Faroughi, Azadeh
    Jalili, Mahdi
    EXPERT SYSTEMS WITH APPLICATIONS, 2025, 273
  • [10] Graph similarity learning for cross-level interactions
    Zou, Cuifang
    Lu, Guangquan
    Du, Longqing
    Zeng, Xuxia
    Lin, Shilong
    INFORMATION PROCESSING & MANAGEMENT, 2025, 62 (01)