GuoTianYu2000 / Active-Dormant-AttentionLinks
codes and plots for "Active-Dormant Attention Heads: Mechanistically Demystifying Extreme-Token Phenomena in LLMs"
☆10Updated 10 months ago
Alternatives and similar repositories for Active-Dormant-Attention
Users that are interested in Active-Dormant-Attention are comparing it to the libraries listed below
Sorting:
- Unofficial Implementation of Selective Attention Transformer☆17Updated last year
- SLTrain: a sparse plus low-rank approach for parameter and memory efficient pretraining (NeurIPS 2024)☆36Updated last year
- ☆35Updated last year
- ☆19Updated 7 months ago
- [ICLR 2025] When Attention Sink Emerges in Language Models: An Empirical View (Spotlight)☆135Updated 4 months ago
- The official implementation for [NeurIPS2025 Oral] Gated Attention for Large Language Models: Non-linearity, Sparsity, and Attention-Sink…☆105Updated 2 months ago
- Stick-breaking attention☆61Updated 4 months ago
- Implementation of CoLA: Compute-Efficient Pre-Training of LLMs via Low-Rank Activation☆24Updated 9 months ago
- ☆20Updated 2 weeks ago
- Kinetics: Rethinking Test-Time Scaling Laws☆82Updated 4 months ago
- [NeurIPS '25] Multi-Token Prediction Needs Registers☆24Updated 2 months ago
- Code for NeurIPS 2024 Spotlight: "Scaling Laws and Compute-Optimal Training Beyond Fixed Training Durations"☆85Updated last year
- AdaSplash: Adaptive Sparse Flash Attention (aka Flash Entmax Attention)☆29Updated last month
- LISA: Layerwise Importance Sampling for Memory-Efficient Large Language Model Fine-Tuning☆35Updated last year
- ☆33Updated last year
- Code for ICLR 2025 Paper "What is Wrong with Perplexity for Long-context Language Modeling?"☆105Updated last month
- Code for the paper: Why Transformers Need Adam: A Hessian Perspective☆63Updated 8 months ago
- Activation-aware Singular Value Decomposition for Compressing Large Language Models☆80Updated last year
- Code for the EMNLP24 paper "A simple and effective L2 norm based method for KV Cache compression."☆17Updated 11 months ago
- Test-time-training on nearest neighbors for large language models☆46Updated last year
- [ICML 2024] SPP: Sparsity-Preserved Parameter-Efficient Fine-Tuning for Large Language Models☆21Updated last year
- source code for paper "Riemannian Preconditioned LoRA for Fine-Tuning Foundation Models"☆32Updated last year
- Position Coupling: Improving Length Generalization of Arithmetic Transformers Using Task Structure (NeurIPS 2024) + Arithmetic Transfor…☆11Updated 3 weeks ago
- Multi-Layer Sparse Autoencoders (ICLR 2025)☆26Updated 9 months ago
- ☆47Updated 3 weeks ago
- official implementation of ICLR'2025 paper: Rethinking Bradley-Terry Models in Preference-based Reward Modeling: Foundations, Theory, and…☆69Updated 7 months ago
- A Sober Look at Language Model Reasoning☆87Updated this week
- [ICLR 2025] Linear Combination of Saved Checkpoints Makes Consistency and Diffusion Models Better☆16Updated 9 months ago
- ☆61Updated 4 months ago
- ☆95Updated 8 months ago