Matching Tasks to Objectives: Fine-Tuning and Prompt-Tuning Strategies for Encoder-Decoder Pre-trained Language Models
About
Prompt-based learning has emerged as a dominant paradigm in natural language processing. This study explores the impact of diverse pre-training objectives on the performance of encoder-decoder pre-trained language models across generation and question answering tasks, with a focus on commonsense knowledge retrieval and completion. We highlight the benefits of incorporating multiple objectives during both pre-training and fine-tuning stages. We introduce the Match Task to Objective (MTO) framework and methods for determining the appropriate objective for a given task. This framework offers automated methods to prepare task-related data for adaptation through unsupervised training, based on the identified objective. In the fine-tuning stage, we design novel templates that align with the objectives of the pre-training and adaptation stages. When aligned with task requirements, these strategies can achieve a performance gain of over 120\% compared to conventional methods in few-shot settings. They significantly outperform related works in few-shot settings and exceed the baseline even in full-dataset scenarios. Furthermore, we extend this approach to include prompt-tuning methodologies, providing guidance for more effective soft prompt engineering and optimization. Our strategies significantly enhance prompt-tuning performance as well. These insights hold substantial value, precisely guiding the selection and optimization of models customized for specific tasks. Code is available at https://github.com/puraminy/MTO/
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Question Answering | OpenBookQA | Accuracy65.2 | 319 | |
| Question Answering | CommonsenseQA | Accuracy71.74 | 172 | |
| Commonsense Question Answering | CommonSenseQA n=425 (5%) | Accuracy61.22 | 13 | |
| Commonsense Question Answering | CommonSenseQA n=1700 (20%) | Accuracy68.12 | 13 | |
| Open-Book Question Answering | OpenBookQA 5% n=298 | Accuracy53.6 | 13 | |
| Open-Book Question Answering | OpenBookQA n=991 (20%) | Accuracy57.52 | 13 | |
| Map-Phrasal | ATOMIC selected tasks 20 | ROUGE F142.81 | 12 | |
| Commonsense Question Answering | CommonSenseQA FS n=30 | Accuracy52.5 | 6 | |
| Mask-Filling | ATOMIC 20 (selected tasks) | ROUGE19.92 | 6 | |
| Mask-Filling | ATOMIC20 few-shot n = 30 | ROUGE Score16.65 | 6 |