LightPAFF: A Two-Stage Distillation Framework for Pre-training and
  Fine-tuning
  While pre-training and fine-tuning, e.g., BERT~\citep{devlin2018bert},
GPT-2~\citep{radford2019language}, have achieved great success in language
understanding and generation tasks, the pre-trained models are usually too big
for online deployment in terms of both memory cost and inference speed, which
hinders them from practical online usage. In this paper, we propose LightPAFF,
a Lightweight Pre-training And Fine-tuning Framework that leverages two-stage
knowledge distillation to transfer knowledge from a big teacher model to a
lightweight student model in both pre-training and fine-tuning stages. In this
way the lightweight model can achieve similar accuracy as the big teacher
model, but with much fewer parameters and thus faster online inference speed.
LightPAFF can support different pre-training methods (such as BERT, GPT-2 and
MASS~\citep{song2019mass}) and be applied to many downstream tasks. Experiments
on three language understanding tasks, three language modeling tasks and three
sequence to sequence generation tasks demonstrate that while achieving similar
accuracy with the big BERT, GPT-2 and MASS models, LightPAFF reduces the model
size by nearly 5x and improves online inference speed by 5x-7x.
