English

IDPG: An Instance-Dependent Prompt Generation Method

Computation and Language 2022-04-12 v1 Machine Learning

Abstract

Prompt tuning is a new, efficient NLP transfer learning paradigm that adds a task-specific prompt in each input instance during the model training stage. It freezes the pre-trained language model and only optimizes a few task-specific prompts. In this paper, we propose a conditional prompt generation method to generate prompts for each input instance, referred to as the Instance-Dependent Prompt Generation (IDPG). Unlike traditional prompt tuning methods that use a fixed prompt, IDPG introduces a lightweight and trainable component to generate prompts based on each input sentence. Extensive experiments on ten natural language understanding (NLU) tasks show that the proposed strategy consistently outperforms various prompt tuning baselines and is on par with other efficient transfer learning methods such as Compacter while tuning far fewer model parameters.

Keywords

Cite

@article{arxiv.2204.04497,
  title  = {IDPG: An Instance-Dependent Prompt Generation Method},
  author = {Zhuofeng Wu and Sinong Wang and Jiatao Gu and Rui Hou and Yuxiao Dong and V. G. Vinod Vydiswaran and Hao Ma},
  journal= {arXiv preprint arXiv:2204.04497},
  year   = {2022}
}

Comments

To appear at the NAACL 2022 main conference

R2 v1 2026-06-24T10:43:17.502Z