TBERT: Dynamic BERT Inference with Top-k Based Predictors
Liu, Zejian2,3; Zhao, Kun3; Cheng, Jian1,2,3
2023-04
会议日期2023-4-17
会议地点Antwerp, Belgium
关键词Transformer Dynamic Inference Pruning
英文摘要

Dynamic inference is a compression method that adaptively prunes unimportant components according to the input at the inference stage, which can achieve a better tradeoff between computational complexity and model accuracy than static compression methods. However, there are two limitations in previous works. The first one is that they usually need to search the threshold on the evaluation dataset to achieve the target compression ratio, but the search process is non-trivial. The second one is that these methods are unstable. Their performance will be significantly degraded on some datasets, especially when the compression ratio is high. In this paper, we propose TBERT, a simple yet stable dynamic inference method. TBERT utilizes the top-k-based pruning strategy which allows accurate control of the compression ratio. To enable stable end-to-end training of the model, we carefully design the structure of the predictor. Moreover, we propose adding auxiliary classifiers to help the model’s training. Experimental results on the GLUE benchmark demonstrate that our method achieves higher performance than previous state-of-the-art methods.

语种英语
内容类型会议论文
源URL[http://ir.ia.ac.cn/handle/173211/52036]  
专题类脑芯片与系统研究
通讯作者Cheng, Jian
作者单位1.AiRiA
2.School of Future Technology, University of Chinese Academy of Sciences
3.National Laboratory of Pattern Recognition, Institute of Automation, Chinese Academy of Sciences
推荐引用方式
GB/T 7714
Liu, Zejian,Zhao, Kun,Cheng, Jian. TBERT: Dynamic BERT Inference with Top-k Based Predictors[C]. 见:. Antwerp, Belgium. 2023-4-17.
个性服务
查看访问统计
相关权益政策
暂无数据
收藏/分享
所有评论 (0)
暂无评论
 

除非特别说明,本系统中所有内容都受版权保护,并保留所有权利。


©版权所有 ©2017 CSpace - Powered by CSpace