资源论文PROBABILITY CALIBRATION FOR KNOWLEDGEG RAPH EMBEDDING MODELS

PROBABILITY CALIBRATION FOR KNOWLEDGEG RAPH EMBEDDING MODELS

2020-01-02 | |  67 |   38 |   0

Abstract
Knowledge graph embedding research has overlooked the problem of probability calibration. We show popular embedding models are indeed uncalibrated. That means probability estimates associated to predicted triples are unreliable. We present a novel method to calibrate a model when ground truth negatives are not available, which is the usual case in knowledge graphs. We propose to use Platt scaling and isotonic regression alongside our method. Experiments on three datasets with ground truth negatives show our contribution leads to well calibrated models when compared to the gold standard of using negatives. We get significantly better results than the uncalibrated models from all calibration methods. We show isotonic regression offers the best the performance overall, not without trade-offs. We also show that calibrated models reach state-of-the-art accuracy without the need to define relation-specific decision thresholds.

上一篇:TENSOR DECOMPOSITIONS FORT EMPORAL KNOWLEDGE BASE COMPLETION

下一篇:LOW-R ESOURCE KNOWLEDGE -G ROUNDEDD IALOGUE GENERATION

用户评价
全部评价

热门资源

  • Learning to Predi...

    Much of model-based reinforcement learning invo...

  • Stratified Strate...

    In this paper we introduce Stratified Strategy ...

  • The Variational S...

    Unlike traditional images which do not offer in...

  • A Mathematical Mo...

    Direct democracy, where each voter casts one vo...

  • Rating-Boosted La...

    The performance of a recommendation system reli...