lena-voita / the-story-of-headsLinks
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".
☆314Updated 3 years ago
Alternatives and similar repositories for the-story-of-heads
Users that are interested in the-story-of-heads are comparing it to the libraries listed below
Sorting:
- Code for the paper "Are Sixteen Heads Really Better than One?"☆172Updated 5 years ago
- Understanding the Difficulty of Training Transformers☆329Updated 3 years ago
- Repository containing code for "How to Train BERT with an Academic Budget" paper☆313Updated last year
- ☆85Updated last year
- 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…☆244Updated 3 years ago
- ☆321Updated 4 years ago
- For the code release of our arXiv paper "Revisiting Few-sample BERT Fine-tuning" (https://arxiv.org/abs/2006.05987).☆184Updated 2 years ago
- [ACL 2022] Structured Pruning Learns Compact and Accurate Models https://arxiv.org/abs/2204.00408☆195Updated 2 years ago
- ☆492Updated last year
- An original implementation of "MetaICL Learning to Learn In Context" by Sewon Min, Mike Lewis, Luke Zettlemoyer and Hannaneh Hajishirzi☆266Updated 2 years ago
- DeeBERT: Dynamic Early Exiting for Accelerating BERT Inference☆156Updated 3 years ago
- A PyTorch implementation of Transformer in "Attention is All You Need"☆106Updated 4 years ago
- Code for Multi-Head Attention: Collaborate Instead of Concatenate☆152Updated 2 years ago
- ☆464Updated 4 years ago
- Fully featured implementation of Routing Transformer☆295Updated 3 years ago
- Neural Text Generation with Unlikelihood Training☆309Updated 3 years ago
- Implementation of Universal Transformer in Pytorch☆260Updated 6 years ago
- [NeurIPS 2020] "The Lottery Ticket Hypothesis for Pre-trained BERT Networks", Tianlong Chen, Jonathan Frankle, Shiyu Chang, Sijia Liu, Ya…☆140Updated 3 years ago
- Transformers without Tears: Improving the Normalization of Self-Attention☆131Updated last year
- ☆315Updated 3 years ago
- PyTorch implementation of BERT in "BERT: Pre-training of Deep Bidirectional Transformers for Language Understanding"☆107Updated 6 years ago
- ☆292Updated 2 years ago
- PyTorch + HuggingFace code for RetoMaton: "Neuro-Symbolic Language Modeling with Automaton-augmented Retrieval" (ICML 2022), including an…☆274Updated 2 years ago
- The entmax mapping and its loss, a family of sparse softmax alternatives.☆439Updated last year
- Optimus: the first large-scale pre-trained VAE language model☆387Updated last year
- Transformer with Untied Positional Encoding (TUPE). Code of paper "Rethinking Positional Encoding in Language Pre-training". Improve exis…☆251Updated 3 years ago
- Simple Parameter-efficient Fine-tuning for Transformer-based Masked Language-models☆142Updated 2 years ago
- ☆207Updated last year
- Package to compute Mauve, a similarity score between neural text and human text. Install with `pip install mauve-text`.☆292Updated 11 months ago
- Tracking the progress in non-autoregressive generation (translation, transcription, etc.)☆306Updated 2 years ago