EELBERT: Tiny Models through Dynamic Embeddings

Gabrielle Cohn, Rishika Agarwal, Deepanshu Gupta, Siddharth Patwardhan


Abstract
We introduce EELBERT, an approach for compression of transformer-based models (e.g., BERT), with minimal impact on the accuracy of downstream tasks. This is achieved by replacing the input embedding layer of the model with dynamic, i.e. on-the-fly, embedding computations. Since the input embedding layer occupies a large portion of the model size, especially for the smaller BERT variants, replacing this layer with an embedding computation function helps us reduce the model size significantly. Empirical evaluation on the GLUE benchmark shows that our BERT variants (EELBERT) suffer minimal regression compared to the traditional BERT models. Through this approach, we are able to develop our smallest model UNO-EELBERT, which achieves a GLUE score within 4% of fully trained BERT-tiny, while being 15x smaller (1.2 MB) in size.
Anthology ID:
2023.emnlp-industry.43
Volume:
Proceedings of the 2023 Conference on Empirical Methods in Natural Language Processing: Industry Track
Month:
December
Year:
2023
Address:
Singapore
Editors:
Mingxuan Wang, Imed Zitouni
Venue:
EMNLP
SIG:
Publisher:
Association for Computational Linguistics
Note:
Pages:
451–459
Language:
URL:
https://aclanthology.org/2023.emnlp-industry.43
DOI:
10.18653/v1/2023.emnlp-industry.43
Bibkey:
Cite (ACL):
Gabrielle Cohn, Rishika Agarwal, Deepanshu Gupta, and Siddharth Patwardhan. 2023. EELBERT: Tiny Models through Dynamic Embeddings. In Proceedings of the 2023 Conference on Empirical Methods in Natural Language Processing: Industry Track, pages 451–459, Singapore. Association for Computational Linguistics.
Cite (Informal):
EELBERT: Tiny Models through Dynamic Embeddings (Cohn et al., EMNLP 2023)
Copy Citation:
PDF:
https://aclanthology.org/2023.emnlp-industry.43.pdf
Video:
 https://aclanthology.org/2023.emnlp-industry.43.mp4