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

神经随机块模型与可扩展的基于社区的图学习

Neural Stochastic Block Model & Scalable Community-Based Graph Learning

Amazon Alexa AI(亚马逊Alexa人工智能部门);Temple University(天普大学)

Zheng Chen, Xinli Yu, Yuan Ling, Xiaohua Hu

摘要:

本文提出了一种新的可扩展的基于社区的图学习神经框架。该框架通过社区发现和链接预测的任务来学习图拓扑,通过我们提出的联合SBM损失函数进行优化,该函数来自经典随机块模型(SBM)的似然函数的非平凡适应。与SBM相比,我们的框架是灵活的,自然允许软标签和消化复杂的节点属性。主要目标是对复杂图形数据进行有效评估,因此我们的设计旨在适应大数据,并确保有一个有效评估的单一向前传递。对于大型图,如何有效地利用其底层结构进行各种图学习任务仍然是一个悬而未决的问题。以前可能是繁重的工作。通过我们基于社区的框架,这变得不那么困难,并允许任务模型基本上即插即用并执行联合训练。我们目前正在研究两个特定的应用程序,图对齐和异常相关检测,并讨论如何利用我们的框架来解决这两个问题。大量的实验证明了我们的方法的有效性。我们还贡献了经典技术的调整,我们发现这对性能和可扩展性很有帮助。例如,1)GAT+,GAT(图注意力网络)的改进设计,缩放余弦相似性,以及基于卷积/注意力和基于随机游走的神经图模型的统一实现。

英文摘要:

This paper proposes a novel scalable community-based neural framework for graph learning. The framework learns the graph topology through the task of community detection and link prediction by optimizing with our proposed joint SBM loss function, which results from a non-trivial adaptation of the likelihood function of the classic Stochastic Block Model (SBM). Compared with SBM, our framework is flexible, naturally allows soft labels and digestion of complex node attributes. The main goal is efficient valuation of complex graph data, therefore our design carefully aims at accommodating large data, and ensures there is a single forward pass for efficient evaluation. For large graph, it remains an open problem of how to efficiently leverage its underlying structure for various graph learning tasks. Previously it can be heavy work. With our community-based framework, this becomes less difficult and allows the task models to basically plug-in-and-play and perform joint training. We currently look into two particular applications, the graph alignment and the anomalous correlation detection, and discuss how to make use of our framework to tackle both problems. Extensive experiments are conducted to demonstrate the effectiveness of our approach. We also contributed tweaks of classic techniques which we find helpful for performance and scalability. For example, 1) the GAT+, an improved design of GAT (Graph Attention Network), the scaled-cosine similarity, and a unified implementation of the convolution/attention based and the random-walk based neural graph models.

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

↑