arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2004.01468cs.LGstat.ML

Epitomic Variational Graph Autoencoder

Technical University of Munich(慕尼黑工业大学);Mercateo AG(Mercateo公司)

Rayyan Ahmad Khan, Muhammad Umer Anwaar, Martin Kleinsteuber

英文摘要:

Variational autoencoder (VAE) is a widely used generative model for learning latent representations. Burda et al. in their seminal paper showed that learning capacity of VAE is limited by over-pruning. It is a phenomenon where a significant number of latent variables fail to capture any information about the input data and the corresponding hidden units become inactive. This adversely affects learning diverse and interpretable latent representations. As variational graph autoencoder (VGAE) extends VAE for graph-structured data, it inherits the over-pruning problem. In this paper, we adopt a model based approach and propose epitomic VGAE (EVGAE),a generative variational framework for graph datasets which successfully mitigates the over-pruning problem and also boosts the generative ability of VGAE. We consider EVGAE to consist of multiple sparse VGAE models, called epitomes, that are groups of latent variables sharing the latent space. This approach aids in increasing active units as epitomes compete to learn better representation of the graph data. We verify our claims via experiments on three benchmark datasets. Our experiments show that EVGAE has a better generative ability than VGAE. Moreover, EVGAE outperforms VGAE on link prediction task in citation networks.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑