VITA-Group / SSM-Bottleneck
[ICLR'25] "Understanding Bottlenecks of State Space Models through the Lens of Recency and Over-smoothing" by Peihao Wang, Ruisi Cai, Yuehao Wang, Jiajun Zhu, Pragya Srivastava, Zhangyang Wang, Pan Li
☆11Updated last week
Alternatives and similar repositories for SSM-Bottleneck:
Users that are interested in SSM-Bottleneck are comparing it to the libraries listed below
- Official code for the paper "Attention as a Hypernetwork"☆25Updated 9 months ago
- Code for the paper "Stack Attention: Improving the Ability of Transformers to Model Hierarchical Patterns"☆17Updated last year
- Unofficial implementation of paper : Exploring the Space of Key-Value-Query Models with Intention☆11Updated last year
- HGRN2: Gated Linear RNNs with State Expansion☆53Updated 7 months ago
- The official repository for our paper "The Dual Form of Neural Networks Revisited: Connecting Test Time Predictions to Training Patterns …☆16Updated last year
- Official code for "Accelerating Feedforward Computation via Parallel Nonlinear Equation Solving", ICML 2021☆27Updated 3 years ago
- Implementation of the model "Hedgehog" from the paper: "The Hedgehog & the Porcupine: Expressive Linear Attentions with Softmax Mimicry"☆13Updated last year
- ☆17Updated 2 months ago
- ☆12Updated last month
- Curse-of-memory phenomenon of RNNs in sequence modelling☆19Updated this week
- Official PyTorch Implementation of the Longhorn Deep State Space Model☆50Updated 3 months ago
- LV-BERT: Exploiting Layer Variety for BERT (Findings of ACL 2021)☆18Updated last year
- ☆16Updated 2 years ago
- Official implementation of "Continual Learning by Modeling Intra-Class Variation" (MOCA). [TMLR 2023]☆16Updated 2 years ago
- Directed masked autoencoders☆14Updated 2 years ago
- [NeurIPS 2024, spotlight] Multivariate Learned Adaptive Noise for Diffusion Models☆18Updated 3 months ago
- [ICLR 2025] Official PyTorch implementation of "Forgetting Transformer: Softmax Attention with a Forget Gate"☆80Updated last week
- Official repository of paper "RNNs Are Not Transformers (Yet): The Key Bottleneck on In-context Retrieval"☆26Updated 11 months ago
- ☆13Updated 2 years ago
- Code for "Are “Hierarchical” Visual Representations Hierarchical?" in NeurIPS Workshop for Symmetry and Geometry in Neural Representation…☆20Updated last year
- Combining SOAP and MUON☆13Updated last month
- [ICCV23] Official implementation of eP-ALM: Efficient Perceptual Augmentation of Language Models.☆27Updated last year
- An adaptive training algorithm for residual network☆15Updated 4 years ago
- implementation of dualformer☆13Updated last month
- [ECCV 2024] This is the official implementation of "Stitched ViTs are Flexible Vision Backbones".☆27Updated last year
- Reference implementation of "Softmax Attention with Constant Cost per Token" (Heinsen, 2024)☆24Updated 9 months ago
- [ICML 2023] "Data Efficient Neural Scaling Law via Model Reusing" by Peihao Wang, Rameswar Panda, Zhangyang Wang☆14Updated last year
- A big_vision inspired repo that implements a generic Auto-Encoder class capable in representation learning and generative modeling.☆34Updated 9 months ago
- Does VLM Classification Benefit from LLM Description Semantics? (AAAI 2025)☆16Updated 2 months ago
- ☆23Updated 6 months ago