English

ConvFiT: Conversational Fine-Tuning of Pretrained Language Models

Computation and Language 2021-09-22 v1

Abstract

Transformer-based language models (LMs) pretrained on large text collections are proven to store a wealth of semantic knowledge. However, 1) they are not effective as sentence encoders when used off-the-shelf, and 2) thus typically lag behind conversationally pretrained (e.g., via response selection) encoders on conversational tasks such as intent detection (ID). In this work, we propose ConvFiT, a simple and efficient two-stage procedure which turns any pretrained LM into a universal conversational encoder (after Stage 1 ConvFiT-ing) and task-specialised sentence encoder (after Stage 2). We demonstrate that 1) full-blown conversational pretraining is not required, and that LMs can be quickly transformed into effective conversational encoders with much smaller amounts of unannotated data; 2) pretrained LMs can be fine-tuned into task-specialised sentence encoders, optimised for the fine-grained semantics of a particular task. Consequently, such specialised sentence encoders allow for treating ID as a simple semantic similarity task based on interpretable nearest neighbours retrieval. We validate the robustness and versatility of the ConvFiT framework with such similarity-based inference on the standard ID evaluation sets: ConvFiT-ed LMs achieve state-of-the-art ID performance across the board, with particular gains in the most challenging, few-shot setups.

Keywords

Cite

@article{arxiv.2109.10126,
  title  = {ConvFiT: Conversational Fine-Tuning of Pretrained Language Models},
  author = {Ivan Vulić and Pei-Hao Su and Sam Coope and Daniela Gerz and Paweł Budzianowski and Iñigo Casanueva and Nikola Mrkšić and Tsung-Hsien Wen},
  journal= {arXiv preprint arXiv:2109.10126},
  year   = {2021}
}

Comments

EMNLP 2021 (long paper)