ACTC: Active Threshold Calibration for Cold-Start Knowledge Graph Completion

Anastasiia Sedova, Benjamin Roth


Abstract
Self-supervised knowledge-graph completion (KGC) relies on estimating a scoring model over (entity, relation, entity)-tuples, for example, by embedding an initial knowledge graph. Prediction quality can be improved by calibrating the scoring model, typically by adjusting the prediction thresholds using manually annotated examples. In this paper, we attempt for the first time cold-start calibration for KGC, where no annotated examples exist initially for calibration, and only a limited number of tuples can be selected for annotation. Our new method ACTC finds good per-relation thresholds efficiently based on a limited set of annotated tuples. Additionally to a few annotated tuples, ACTC also leverages unlabeled tuples by estimating their correctness with Logistic Regression or Gaussian Process classifiers. We also experiment with different methods for selecting candidate tuples for annotation: density-based and random selection. Experiments with five scoring models and an oracle annotator show an improvement of 7% points when using ACTC in the challenging setting with an annotation budget of only 10 tuples, and an average improvement of 4% points over different budgets.
Anthology ID:
2023.acl-short.158
Volume:
Proceedings of the 61st Annual Meeting of the Association for Computational Linguistics (Volume 2: Short Papers)
Month:
July
Year:
2023
Address:
Toronto, Canada
Editors:
Anna Rogers, Jordan Boyd-Graber, Naoaki Okazaki
Venue:
ACL
SIG:
Publisher:
Association for Computational Linguistics
Note:
Pages:
1853–1863
Language:
URL:
https://aclanthology.org/2023.acl-short.158
DOI:
10.18653/v1/2023.acl-short.158
Bibkey:
Cite (ACL):
Anastasiia Sedova and Benjamin Roth. 2023. ACTC: Active Threshold Calibration for Cold-Start Knowledge Graph Completion. In Proceedings of the 61st Annual Meeting of the Association for Computational Linguistics (Volume 2: Short Papers), pages 1853–1863, Toronto, Canada. Association for Computational Linguistics.
Cite (Informal):
ACTC: Active Threshold Calibration for Cold-Start Knowledge Graph Completion (Sedova & Roth, ACL 2023)
Copy Citation:
PDF:
https://aclanthology.org/2023.acl-short.158.pdf
Video:
 https://aclanthology.org/2023.acl-short.158.mp4