papers AI Learner
The Github is limit! Click to go to the new site.

Model Compression with Multi-Task Knowledge Distillation for Web-scale Question Answering System

2019-04-21
Ze Yang, Linjun Shou, Ming Gong, Wutao Lin, Daxin Jiang

Abstract

Deep pre-training and fine-tuning models (like BERT, OpenAI GPT) have demonstrated excellent results in question answering areas. However, due to the sheer amount of model parameters, the inference speed of these models is very slow. How to apply these complex models to real business scenarios becomes a challenging but practical problem. Previous works often leverage model compression approaches to resolve this problem. However, these methods usually induce information loss during the model compression procedure, leading to incomparable results between compressed model and the original model. To tackle this challenge, we propose a Multi-task Knowledge Distillation Model (MKDM for short) for web-scale Question Answering system, by distilling knowledge from multiple teacher models to a light-weight student model. In this way, more generalized knowledge can be transferred. The experiment results show that our method can significantly outperform the baseline methods and even achieve comparable results with the original teacher models, along with significant speedup of model inference.

Abstract (translated by Google)
URL

http://arxiv.org/abs/1904.09636

PDF

http://arxiv.org/pdf/1904.09636


Comments

Content