TBERT: Dynamic BERT Inference with Top-k Based Predictors
文献类型:会议论文
作者 | Liu, Zejian2,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.National Laboratory of Pattern Recognition, Institute of Automation, Chinese Academy of Sciences 3.School of Future Technology, University of 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. |
入库方式: OAI收割
来源:自动化研究所
其他版本
除非特别说明,本系统中所有内容都受版权保护,并保留所有权利。