[ICML 2021 Oral] We show pure attention suffers rank collapse, and how different mechanisms combat it.
☆173Mar 8, 2021Updated 5 years ago
Alternatives and similar repositories for attention-rank-collapse
Users that are interested in attention-rank-collapse are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Co-Supervised Learning: Improving Weak-to-Strong Generalization with Hierarchical Mixture of Experts☆15Feb 26, 2024Updated 2 years ago
- ☆20Aug 10, 2024Updated 2 years ago
- ☆12Sep 26, 2019Updated 7 years ago
- We investigated corruption robustness across different architectures including Convolutional Neural Networks, Vision Transformers, and th…☆16Oct 28, 2021Updated 4 years ago
- ☆13Feb 16, 2021Updated 5 years ago
- End-to-end encrypted email - Proton Mail • AdSpecial offer: 40% Off Yearly / 80% Off First Month. All Proton services are open source and independently audited for security.
- Locally Enhanced Self-Attention: Rethinking Self-Attention as Local and Context Terms☆20Nov 29, 2021Updated 4 years ago
- Code Release for the 2023 NeurIPS Paper How does GPT-2 compute greater-than?: Interpreting mathematical abilities in a pre-trained langua…☆17Dec 6, 2024Updated last year
- Code for Branched Schrödinger Bridge Matching (ICLR 2026)☆20Mar 20, 2026Updated 6 months ago
- Efficient Householder Transformation in PyTorch☆70Jul 6, 2021Updated 5 years ago
- [NeurIPS'20] Code for the Paper Compositional Visual Generation and Inference with Energy Based Models☆48Mar 24, 2023Updated 3 years ago
- ☆390Oct 18, 2023Updated 2 years ago
- Implementation of the GBST block from the Charformer paper, in Pytorch☆118Jul 15, 2021Updated 5 years ago
- A simple Transformer where the softmax has been replaced with normalization☆20Sep 11, 2020Updated 6 years ago
- Variational Walkback, NIPS'17☆28Oct 18, 2017Updated 8 years ago
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- Implementation of RETRO, Deepmind's Retrieval based Attention net, in Pytorch☆878Oct 30, 2023Updated 2 years ago
- ☆26Nov 23, 2023Updated 2 years ago
- ☆20Feb 26, 2021Updated 5 years ago
- Code for 'The Linear Representation Hypothesis and the Geometry of Large Language Models' (ICML 2024)☆134Feb 11, 2025Updated last year
- Implementation of N-Grammer, augmenting Transformers with latent n-grams, in Pytorch☆82Dec 4, 2022Updated 3 years ago
- This repository is for the paper "A generative nonparametric Bayesian model for whole genomes"☆16Jun 7, 2023Updated 3 years ago
- ☆14Feb 26, 2024Updated 2 years ago
- Training scripts for paper Miceli Barone et al. 2017 "Deep Architectures for Neural Machine Translation"☆11Jul 13, 2017Updated 9 years ago
- RelationNet++: Bridging Visual Representations for Object Detection via Transformer Decoder☆211Mar 18, 2021Updated 5 years ago
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- ☆32Oct 13, 2021Updated 4 years ago
- Implementation of H-Transformer-1D, Hierarchical Attention for Sequence Learning☆166Feb 12, 2024Updated 2 years ago
- Implementation of Fast Transformer in Pytorch☆176Aug 26, 2021Updated 5 years ago
- ☆45Mar 3, 2023Updated 3 years ago
- Fine-Tuning Pre-trained Transformers into Decaying Fast Weights☆20Oct 9, 2022Updated 4 years ago
- ☆20Mar 22, 2024Updated 2 years ago
- ☆46Apr 13, 2022Updated 4 years ago
- Disentangled Non-Local Neural Networks☆84Dec 7, 2020Updated 5 years ago
- This is the official GitHub for paper: On the Versatile Uses of Partial Distance Correlation in Deep Learning, in ECCV 2022☆176Jun 4, 2023Updated 3 years ago
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting for WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Deploy in minutes on Cloudways by DigitalOcean.
- Transformer based on a variant of attention that is linear complexity in respect to sequence length☆845May 5, 2024Updated 2 years ago
- Pretraining summarization models using a corpus of nonsense☆13Sep 28, 2021Updated 5 years ago
- [NeurIPS‘2021] "TransGAN: Two Pure Transformers Can Make One Strong GAN, and That Can Scale Up", Yifan Jiang, Shiyu Chang, Zhangyang Wang☆1,694Nov 3, 2022Updated 3 years ago
- Official repository for the "Big Transfer (BiT): General Visual Representation Learning" paper.☆1,540Jul 30, 2024Updated 2 years ago
- Implementation of paper "D2Match: Leveraging Deep Learning and Degeneracy for Subgraph Matching"☆14Dec 30, 2023Updated 2 years ago
- Source codes for "Preference-grounded Token-level Guidance for Language Model Fine-tuning" (NeurIPS 2023).☆17Jan 8, 2025Updated last year
- Long Range Arena for Benchmarking Efficient Transformers☆788Dec 16, 2023Updated 2 years ago