English

SDBERT: SparseDistilBERT, a faster and smaller BERT model

Computation and Language 2022-08-23 v1 Information Theory Machine Learning math.IT

Abstract

In this work we introduce a new transformer architecture called SparseDistilBERT (SDBERT), which is a combination of sparse attention and knowledge distillantion (KD). We implemented sparse attention mechanism to reduce quadratic dependency on input length to linear. In addition to reducing computational complexity of the model, we used knowledge distillation (KD). We were able to reduce the size of BERT model by 60% while retaining 97% performance and it only took 40% of time to train.

Cite

@article{arxiv.2208.10246,
  title  = {SDBERT: SparseDistilBERT, a faster and smaller BERT model},
  author = {Devaraju Vinoda and Pawan Kumar Yadav},
  journal= {arXiv preprint arXiv:2208.10246},
  year   = {2022}
}
R2 v1 2026-06-25T01:52:09.804Z