TY - JOUR
T1 - 融合社团信息的时序图链路预测算法
AU - Wu, Xiang
AU - Gao, Yujin
AU - Li, Ronghua
AU - Wang, Guoren
N1 - Publisher Copyright:
© 2024 Journal of Computer Engineering and Applications Beijing Co., Ltd.; Science Press. All rights reserved.
PY - 2024/10/1
Y1 - 2024/10/1
N2 - Temporal graphs aim to characterize the dynamic interactions among entities in real- world scenarios, while temporal link prediction is an essential method for modeling such relationships. Existing representation learning-based methods for temporal graph link prediction typically involve designing temporal graph neural networks to model interactions between nodes. The message- passing mechanisms of graph neural networks are incorporated with temporal information, enabling the models to generate time- specified embeddings for link prediction tasks. However, existing methods only consider interactions between nodes, neglecting the prevalent community structure in temporal graphs. To address this limitation, this paper proposes a temporal link prediction algorithm, TLPC (temporal link prediction enhanced by community-level information). Unlike traditional community detection methods, this paper focuses on representing the neighbor communities of a node. Leveraging contrastive learning techniques on top of node features, positive and negative samples are sampled based on neighborhood structural features for contrastive learning constraints, encoding the neighbor communities of nodes. The learnt encoding of node neighborhood communities effectively enhances the representation ability of node embeddings, thereby improving link prediction effectiveness. Experimental results on four real-life datasets for temporal link prediction tasks demonstrate that TLPC achieves an average accuracy increase of 6.47% and an average F1 score increase of 6.77% compared with existing methods, while simultaneously reducing training time by an average of 62.27%.
AB - Temporal graphs aim to characterize the dynamic interactions among entities in real- world scenarios, while temporal link prediction is an essential method for modeling such relationships. Existing representation learning-based methods for temporal graph link prediction typically involve designing temporal graph neural networks to model interactions between nodes. The message- passing mechanisms of graph neural networks are incorporated with temporal information, enabling the models to generate time- specified embeddings for link prediction tasks. However, existing methods only consider interactions between nodes, neglecting the prevalent community structure in temporal graphs. To address this limitation, this paper proposes a temporal link prediction algorithm, TLPC (temporal link prediction enhanced by community-level information). Unlike traditional community detection methods, this paper focuses on representing the neighbor communities of a node. Leveraging contrastive learning techniques on top of node features, positive and negative samples are sampled based on neighborhood structural features for contrastive learning constraints, encoding the neighbor communities of nodes. The learnt encoding of node neighborhood communities effectively enhances the representation ability of node embeddings, thereby improving link prediction effectiveness. Experimental results on four real-life datasets for temporal link prediction tasks demonstrate that TLPC achieves an average accuracy increase of 6.47% and an average F1 score increase of 6.77% compared with existing methods, while simultaneously reducing training time by an average of 62.27%.
KW - community structure
KW - contrastive learning
KW - temporal link prediction
UR - http://www.scopus.com/inward/record.url?scp=85206505071&partnerID=8YFLogxK
U2 - 10.3778/j.issn.1673-9418.2310022
DO - 10.3778/j.issn.1673-9418.2310022
M3 - 文章
AN - SCOPUS:85206505071
SN - 1673-9418
VL - 18
SP - 2668
EP - 2677
JO - Journal of Frontiers of Computer Science and Technology
JF - Journal of Frontiers of Computer Science and Technology
IS - 10
ER -