Mikhail Galkin, Priyansh Trivedi, Gaurav Maheshwari, Ricardo Usbeck, Jens Lehmann
Hyper-relational knowledge graphs (KGs) (e.g., Wikidata) enable associating additional key-value pairs along with the main triple to disambiguate, or restrict the validity of a fact. In this work, we propose a message passing based graph encoder - StarE capable of modeling such hyper-relational KGs. Unlike existing approaches, StarE can encode an arbitrary number of additional information (qualifiers) along with the main triple while keeping the semantic roles of qualifiers and triples intact. We also demonstrate that existing benchmarks for evaluating link prediction (LP) performance on hyper-relational KGs suffer from fundamental flaws and thus develop a new Wikidata-based dataset - WD50K. Our experiments demonstrate that StarE based LP model outperforms existing approaches across multiple benchmarks. We also confirm that leveraging qualifiers is vital for link prediction with gains up to 25 MRR points compared to triple-based representations.
| Task | Dataset | Metric | Value | Model |
|---|---|---|---|---|
| Link Prediction | JF17K | Hit@1 | 0.496 | StarE (H) + Transformer (H) |
| Link Prediction | JF17K | Hit@10 | 0.725 | StarE (H) + Transformer (H) |
| Link Prediction | JF17K | Hit@5 | 0.658 | StarE (H) + Transformer (H) |
| Link Prediction | JF17K | MRR | 0.574 | StarE (H) + Transformer (H) |
| Link Prediction | Temp8 | Hit@1 | 0.271 | StarE (H) + Transformer (H) |
| Link Prediction | Temp8 | Hit@10 | 0.496 | StarE (H) + Transformer (H) |
| Link Prediction | Temp8 | MRR | 0.349 | StarE (H) + Transformer (H) |