Subgraph representation learning with self-attention and free adversarial training

Denggao Qin,Xianghong Tang,Jianguang Lu
DOI: https://doi.org/10.1007/s10489-024-05542-7
IF: 5.3
2024-06-01
Applied Intelligence
Abstract:Due to its capacity to capture subgraph information within graph data, subgraph representation learning has garnered considerable attention in recent years. However, current subgraph representation learning methods neglect the relational information among the subgraphs and noise interference from the subgraph nodes. To address these issues, this paper proposes a method of subgraph representation learning with self-attention and free adversarial training (SGSFA). Specifically, this method employs a self-attention mechanism to calculate the similarity among the subgraphs and then captures the relational information among the subgraphs based on this similarity. Additionally, this method utilizes an improved free adversarial training approach to generate iterative perturbations. These iterative perturbations can eliminate the noise that hampers subgraph representation. Simultaneously, an iterative perturbation is applied to the subgraph nodes to enhance the subgraph's node features. Extensive results on four real-world subgraph datasets demonstrate that the proposed method outperforms state-of-the-art baselines. The source code of SGSFA is publicly available at https://github.com/denggaoqin/SGSFA.
computer science, artificial intelligence
What problem does this paper attempt to address?