English

Block-wise Bit-Compression of Transformer-based Models

Computation and Language 2023-04-04 v2 Machine Learning

Abstract

With the popularity of the recent Transformer-based models represented by BERT, GPT-3 and ChatGPT, there has been state-of-the-art performance in a range of natural language processing tasks. However, the massive computations, huge memory footprint, and thus high latency of Transformer-based models is an inevitable challenge for the cloud with high real-time requirement. To tackle the issue, we propose BBCT, a method of block-wise bit-compression for transformer without retraining. Our method achieves more fine-grained compression of the whole transformer, including embedding, matrix multiplication, GELU, softmax, layer normalization, and all the intermediate results. As a case, we compress an efficient BERT with the method of BBCT. Our benchmark test results on General Language Understanding Evaluation (GLUE) show that BBCT can achieve less than 1% accuracy drop in most tasks.

Keywords

Cite

@article{arxiv.2303.09184,
  title  = {Block-wise Bit-Compression of Transformer-based Models},
  author = {Gaochen Dong and Wei Chen},
  journal= {arXiv preprint arXiv:2303.09184},
  year   = {2023}
}

Comments

Need to add figures and adjust languages to improve readability

R2 v1 2026-06-28T09:19:56.590Z