Graph representation learning based on deep generative gaussian mixture models

doi:10.1016/j.neucom.2022.11.087

Graph representation learning based on deep generative gaussian mixture models

2023-02-01

8

OA

AI

DOI：10.1016/j.neucom.2022.11.087

Original

Save

View PDF

Abstract

En 中文

Graph representation learning is an effective tool for facilitating graph analysis with machine learning methods. Most GNNs, including Graph Convolutional Networks (GCN), Graph Recurrent Neural Networks (GRNN), and Graph Auto-Encoders (GAE), employ vectors to represent nodes in a deterministic way without exploiting the uncertainty in hidden variables. Deep generative models are combined with GAE in the Variational Graph Auto-Encoder (VGAE) framework to address this issue. While traditional VGAE-based methods can capture hidden and hierarchical dependencies in latent spaces, they are limited by the data's multimodality. Here, we propose the Gaussian Mixture Model (GMM) to model the prior distribution in VGAE. Furthermore, an adversarial regularization is incorporated into the proposed approach to ensure the fruitful impact of the latent representations on the results. We demonstrate the performance of the proposed method on clustering and link prediction tasks. Our experimental results on real datasets show remarkable performance compared to state-of-the-art methods.(c) 2022 Elsevier B.V. All rights reserved.

Keywords:

Graph representation learning

Node embedding

Variational graph auto -encoder

Adversarial mechanism

Gaussian mixture model

Graph representation learning based on deep generative gaussian mixture models

Abstract

AI Summary

Journal

Organization