← 返回资源分享
Prefix-Tuning: Optimizing Continuous Prompts for Generation
论文
论文
发布时间2021-01-01
发表ACL 2021 5 · arXiv:2101.00190
作者:Percy Liang,Xiang Lisa Li
详细介绍
Fine-tuning is the de facto way to leverage large pretrained language models to perform downstream tasks. However, it modifies all the language model parameters and therefore necessitates storing a full copy for each task. In this paper, we propose prefix-tuning, a lightweight alternative to fine-tuning for natural language generation tasks, which keeps language model parameters frozen, but optimizes a small continuous task-specific vector (called the prefix). Prefix-tuning draws inspiration from prompting, allowing subsequent tokens to attend to this prefix as if it were "virtual tokens". We apply prefix-tuning to GPT-2 for table-to-text generation and to BART for summarization. We find that by learning only 0.1\% of the parameters, prefix-tuning obtains comparable performance in the full data setting, outperforms fine-tuning in low-data settings, and extrapolates better to examples with topics unseen during training.
代码仓库 (14)
ga642381/SpeechPrompt官方PyTorch
XiangLi1999/PrefixTuning官方PyTorch
Zeng-WH/PrefixTuning-FixPyTorch
DKhomi/PrefixTuning
rmokady/clip_prefix_captionPyTorch
jordiclive/ControlPrefixesPyTorch
rucaibox/mvp
NVIDIA/FasterTransformerPyTorch
lostoxygen/llm-confidentialityPyTorch
hellokevin07/elastictrainerTensorFlow
