Control Prefixes for Parameter-Efficient Text Generation
This work addresses the problem of parameter-efficient and controlled text generation for natural language processing applications, offering an incremental improvement over existing prefix-tuning methods.
The paper tackles the limitation of prefix-tuning by introducing Control Prefixes, a dynamic method that incorporates conditional input-dependent information to guide text generation, achieving state-of-the-art results on datasets like WebNLG and outperforming full fine-tuning in some cases.
Prefix-tuning is a powerful lightweight technique for adapting a large pre-trained language model to a downstream application. However, it uses the same dataset-level tuned prompt for all examples in the dataset. We extend this idea and propose a dynamic method, Control Prefixes, which allows for the inclusion of conditional input-dependent information, combining the benefits of prompt tuning and controlled generation. The method incorporates attribute-level learnable representations into different layers of a pre-trained transformer, allowing for the generated text to be guided in a particular direction. We provide a systematic evaluation of the technique and apply it to five datasets from the GEM benchmark for natural language generation (NLG). Although the aim is to develop a parameter-efficient model, we show Control Prefixes can even outperform full fine-tuning methods. We present state-of-the-art results on several data-to-text datasets, including WebNLG.