用非欧几何建模社交网络中的同质性与影响力,提升链接生成效果。
Non-Euclidean Mixture Model for Social Network Embedding
- 将同质性建模为球面空间,影响力建模为双曲空间,分别捕捉循环与层级结构。
- 在多个公开数据集上,该模型在链接生成和分类任务中显著优于现有方法。
- 适合对社交网络形成机制、图神经网络几何建模感兴趣的读者。
社交网络中的链接形成主要源于同质性或社会影响。受此启发,我们提出一种基于嵌入的图生成模型,通过混合两种因素来理解链接生成过程。不同于传统图表示学习中将链接生成概率定义为节点嵌入的简单函数,本文将链接生成建模为两个因素的混合模型。其中,同质性因素在球面空间中建模,影响力因素在双曲空间中建模,以适应网络中存在循环和层级结构的事实。同时设计特殊投影实现两空间对齐。该模型称为非欧混合模型(NMM)。进一步将NMM集成到非欧图变分自编码器框架NMM-GNN中,该框架采用统一的非欧度量:非欧图神经网络编码器、非欧高斯先验、非欧解码器以及新颖的空间统一损失,实现不同非欧几何空间的融合。在多个公开数据集上的实验表明,NMM-GNN在社交网络生成与分类任务中显著优于现有先进基线,验证了其对社交网络形成机制的更好解释能力。
原文摘要 · Abstract (English)
It is largely agreed that social network links are formed due to either homophily or social influence. Inspired by this, we aim at understanding the generation of links via providing a novel embedding-based graph formation model. Different from existing graph representation learning, where link generation probabilities are defined as a simple function of the corresponding node embeddings, we model the link generation as a mixture model of the two factors. In addition, we model the homophily factor in spherical space and the influence factor in hyperbolic space to accommodate the fact that (1) homophily results in cycles and (2) influence results in hierarchies in networks. We also design a special projection to align these two spaces. We call this model Non-Euclidean Mixture Model, i.e., NMM. We further integrate NMM with our non-Euclidean graph variational autoencoder (VAE) framework, NMM-GNN. NMM-GNN learns embeddings through a unified framework which uses non-Euclidean GNN encoders, non-Euclidean Gaussian priors, a non-Euclidean decoder, and a novel space unification loss component to unify distinct non-Euclidean geometric spaces. Experiments on public datasets show NMM-GNN significantly outperforms state-of-the-art baselines on social network generation and classification tasks, demonstrating its ability to better explain how the social network is formed.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。