arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 1706.05136stat.ML
收藏

带有边信息的关系数据的深度生成模型

Deep Generative Models for Relational Data with Side Information

Changwei Hu, Piyush Rai, Lawrence Carin

摘要:

我们提出了一个概率框架重叠的社区发现和链接预测的关系数据,作为一个图形。拟议框架包括:(1)深度架构,其使我们能够推断每个节点的多层潜在特征/社区,在更复杂的网络上提供更好的链路预测性能和潜在特征的更好的可解释性;以及(2)回归模型,其允许直接根据以节点属性的形式可用的边信息来调节节点潜在特征。我们的框架通过一个干净、统一的模型来处理(1)和(2),该模型通过数据增强来享受完全的局部共轭性,并通过封闭形式的吉布斯抽样来促进有效的推理。此外,推理成本在边缘的数量上成比例,这对于大规模但稀疏的网络是有吸引力的。我们的框架也很容易扩展到模型加权网络计数值的边缘。我们比较各种国家的最先进的方法和报告结果,定量和定性,在几个基准数据集。

英文摘要:

We present a probabilistic framework for overlapping community discovery and link prediction for relational data, given as a graph. The proposed framework has: (1) a deep architecture which enables us to infer multiple layers of latent features/communities for each node, providing superior link prediction performance on more complex networks and better interpretability of the latent features; and (2) a regression model which allows directly conditioning the node latent features on the side information available in form of node attributes. Our framework handles both (1) and (2) via a clean, unified model, which enjoys full local conjugacy via data augmentation, and facilitates efficient inference via closed form Gibbs sampling. Moreover, inference cost scales in the number of edges which is attractive for massive but sparse networks. Our framework is also easily extendable to model weighted networks with count-valued edges. We compare with various state-of-the-art methods and report results, both quantitative and qualitative, on several benchmark data sets.

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

↑