This repository contains the Pytorch implementation of [1](https://arxiv.org/abs/2109.03792).
Here the link to pre-processed data used for this work (i.e., training, validation and test splits of AIDA as well as the KB with the entities) and the released model.
- python>=3.8
- pytorch>=1.7
- pytorch_lightning>=1.3
- transformers>=4.0
- src: The source code of the model. In src/data there is an class of a dataset for Entity Linking. In src/model there are three classes that implement our EL model. One for the Entity Disambiuation part, one for the (autoregresive) Entity Liking part, and one for the entire model (which also contains the training and validation loops).
- notebooks: Example code for loading our Entity Linking model, evaluate it on AIDA, and run inference on a test document.
Please have a look into the notebooks folder to see hot to load our Entity Linking model, evaluate it on AIDA, and run inference on a test document.
Here a minimal example that demonstrate how to use our model:
from src.model.efficient_el import EfficientEL
from IPython.display import Markdown
from src.utils import
# loading the model on GPU and setting the the threshold to the
# optimal value (based on AIDA validation set)
model = EfficientEL.load_from_checkpoint("../models/model.ckpt").eval().cuda()
model.hparams.threshold = -3.2
# loading the KB with the entities
model.generate_global_trie()
# document which we want to apply EL on
s = """CRICKET - LEICESTERSHIRE TAKE OVER AT TOP AFTER INNINGS VICTORY . LONDON 1996-08-30 \
West Indian all-rounder Phil Simmons took four for 38 on Friday as Leicestershire beat Somerset \
by an innings and 39 runs in two days to take over at the head of the county championship ."""
# getting spans from the model and converting the result into Markdown for visualization
Markdown(
get_markdown(
[s],
[[(s[0], s[1], s[2][0][0]) for s in spans]
for spans in model.sample([s])]
)[0]
)CRICKET - LEICESTERSHIRE TAKE OVER AT TOP AFTER INNINGS VICTORY . LONDON 1996-08-30 West Indian all-rounder Phil Simmons took four for 38 on Friday as Leicestershire beat Somerset by an innings and 39 runs in two days to take over at the head of the county championship .
Please cite [1] in your work when using this library in your experiments.
For questions and comments, feel free to contact Nicola De Cao.
MIT
[1] De Cao, N. et al. (2020).
Highly Parallel Autoregressive Entity Linking with Discriminative Correction.
Proceedings of the 2020 Conference on Empirical Methods in Natural
Language Processing (EMNLP2021).
BibTeX format:
@article{decao2021parallel,
title={{Highly Parallel Autoregressive Entity Linking with Discriminative Correction}},
author={
De Cao, Nicola and
Aziz, Wilker and
Titov, Ivan},
journal={Proceedings of the 2020 Conference on Empirical Methods in
Natural Language Processing (EMNLP2021)},
url={https://arxiv.org/abs/2109.03792},
year={2021}
}