Treat Different Negatives Differently: Enriching Loss Functions with Domain and Range Constraints for Link Prediction

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Hubert, Nicolas, Monnin, Pierre, Brun, Armelle, Monticolo, Davy
Formato: Preprint
Publicado: 2023
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866910354480562176
author Hubert, Nicolas
Monnin, Pierre
Brun, Armelle
Monticolo, Davy
author_facet Hubert, Nicolas
Monnin, Pierre
Brun, Armelle
Monticolo, Davy
contents Knowledge graph embedding models (KGEMs) are used for various tasks related to knowledge graphs (KGs), including link prediction. They are trained with loss functions that consider batches of true and false triples. However, different kinds of false triples exist and recent works suggest that they should not be valued equally, leading to specific negative sampling procedures. In line with this recent assumption, we posit that negative triples that are semantically valid w.r.t. signatures of relations (domain and range) are high-quality negatives. Hence, we enrich the three main loss functions for link prediction such that all kinds of negatives are sampled but treated differently based on their semantic validity. In an extensive and controlled experimental setting, we show that the proposed loss functions systematically provide satisfying results which demonstrates both the generality and superiority of our proposed approach. In fact, the proposed loss functions (1) lead to better MRR and Hits@10 values, and (2) drive KGEMs towards better semantic correctness as measured by the Sem@K metric. This highlights that relation signatures globally improve KGEMs, and thus should be incorporated into loss functions. Domains and ranges of relations being largely available in schema-defined KGs, this makes our approach both beneficial and widely usable in practice.
format Preprint
id arxiv_https___arxiv_org_abs_2303_00286
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Treat Different Negatives Differently: Enriching Loss Functions with Domain and Range Constraints for Link Prediction
Hubert, Nicolas
Monnin, Pierre
Brun, Armelle
Monticolo, Davy
Machine Learning
Artificial Intelligence
Knowledge graph embedding models (KGEMs) are used for various tasks related to knowledge graphs (KGs), including link prediction. They are trained with loss functions that consider batches of true and false triples. However, different kinds of false triples exist and recent works suggest that they should not be valued equally, leading to specific negative sampling procedures. In line with this recent assumption, we posit that negative triples that are semantically valid w.r.t. signatures of relations (domain and range) are high-quality negatives. Hence, we enrich the three main loss functions for link prediction such that all kinds of negatives are sampled but treated differently based on their semantic validity. In an extensive and controlled experimental setting, we show that the proposed loss functions systematically provide satisfying results which demonstrates both the generality and superiority of our proposed approach. In fact, the proposed loss functions (1) lead to better MRR and Hits@10 values, and (2) drive KGEMs towards better semantic correctness as measured by the Sem@K metric. This highlights that relation signatures globally improve KGEMs, and thus should be incorporated into loss functions. Domains and ranges of relations being largely available in schema-defined KGs, this makes our approach both beneficial and widely usable in practice.
title Treat Different Negatives Differently: Enriching Loss Functions with Domain and Range Constraints for Link Prediction
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2303.00286