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

基于邻域混合模型的图节点相关性建模

Modeling Graph Node Correlations with Neighbor Mixture Models

Tufts University(塔夫茨大学)

Linfeng Liu, Michael C. Hughes, Li-Ping Liu

摘要:

我们提出了一个新的模型,邻居混合模型(NMM),在一个图中的节点标签建模。该模型旨在捕获局部邻域中节点标签之间的相关性。我们仔细设计了模型,使其可以替代马尔可夫随机场,但更实惠的计算。特别是,绘制样本和评估单个标签的边际概率可以在线性时间内完成。为了将计算扩展到大型图,我们设计了一个变分近似,而不引入额外的参数。我们进一步使用图神经网络(GNNs)来参数化NMM,这减少了可学习参数的数量,同时允许表达性表示学习。所提出的模型可以直接拟合大型观察图,也可以用于实现可扩展的推理,以保留其他分布(例如深度生成图模型)的相关性。在一组不同的节点分类,图像去噪,和链接预测任务,我们表明我们提出的NMM先进的最先进的建模现实世界的标记图。

英文摘要:

We propose a new model, the Neighbor Mixture Model (NMM), for modeling node labels in a graph. This model aims to capture correlations between the labels of nodes in a local neighborhood. We carefully design the model so it could be an alternative to a Markov Random Field but with more affordable computations. In particular, drawing samples and evaluating marginal probabilities of single labels can be done in linear time. To scale computations to large graphs, we devise a variational approximation without introducing extra parameters. We further use graph neural networks (GNNs) to parameterize the NMM, which reduces the number of learnable parameters while allowing expressive representation learning. The proposed model can be either fit directly to large observed graphs or used to enable scalable inference that preserves correlations for other distributions such as deep generative graph models. Across a diverse set of node classification, image denoising, and link prediction tasks, we show our proposed NMM advances the state-of-the-art in modeling real-world labeled graphs.

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

↑