English

To Tune or Not to Tune? Adapting Pretrained Representations to Diverse Tasks

Computation and Language 2019-06-12 v2 Machine Learning

Abstract

While most previous work has focused on different pretraining objectives and architectures for transfer learning, we ask how to best adapt the pretrained model to a given target task. We focus on the two most common forms of adaptation, feature extraction (where the pretrained weights are frozen), and directly fine-tuning the pretrained model. Our empirical results across diverse NLP tasks with two state-of-the-art models show that the relative performance of fine-tuning vs. feature extraction depends on the similarity of the pretraining and target tasks. We explore possible explanations for this finding and provide a set of adaptation guidelines for the NLP practitioner.

Keywords

Cite

@article{arxiv.1903.05987,
  title  = {To Tune or Not to Tune? Adapting Pretrained Representations to Diverse Tasks},
  author = {Matthew E. Peters and Sebastian Ruder and Noah A. Smith},
  journal= {arXiv preprint arXiv:1903.05987},
  year   = {2019}
}

Comments

Proceedings of the 4th Workshop on Representation Learning for NLP