Differentiable Prompt Makes Pre-trained Language Models Better Few-shot Learners

Ningyu Zhang, Luoqiu Li, Xiang Chen, Shumin Deng, Zhen Bi +3 more
8/30/2021
cs.CLcs.AIcs.CV

Official code: zjunlp/DART (131 stars, MIT, PyTorch, last commit 2022-12-07).

Abstract

Large-scale pre-trained language models have contributed significantly to natural language processing by demonstrating remarkable abilities as few-shot learners. However, their effectiveness depends mainly on scaling the model parameters and prompt design, hindering their implementation in most real-world applications. This study proposes a novel pluggable, extensible, and efficient approach named DifferentiAble pRompT (DART), which can convert small language models into better few-shot learners without any prompt engineering. The main principle behind this approach involves reformulating potential natural language processing tasks into the task of a pre-trained language model and differentially optimizing the prompt template as well as the target label with backpropagation. Furthermore, the proposed approach can be: (i) Plugged to any pre-trained language models; (ii) Extended to widespread classification tasks. A comprehensive evaluation of standard NLP tasks demonstrates that the proposed approach achieves a better few-shot performance. Code is available in https://github.com/zjunlp/DART.

View on arXivView PDF

Code Implementations(1)

zjunlp/DARTOfficial100%
PyTorchMIT

[ICLR 2022] Differentiable Prompt Makes Pre-trained Language Models Better Few-shot Learners

13117PythonDec 18, 20213 years agoMIT
dartfew-shot-learningiclriclr2022language-models+5 more

Similar Papers

Cite this paper

@article{zhang2021differentiable,
  title  = {Differentiable Prompt Makes Pre-trained Language Models Better Few-shot Learners},
  author = {Ningyu Zhang and Luoqiu Li and Xiang Chen and Shumin Deng and Zhen Bi and Chuanqi Tan and Fei Huang and Huajun Chen},
  year   = {2021},
  eprint = {2108.13161},
  archivePrefix = {arXiv},
  url    = {https://arxiv.org/abs/2108.13161}
}

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

Discussion