Dynamic Entity Representations in Neural Language Models

Yangfeng Ji, Chenhao Tan, Sebastian Martschat, Yejin Choi, Noah A. Smith
In Proceedings of EMNLP 2017

Understanding a long document requires tracking how entities are introduced and evolve over time. We present a new type of language model, ENTITYNLM, that can explicitly model entities, dynamically update their representations, and contextually generate their mentions. Our model is generative and flexible; it can model an arbitrary number of entities in context while generating each entity mention at an arbitrary length. In addition, it can be used for several different tasks such as language modeling, coreference resolution, and entity prediction. Experimental results with all these tasks demonstrate that our model consistently outperforms strong baselines and prior work.

[PDF] [Code]

     author = {Chenhao Tan and Dallas Card and Noah A. Smith},
     title = {Friendships, Rivalries, and Trysts: Characterizing Relations between Ideas in Texts},
     year = {2017},
     booktitle = {Proceedings of ACL}