References & Citations
Computer Science > Computation and Language
Title: Pruning Attention Heads of Transformer Models Using A* Search: A Novel Approach to Compress Big NLP Architectures
(Submitted on 28 Oct 2021 (this version), latest version 17 Nov 2021 (v3))
Abstract: Recent years have seen a growing adoption of Transformer models such as BERT in Natural Language Processing and even in Computer Vision. However, due to the size, there has been limited adoption of such models within resource-constrained computing environments This paper proposes novel pruning algorithms to compress transformer models by eliminating redundant Attention Heads. We apply the A* search algorithm to obtain a pruned model with minimal accuracy guarantees. Our results indicate that the method could eliminate as much as 40% of the attention heads in the BERT transformer model with almost no loss in accuracy.
Submission history
From: Archit Parnami [view email][v1] Thu, 28 Oct 2021 15:39:11 GMT (1150kb)
[v2] Thu, 4 Nov 2021 15:06:43 GMT (1149kb)
[v3] Wed, 17 Nov 2021 14:50:51 GMT (1149kb)
Link back to: arXiv, form interface, contact.