In plain words: Instead of tuning every added prompt word freely, this stores the prompt settings as two smaller pieces multiplied together, shrinking what must be learned. It matched full prompt tuning's results while cutting trainable parameters fivefold, beating rivals that need 10 to 20 times more.
Abstract
In prompt tuning, a prefix or suffix text is added to the prompt, and the embeddings (soft prompts) or token indices (hard prompts) of the prefix/suffix are optimized to gain more control over language models for specific tasks. This approach eliminates the need for hand-crafted prompt engineering or explicit model fine-tuning. Prompt tuning is significantly more parameter-efficient than model fine-tuning, as it involves optimizing partial inputs of language models to produce desired outputs. In this work, we aim to further reduce the amount of trainable parameters required for a language model to perform well on specific tasks. We propose Low-rank Prompt Tuning (LoPT), a low-rank model for prompts that achieves efficient prompt optimization. The proposed method demonstrates similar outcomes to full parameter prompt tuning while reducing the number of trainable parameters by a factor of 5. It also provides promising results compared to the state-of-the-art methods that would require 10 to 20 times more parameters.
Shouchang Guo, Sonam Damani, Keng-hao Chang
arXiv:2406.19486 · cs.CL, cs.AI, cs.ET, cs.LG, eess.SP · submitted Jun 27, 2024
abstract · pdf · html