about
Efficient Transformer Knowledge Distillation: A Performance Review (arxiv.org)
63 points by PaulHoule on Dec 7, 2023 | hide | past | pdf | 5 comments on HN

In plain words: They tested shrinking transformers that use cheaper attention by training small copies to mimic big ones, across short and long tasks including a new long-document name-finding dataset. The shrunk models kept up to 98.6% of the original accuracy while running more than twice as fast.

Abstract

As pretrained transformer language models continue to achieve state-of-the-art performance, the Natural Language Processing community has pushed for advances in model compression and efficient attention mechanisms to address high computational requirements and limited input sequence length. Despite these separate efforts, no investigation has been done into the intersection of these two fields. In this work, we provide an evaluation of model compression via knowledge distillation on efficient attention transformers. We provide cost-performance trade-offs for the compression of state-of-the-art efficient attention architectures and the gains made in performance in comparison to their full attention counterparts. Furthermore, we introduce a new long-context Named Entity Recognition dataset, GONERD, to train and test the performance of NER models on long sequences. We find that distilled efficient attention transformers can preserve a significant amount of original model performance, preserving up to 98.6% across short-context tasks (GLUE, SQUAD, CoNLL-2003), up to 94.6% across long-context Question-and-Answering tasks (HotpotQA, TriviaQA), and up to 98.8% on long-context Named Entity Recognition (GONERD), while decreasing inference times by up to 57.8%. We find that, for most models on most tasks, performing knowledge distillation is an effective method to yield high-performing efficient attention models with low costs.

Nathan Brown, Ashton Williamson, Tahj Anderson, Logan Lawrence
arXiv:2311.13657 · cs.CL, cs.LG · submitted Nov 22, 2023
abstract · pdf · html · Accepted to EMNLP 2023. 12 pages, 1 figure, 11 tables. Models and data available at https://huggingface.co/giant-oak

add comment on HN

This paper combines knowledge distillation and efficient attention mechanisms.

=> It works (still efficient, lower cost).

Not an unexpected result, but to their credit, they established a new benchmark to test these combinations. KD+LongFormer is one of the best ones, retaining 95.9% of the performance for 50.7% of the cost.

Appendix A

A.1 Data Collection Data for GONERD was obtained through Giant Oak’s GONER software, which scraped web ar- ticles from public facing online news sources as well as the U.S. Department of Justice’s justice.gov domain. This webtext data was randomly sampled with an upweighted probability toward documents from justice.gov so that justice.gov consisted of roughly 25% of the total GONERD dataset.

I skimmed the paper but I don't really understand what knowledge generation actually entails.
Don’t mean to be flip at all, may I suggest:

1. Use Gpt-4 with something like this:

“Help me understand what this paper is about and estimate whether the relevance and impact to the field is likely to be low, medium, or high.

Explain jargon that may be specific to AI research, but don’t bother explaining or expanding on terms familiar to a working software developer or basic undergraduate computer science.”

2. Follow the above prompt with either the abstract or the full text of the paper.

3. Post something useful here and save others the time.

Do not - in my opinion - copy/paste LLM output as a comment. But would love to hear your own succinct, human sounding, HN guideline compatible thoughts.