Improving link prediction accuracy of network embedding algorithms via rich node attribute information

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Gu, Weiwei, Hou, Jinqiang, Gu, Weiyi
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866909131120574464
author Gu, Weiwei
Hou, Jinqiang
Gu, Weiyi
author_facet Gu, Weiwei
Hou, Jinqiang
Gu, Weiyi
contents Complex networks are widely used to represent an abundance of real-world relations ranging from social networks to brain networks. Inferring missing links or predicting future ones based on the currently observed network is known as the link prediction task.Recent network embedding based link prediction algorithms have demonstrated ground-breaking performance on link prediction accuracy. Those algorithms usually apply node attributes as the initial feature input to accelerate the convergence speed during the training process. However, they do not take full advantage of node feature information. In this paper,besides applying feature attributes as the initial input, we make better utilization of node attribute information by building attributable networks and plugging attributable networks into some typical link prediction algorithms and naming this algorithm Attributive Graph Enhanced Embedding (AGEE). AGEE is able to automatically learn the weighting trades-off between the structure and the attributive networks. Numerical experiments show that AGEE can improve the link prediction accuracy by around 3% compared with link prediction framework SEAL, Variational Graph AutoEncoder (VGAE), and Node2vec.
format Preprint
id arxiv_https___arxiv_org_abs_2403_04282
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Improving link prediction accuracy of network embedding algorithms via rich node attribute information
Gu, Weiwei
Hou, Jinqiang
Gu, Weiyi
Social and Information Networks
Complex networks are widely used to represent an abundance of real-world relations ranging from social networks to brain networks. Inferring missing links or predicting future ones based on the currently observed network is known as the link prediction task.Recent network embedding based link prediction algorithms have demonstrated ground-breaking performance on link prediction accuracy. Those algorithms usually apply node attributes as the initial feature input to accelerate the convergence speed during the training process. However, they do not take full advantage of node feature information. In this paper,besides applying feature attributes as the initial input, we make better utilization of node attribute information by building attributable networks and plugging attributable networks into some typical link prediction algorithms and naming this algorithm Attributive Graph Enhanced Embedding (AGEE). AGEE is able to automatically learn the weighting trades-off between the structure and the attributive networks. Numerical experiments show that AGEE can improve the link prediction accuracy by around 3% compared with link prediction framework SEAL, Variational Graph AutoEncoder (VGAE), and Node2vec.
title Improving link prediction accuracy of network embedding algorithms via rich node attribute information
topic Social and Information Networks
url https://arxiv.org/abs/2403.04282