arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 1908.04003cs.LGcs.SIstat.ML
收藏

RWR-GAE: Random Walk Regularization for Graph Auto Encoders

Carnegie Mellon University(卡内基梅隆大学)

Vaibhav, Po-Yao Huang, Robert Frederking

英文摘要:

Node embeddings have become an ubiquitous technique for representing graph data in a low dimensional space. Graph autoencoders, as one of the widely adapted deep models, have been proposed to learn graph embeddings in an unsupervised way by minimizing the reconstruction error for the graph data. However, its reconstruction loss ignores the distribution of the latent representation, and thus leading to inferior embeddings. To mitigate this problem, we propose a random walk based method to regularize the representations learnt by the encoder. We show that the proposed novel enhancement beats the existing state-of-the-art models by a large margin (upto 7.5\%) for node clustering task, and achieves state-of-the-art accuracy on the link prediction task for three standard datasets, cora, citeseer and pubmed. Code available at https://github.com/MysteryVaibhav/DW-GAE.

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

↑