HyperAIHyperAI

Command Palette

Search for a command to run...

5 months ago

SpEL: Structured Prediction for Entity Linking

Hassan S. Shavarani; Anoop Sarkar

SpEL: Structured Prediction for Entity Linking

Abstract

Entity linking is a prominent thread of research focused on structured data creation by linking spans of text to an ontology or knowledge source. We revisit the use of structured prediction for entity linking which classifies each individual input token as an entity, and aggregates the token predictions. Our system, called SpEL (Structured prediction for Entity Linking) is a state-of-the-art entity linking system that uses some new ideas to apply structured prediction to the task of entity linking including: two refined fine-tuning steps; a context sensitive prediction aggregation strategy; reduction of the size of the model's output vocabulary, and; we address a common problem in entity-linking systems where there is a training vs. inference tokenization mismatch. Our experiments show that we can outperform the state-of-the-art on the commonly used AIDA benchmark dataset for entity linking to Wikipedia. Our method is also very compute efficient in terms of number of parameters and speed of inference.

Code Repositories

shavarani/spel
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
entity-linking-on-aida-conllSpEL-base (2023)
Micro-F1 strong: 88.1
entity-linking-on-aida-conllSpEL-large (2023)
Micro-F1 strong: 88.6
entity-linking-on-aida-testcSpEL-large (2023)
Micro-F1 strong: 77.5
entity-linking-on-aida-testcSpEL-base (2023)
Micro-F1 strong: 73.7

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
SpEL: Structured Prediction for Entity Linking | Papers | HyperAI