English

Prototypical Verbalizer for Prompt-based Few-shot Tuning

Computation and Language 2022-03-21 v1 Machine Learning

Abstract

Prompt-based tuning for pre-trained language models (PLMs) has shown its effectiveness in few-shot learning. Typically, prompt-based tuning wraps the input text into a cloze question. To make predictions, the model maps the output words to labels via a verbalizer, which is either manually designed or automatically built. However, manual verbalizers heavily depend on domain-specific prior knowledge and human efforts, while finding appropriate label words automatically still remains challenging.In this work, we propose the prototypical verbalizer (ProtoVerb) which is built directly from training data. Specifically, ProtoVerb learns prototype vectors as verbalizers by contrastive learning. In this way, the prototypes summarize training instances and are able to enclose rich class-level semantics. We conduct experiments on both topic classification and entity typing tasks, and the results demonstrate that ProtoVerb significantly outperforms current automatic verbalizers, especially when training data is extremely scarce. More surprisingly, ProtoVerb consistently boosts prompt-based tuning even on untuned PLMs, indicating an elegant non-tuning way to utilize PLMs. Our codes are avaliable at https://github.com/thunlp/OpenPrompt.

Keywords

Cite

@article{arxiv.2203.09770,
  title  = {Prototypical Verbalizer for Prompt-based Few-shot Tuning},
  author = {Ganqu Cui and Shengding Hu and Ning Ding and Longtao Huang and Zhiyuan Liu},
  journal= {arXiv preprint arXiv:2203.09770},
  year   = {2022}
}

Comments

11 pages. ACL 2022 main conference

R2 v1 2026-06-24T10:18:00.996Z