On Conditional and Compositional Language Model Differentiable Prompting

Jonathan Pilault, Can Liu, Mohit Bansal, Markus Dreyer

Proceedings of the Thirty-Second International Joint Conference on Artificial Intelligence

Main Track. Pages 4136-4144.

https://doi.org/10.24963/ijcai.2023/460

PDF BibTeX

Abstract

Prompts have been shown to be an effective method to adapt a frozen Pretrained Language Model (PLM) to perform well on downstream tasks. Prompts can be represented by a human-engineered word sequence or by a learned continuous embedding.

In this work, we investigate conditional and compositional differentiable prompting.

We propose a new model, Prompt Production System (ProPS), which learns to transform task instructions or input metadata, into continuous prompts that elicit task-specific outputs from the PLM.

Our model uses a modular network structure based on our neural formulation of Production Systems, which allows the model to learn discrete rules -- neural functions that learn to specialize in transforming particular prompt input patterns, making it suitable for compositional transfer learning and few-shot learning.

We present extensive empirical and theoretical analysis and show that ProPS consistently surpasses other PLM adaptation techniques, and often improves upon fully fine-tuned models, on compositional generalization tasks, controllable summarization and multilingual translation, while needing fewer trainable parameters.

Keywords

  • Machine Learning: ML: Multi-task and transfer learning
  • Machine Learning: ML: Neuro-symbolic methods
Welcome Future Conferences AI Journal Division IJCAI Proceedings Travel Grant Program IJCAI Awards