arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2004.12993cs.CLcs.LG

DeeBERT: Dynamic Early Exiting for Accelerating BERT Inference

University of Waterloo(滑铁卢大学);Vector Institute for Artificial Intelligence(向量人工智能研究所)

Ji Xin, Raphael Tang, Jaejun Lee, Yaoliang Yu, Jimmy Lin

英文摘要:

Large-scale pre-trained language models such as BERT have brought significant improvements to NLP applications. However, they are also notorious for being slow in inference, which makes them difficult to deploy in real-time applications. We propose a simple but effective method, DeeBERT, to accelerate BERT inference. Our approach allows samples to exit earlier without passing through the entire model. Experiments show that DeeBERT is able to save up to ~40% inference time with minimal degradation in model quality. Further analyses show different behaviors in the BERT transformer layers and also reveal their redundancy. Our work provides new ideas to efficiently apply deep transformer-based models to downstream tasks. Code is available at https://github.com/castorini/DeeBERT.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑