lena-voita / the-story-of-heads
This is a repository with the code for the ACL 2019 paper "Analyzing Multi-Head Self-Attention: Specialized Heads Do the Heavy Lifting, the Rest Can Be Pruned" and the ACL 2021 paper "Analyzing Source and Target Contributions to NMT Predictions".
☆302Updated 3 years ago
Related projects ⓘ
Alternatives and complementary repositories for the-story-of-heads
- Code for the paper "Are Sixteen Heads Really Better than One?"☆169Updated 4 years ago
- Understanding the Difficulty of Training Transformers☆328Updated 2 years ago
- Repository containing code for "How to Train BERT with an Academic Budget" paper☆309Updated last year
- Pretrain and finetune ELECTRA with fastai and huggingface. (Results of the paper replicated !)☆325Updated 10 months ago
- [ACL 2022] Structured Pruning Learns Compact and Accurate Models https://arxiv.org/abs/2204.00408☆192Updated last year
- ☆315Updated 3 years ago
- Fully featured implementation of Routing Transformer☆284Updated 3 years ago
- A masked language modeling objective to train a model to predict any subset of the target words, conditioned on both the input text and a…☆241Updated 3 years ago
- For the code release of our arXiv paper "Revisiting Few-sample BERT Fine-tuning" (https://arxiv.org/abs/2006.05987).☆184Updated last year
- ☆484Updated last year
- Tracking the progress in non-autoregressive generation (translation, transcription, etc.)☆305Updated last year
- A PyTorch implementation of Transformer in "Attention is All You Need"☆103Updated 3 years ago
- ☆294Updated last year
- DeeBERT: Dynamic Early Exiting for Accelerating BERT Inference☆153Updated 2 years ago
- ☆310Updated 2 years ago
- Neural Text Generation with Unlikelihood Training☆310Updated 3 years ago
- ☆454Updated 3 years ago
- Transformers without Tears: Improving the Normalization of Self-Attention☆130Updated 5 months ago
- Prune a model while finetuning or training.☆394Updated 2 years ago
- PyTorch implementation of BERT in "BERT: Pre-training of Deep Bidirectional Transformers for Language Understanding"☆97Updated 6 years ago
- Transformer with Untied Positional Encoding (TUPE). Code of paper "Rethinking Positional Encoding in Language Pre-training". Improve exis…☆250Updated 3 years ago
- A simple and working implementation of Electra, the fastest way to pretrain language models from scratch, in Pytorch☆222Updated last year
- ☆212Updated 4 years ago
- ☆360Updated last year
- Statistics and accepted paper list of NLP conferences with arXiv link☆430Updated 3 years ago
- ⛵️The official PyTorch implementation for "BERT-of-Theseus: Compressing BERT by Progressive Module Replacing" (EMNLP 2020).☆312Updated last year
- [ACL 2021] LM-BFF: Better Few-shot Fine-tuning of Language Models https://arxiv.org/abs/2012.15723☆721Updated 2 years ago
- Implementation of Mixout with PyTorch☆74Updated last year
- An original implementation of "MetaICL Learning to Learn In Context" by Sewon Min, Mike Lewis, Luke Zettlemoyer and Hannaneh Hajishirzi☆254Updated last year
- A collection of transformer's guides, implementations and variants.☆103Updated 4 years ago