Don’t Mess with Mister-in-Between: Improved Negative Search for Knowledge Graph Completion

Fan Jiang, Tom Drummond, Trevor Cohn


Abstract
The best methods for knowledge graph completion use a ‘dual-encoding’ framework, a form of neural model with a bottleneck that facilitates fast approximate search over a vast collection of candidates. These approaches are trained using contrastive learning to differentiate between known positive examples and sampled negative instances. The mechanism for sampling negatives to date has been very simple, driven by pragmatic engineering considerations (e.g., using mismatched instances from the same batch). We propose several novel means of finding more informative negatives, based on searching for candidates with high lexical overlaps, from the dual-encoder model and according to knowledge graph structures. Experimental results on four benchmarks show that our best single model improves consistently over previous methods and obtains new state-of-the-art performance, including the challenging large-scale Wikidata5M dataset. Combing different kinds of strategies through model ensembling results in a further performance boost.
Anthology ID:
2023.eacl-main.133
Volume:
Proceedings of the 17th Conference of the European Chapter of the Association for Computational Linguistics
Month:
May
Year:
2023
Address:
Dubrovnik, Croatia
Editors:
Andreas Vlachos, Isabelle Augenstein
Venue:
EACL
SIG:
Publisher:
Association for Computational Linguistics
Note:
Pages:
1818–1832
Language:
URL:
https://aclanthology.org/2023.eacl-main.133
DOI:
10.18653/v1/2023.eacl-main.133
Bibkey:
Cite (ACL):
Fan Jiang, Tom Drummond, and Trevor Cohn. 2023. Don’t Mess with Mister-in-Between: Improved Negative Search for Knowledge Graph Completion. In Proceedings of the 17th Conference of the European Chapter of the Association for Computational Linguistics, pages 1818–1832, Dubrovnik, Croatia. Association for Computational Linguistics.
Cite (Informal):
Don’t Mess with Mister-in-Between: Improved Negative Search for Knowledge Graph Completion (Jiang et al., EACL 2023)
Copy Citation:
PDF:
https://aclanthology.org/2023.eacl-main.133.pdf
Video:
 https://aclanthology.org/2023.eacl-main.133.mp4