arXivDaily arXiv每日学术速递 周一至周五更新
北美退伍军人节(Veterans Day),12 日无更新
arXiv 2010.04895cs.LGcs.DBstat.ML
收藏

UniNet:基于Metropolis-Hastings抽样的可扩展网络表示学习

UniNet: Scalable Network Representation Learning with Metropolis-Hastings Sampling

Beijing University of Posts and Telecommunications(北京邮电大学);Peking University(北京大学);Hong Kong University of Science and Technology(香港科技大学)

Xingyu Yao, Yingxia Shao, Bin Cui, Lei Chen

摘要:

网络表示学习(NRL)技术已成功地应用于各种数据采掘和机器学习应用中。基于随机游走的自然学习是一种流行的范式,它使用一组随机游走来捕获网络结构信息,然后使用word 2 vec模型来学习低维表示。然而,到目前为止,缺乏一个框架,统一现有的基于随机游走的NRL模型,并支持有效地从大型网络学习。主要障碍来自不同的随机游走模型和随机游走生成的低效采样方法。本文首先基于Metropolis-Hastings抽样技术提出了一种新的有效的边缘抽样器,并从理论上证明了该边缘抽样器对任意离散概率分布的收敛性。然后,我们提出了一个随机游走模型的抽象,用户可以很容易地定义不同的转移概率,通过指定动态边权重和随机游走状态。我们的边缘采样器有效地支持了这种抽象,因为我们的采样器可以在恒定的时间复杂度下从未归一化的概率分布中提取样本。最后,通过新的边缘采样器和随机游走模型抽象,我们仔细实现了一个可扩展的NRL框架,称为UniNet。我们在11个真实世界的数据集上使用5个基于随机游走的NRL模型进行了全面的实验,结果清楚地表明了UniNet在十亿边网络上的效率。

英文摘要:

Network representation learning (NRL) technique has been successfully adopted in various data mining and machine learning applications. Random walk based NRL is one popular paradigm, which uses a set of random walks to capture the network structural information, and then employs word2vec models to learn the low-dimensional representations. However, until now there is lack of a framework, which unifies existing random walk based NRL models and supports to efficiently learn from large networks. The main obstacle comes from the diverse random walk models and the inefficient sampling method for the random walk generation. In this paper, we first introduce a new and efficient edge sampler based on Metropolis-Hastings sampling technique, and theoretically show the convergence property of the edge sampler to arbitrary discrete probability distributions. Then we propose a random walk model abstraction, in which users can easily define different transition probability by specifying dynamic edge weights and random walk states. The abstraction is efficiently supported by our edge sampler, since our sampler can draw samples from unnormalized probability distribution in constant time complexity. Finally, with the new edge sampler and random walk model abstraction, we carefully implement a scalable NRL framework called UniNet. We conduct comprehensive experiments with five random walk based NRL models over eleven real-world datasets, and the results clearly demonstrate the efficiency of UniNet over billion-edge networks.

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

↑