login

Transformer to CNN: Label-scarce distillation for efficient text classification

arXiv (Cornell University)Published 8 September 2019Open access
Yew Ken Chia, Sam Witteveen, Martin Andrews
Citations21
View PDF

TL;DR

This workshop paper outlines how the proposed convolutional student architecture, having been trained by a distillation process from a large-scale model, can achieve 300x inference speedup and 39x reduction in parameter count.

Abstract

Significant advances have been made in Natural Language Processing (NLP) modelling since the beginning of 2018. The new approaches allow for accurate results, even when there is little labelled data, because these NLP models can benefit from training on both task-agnostic and task-specific unlabelled data. However, these advantages come with significant size and computational costs. This workshop paper outlines how our proposed convolutional student architecture, having been trained by a distillation process from a large-scale model, can achieve 300x inference speedup and 39x reduction in parameter count. In some cases, the student model performance surpasses its teacher on the studied tasks.

Keywords

Computer Science