Papers › MoCoKGC: Momentum Contrast Entity Encoding for Knowledge Graph Completion

MoCoKGC: Momentum Contrast Entity Encoding for Knowledge Graph Completion

12 Nov 2024Empirical Methods in Natural Language Processing 2024 11archive 2025-07-28

Qingyang Li, Yanru Zhong, YuChu Qin

In recent years, numerous studies have sought to enhance the capabilities of pretrained language models (PLMs) for Knowledge Graph Completion (KGC) tasks by integrating structural information from knowledge graphs. However, existing approaches have not effectively combined the structural attributes of knowledge graphs with the textual descriptions of entities to generate robust entity encodings.To address this issue, this paper proposes MoCoKGC (Momentum Contrast Entity Encoding for Knowledge Graph Completion), which incorporates three primary encoders: the entity-relation encoder, the entity encoder, and the momentum entity encoder. Momentum contrastive learning not only provides more negative samples but also allows for the gradual updating of entity encodings. Consequently, we reintroduce the generated entity encodings into the encoder to incorporate the graph’s structural information.Additionally, MoCoKGC enhances the inferential capabilities of the entity-relation encoder through deep prompts of relations. On the standard evaluation metric, Mean Reciprocal Rank (MRR), the MoCoKGC model demonstrates superior performance, achieving a 7.1% improvement on the WN18RR dataset and an 11% improvement on the Wikidata5M dataset, while also surpassing the current best model on the FB15k-237 dataset. Through a series of experiments, this paper thoroughly examines the role and contribution of each component and parameter of the model.

PaperPDF

Code

No code repository is listed for this paper in the archive or in Syntology's graph.

Code Syntology ran Syntology

Not run by Syntology. Nothing on this page verifies that the listed code works.

Tasks

Contrastive LearningKnowledge Graph CompletionKnowledge GraphsLink Prediction

1 archive task tag without a task page not shown.

Results from the paper archive 2025-07-28

TaskDatasetModelMetricValueRank at snapshotLeaderboardReport
Link Prediction FB15k-237 MoCoKGC Hits@1 0.296 #7 of 75 Archive leaderboard report
Link Prediction FB15k-237 MoCoKGC Hits@10 0.580 #7 of 75 Archive leaderboard report
Link Prediction FB15k-237 MoCoKGC Hits@3 0.431 #7 of 75 Archive leaderboard report
Link Prediction FB15k-237 MoCoKGC MRR 0.391 #7 of 75 Archive leaderboard report
Link Prediction WN18RR MoCoKGC Hits@1 0.665 #1 of 75 Archive leaderboard report
Link Prediction WN18RR MoCoKGC Hits@10 0.881 #1 of 75 Archive leaderboard report
Link Prediction WN18RR MoCoKGC Hits@3 0.792 #1 of 75 Archive leaderboard report
Link Prediction WN18RR MoCoKGC MRR 0.742 #1 of 75 Archive leaderboard report
Link Prediction Wikidata5M MoCoKGC Hits@1 0.435 #1 of 14 Archive leaderboard report
Link Prediction Wikidata5M MoCoKGC Hits@10 0.591 #1 of 14 Archive leaderboard report
Link Prediction Wikidata5M MoCoKGC Hits@3 0.517 #1 of 14 Archive leaderboard report
Link Prediction Wikidata5M MoCoKGC MRR 0.490 #1 of 14 Archive leaderboard report

Ranks are positions in the archive's leaderboards as they stood at the 2025-07-28 snapshot. Results published since then are not among these rows, so a rank here is not a current standing.

Methods

Contrastive Learning

Report a problem or propose a change · a person checks every report against the paper or source before anything changes; decisions are listed on /corrections