Edit model card

LUKE: Deep Contextualized Entity Representations with Entity-aware Self-attention

LUKE (Language Understanding with Knowledge-based Embeddings) is a new pre-trained contextualized representation of words and entities based on transformer. LUKE treats words and entities in a given text as independent tokens, and outputs contextualized representations of them. LUKE adopts an entity-aware self-attention mechanism that is an extension of the self-attention mechanism of the transformer, and considers the types of tokens (words or entities) when computing attention scores.

LUKE achieves state-of-the-art results on five popular NLP benchmarks including SQuAD v1.1 (extractive question answering), CoNLL-2003 (named entity recognition), ReCoRD (cloze-style question answering), TACRED (relation classification), and Open Entity (entity typing).

Please check the official repository for more details and updates.

This is the LUKE base model with 12 hidden layers, 768 hidden size. The total number of parameters in this model is 253M. It is trained using December 2018 version of Wikipedia.

Experimental results

The experimental results are provided as follows:

Task Dataset Metric LUKE-large luke-base Previous SOTA
Extractive Question Answering SQuAD v1.1 EM/F1 90.2/95.4 86.1/92.3 89.9/95.1 (Yang et al., 2019)
Named Entity Recognition CoNLL-2003 F1 94.3 93.3 93.5 (Baevski et al., 2019)
Cloze-style Question Answering ReCoRD EM/F1 90.6/91.2 - 83.1/83.7 (Li et al., 2019)
Relation Classification TACRED F1 72.7 - 72.0 (Wang et al. , 2020)
Fine-grained Entity Typing Open Entity F1 78.2 - 77.6 (Wang et al. , 2020)


If you find LUKE useful for your work, please cite the following paper:

  title={LUKE: Deep Contextualized Entity Representations with Entity-aware Self-attention},
  author={Ikuya Yamada and Akari Asai and Hiroyuki Shindo and Hideaki Takeda and Yuji Matsumoto},
Downloads last month

Space using studio-ousia/luke-base 1