Can discrete information extraction prompts generalize across language models?

Nathanaël Carraz Rakotonirina, Roberto Dessì, Fabio Petroni, Sebastian Riedel, Marco Baroni
2/20/2023
cs.CLcs.AIcs.LG

No official code found. Best implementation: ncarraz/prompt_generalization (7 stars, MIT, last commit 2024-03-04).

Abstract

We study whether automatically-induced prompts that effectively extract information from a language model can also be used, out-of-the-box, to probe other language models for the same information. After confirming that discrete prompts induced with the AutoPrompt algorithm outperform manual and semi-manual prompts on the slot-filling task, we demonstrate a drop in performance for AutoPrompt prompts learned on a model and tested on another. We introduce a way to induce prompts by mixing language models at training time that results in prompts that generalize well across models. We conduct an extensive analysis of the induced prompts, finding that the more general prompts include a larger proportion of existing English words and have a less order-dependent and more uniform distribution of information across their component tokens. Our work provides preliminary evidence that it's possible to generate discrete prompts that can be induced once and used with a number of different models, and gives insights on the properties characterizing such prompts.

View on arXivView PDF

Code Implementations(1)

MIT

ICLR 2023: Can discrete information extraction prompts generalize across language models?

70Feb 8, 20232 years agoMIT

Cite this paper

@article{rakotonirina2023discrete,
  title  = {Can discrete information extraction prompts generalize across language models?},
  author = {Nathanaël Carraz Rakotonirina and Roberto Dessì and Fabio Petroni and Sebastian Riedel and Marco Baroni},
  year   = {2023},
  eprint = {2302.09865},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2302.09865v2},
  journal = {ICLR 2023 2023}
}

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/2302.09865.svg)](https://codeofpaper.com/paper/2302.09865)

Discussion