GraKerformer: A Transformer With Graph Kernel for Unsupervised Graph Representation Learning
While highly influential in deep learning, especially in natural language processing, the Transformer model has not exhibited competitive performance in unsupervised graph representation learning (UGRL). Conventional approaches, which focus on local substructures on the graph, offer simplicity but often fall short in encapsulating comprehensive structural information of the graph. This deficiency leads to suboptimal generalization performance. To address this, we proposed the GraKerformer model, a variant of the standard Transformer architecture, to mitigate the shortfall in structural information representation and enhance the performance in UGRL. By leveraging the shortest-path graph kernel (SPGK) to weight attention scores and combining graph neural networks, the GraKerformer effectively encodes the nuanced structural information of graphs. We conducted evaluations on the benchmark datasets for graph classification to validate the superior performance of our approach.
Authors
- Xin Yuan (ORCID: https://orcid.org/0000-0001-5056-171X)
- Lixiang Xu (ORCID: https://orcid.org/0000-0001-8946-620X)
- Enhong Chen (ORCID: https://orcid.org/0000-0002-4835-4102)
- Yuanyan Tang
- Haifeng Liu (ORCID: https://orcid.org/0000-0003-3319-254X)
Institutions
- University of Science and Technology of China (CN)
- University of Macau (MO)
- Hefei University (CN)
- The University of Adelaide (AU)
Publication Details
- Journal
- IEEE Transactions on Cybernetics
- Published
- 2024-10-08
- DOI
- https://doi.org/10.1109/tcyb.2024.3465213
- Citations
- 12
- Primary Topic
- Advanced Graph Neural Networks
- Type
- article
- Field-Weighted Citation Impact
- 4.17
Funders
- National Natural Science Foundation of China