Can LMs Learn New Entities from Descriptions? Challenges in Propagating Injected Knowledge

Yasumasa Onoe, Michael J. Q. Zhang, Shankar Padmanabhan, Greg Durrett, Eunsol Choi
5/2/2023
cs.CL

No official code found. Best implementation: yasumasaonoe/entity_knowledge_propagation (17 stars, last commit 2023-08-02).

Abstract

Pre-trained language models (LMs) are used for knowledge intensive tasks like question answering, but their knowledge gets continuously outdated as the world changes. Prior work has studied targeted updates to LMs, injecting individual facts and evaluating whether the model learns these facts while not changing predictions on other contexts. We take a step forward and study LMs' abilities to make inferences based on injected facts (or propagate those facts): for example, after learning that something is a TV show, does an LM predict that you can watch it? We study this with two cloze-style tasks: an existing dataset of real-world sentences about novel entities (ECBD) as well as a new controlled benchmark with manually designed templates requiring varying levels of inference about injected knowledge. Surprisingly, we find that existing methods for updating knowledge (gradient-based fine-tuning and modifications of this approach) show little propagation of injected knowledge. These methods improve performance on cloze instances only when there is lexical overlap between injected facts and target inferences. Yet, prepending entity definitions in an LM's context improves performance across all settings, suggesting that there is substantial headroom for parameter-updating approaches for knowledge injection.

View on arXivView PDF

Code Implementations(1)

172Shell, PythonMay 1, 20233 years ago

Similar Papers

Cite this paper

@article{onoe2023learn,
  title  = {Can LMs Learn New Entities from Descriptions? Challenges in Propagating Injected Knowledge},
  author = {Yasumasa Onoe and Michael J. Q. Zhang and Shankar Padmanabhan and Greg Durrett and Eunsol Choi},
  year   = {2023},
  eprint = {2305.01651},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2305.01651v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2305.01651.svg)](https://codeofpaper.com/paper/2305.01651)

Discussion