arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2403.08613cs.SIcs.AIcs.LG
收藏

使用表示学习和基于启发式特征的社会网络链接预测

Link Prediction for Social Networks using Representation Learning and Heuristic-based Features

Indian Institute of Engineering Science and Technology, Shibpur(印度工程科学技术学院 Shibpur校区)

Samarth Khanna, Sree Bhattacharyya, Sudipto Ghosh, Kushagra Agarwal, Asit Kumar Das

AI总结:

本文探索多种特征提取技术生成节点和边表示,并融合启发式特征与学习表示,以提升社交网络链接预测性能。

摘要:

社交网络在规模和相关性上的指数级增长使其能够提供广泛的洞察。高效预测社交网络中的缺失链接有助于各种现代商业应用,从生成推荐到影响力分析。针对这一问题,已有多种类别的解决方案。在此,我们探索各种特征提取技术来生成社交网络中节点和边的表示,从而预测缺失链接。我们比较了十种特征提取技术的结果,这些技术分为结构嵌入、基于邻域的嵌入、图神经网络和图启发式方法,随后使用集成分类器和自定义神经网络进行建模。此外,我们提出将基于启发式的特征与学习到的表示相结合,这在社会网络数据集上的链接预测任务中展示了改进的性能。使用这种方法为许多应用生成准确的推荐是进一步研究的方向,且前景十分广阔。所有实验的代码已公开。

英文摘要:

The exponential growth in scale and relevance of social networks enable them to provide expansive insights. Predicting missing links in social networks efficiently can help in various modern-day business applications ranging from generating recommendations to influence analysis. Several categories of solutions exist for the same. Here, we explore various feature extraction techniques to generate representations of nodes and edges in a social network that allow us to predict missing links. We compare the results of using ten feature extraction techniques categorized across Structural embeddings, Neighborhood-based embeddings, Graph Neural Networks, and Graph Heuristics, followed by modeling with ensemble classifiers and custom Neural Networks. Further, we propose combining heuristic-based features and learned representations that demonstrate improved performance for the link prediction task on social network datasets. Using this method to generate accurate recommendations for many applications is a matter of further study that appears very promising. The code for all the experiments has been made public.

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

↑