SOCKET: SOft Collision Kernel EsTimator for Sparse Attention

arXiv:2602.06283 · cs.LG · Submitted 2026-02-06 · Read on arXiv

cs.LG

Submitted: 2026-02-06

Updated: 2026-09-25

Code: https://github.com/amarka8/SOCKET

Terminology

• Sparsity — 38× in this paper · explained in 4 episodes
• Sparse Attention — 30× in this paper · explained in 5 episodes
• Accuracy — 8× in this paper · explained in 1 episode
• Transformer — 7× in this paper · explained in 2 episodes
• Benchmark — 6× in this paper · explained in 2 episodes
• Ablation — 5× in this paper · explained in 1 episode
• Decode — 5× in this paper · explained in 2 episodes
• Locality-Sensitive Hashing — 5× in this paper
• Aggregation — 4× in this paper · explained in 1 episode
• Alignment — 4× in this paper · explained in 1 episode
• KV cache — 4× in this paper · explained in 3 episodes
• Theoretical Guarantees — 3× in this paper · explained in 2 episodes
• Calibration — 2× in this paper · explained in 7 episodes
• Clustering — 2× in this paper · explained in 2 episodes
• Error Decomposition — 2× in this paper · explained in 3 episodes
• In-Context Learning — 2× in this paper · explained in 4 episodes
• Quantization — 2× in this paper · explained in 12 episodes
• Attention Mechanism — 1× in this paper · explained in 2 episodes
• Cosine Similarity — 1× in this paper · explained in 3 episodes
• Densified Winner Take All — 1× in this paper
• Distribution Shift — 1× in this paper · explained in 5 episodes
• Fidelity — 1× in this paper · explained in 2 episodes
• Generalization — 1× in this paper · explained in 13 episodes
• Generative AI — 1× in this paper · explained in 2 episodes
• Hybrid Architectures — 1× in this paper · explained in 3 episodes
• Large Language Models (LLMs) — 1× in this paper · explained in 6 episodes
• Normalized Discounted Cumulative Gain — 1× in this paper
• Semantic Similarity — 1× in this paper · explained in 2 episodes
• Transfer Learning — 1× in this paper · explained in 10 episodes

Sources

• GPT-4 Technical Report
• Longformer: The Long-Document Transformer
• FlashAttention-2: Faster Attention with Better Parallelism and Work Partitioning
• The Llama 3 Herd of Models
• LLaMA: Open and Efficient Foundation Language Models
• Adamas: Hadamard Sparse Attention for Efficient Long-Context Inference
• Qwen3 Technical Report
• Post-Training Sparse Attention with Double Sparsity

Related papers

• Polynomial-Augmented Neural Networks (PANNs) with Weak Orthogonality Constraints for Enhanced Function and PDE Approximation
• AIRL-S: Unifying Reinforcement Learning and Search-Based Test-Time Scaling via Adversarial Inverse Reinforcement Learning
• Transformers as Bayesian In-Context Experimenters: Smoothness-Adaptive Efficient ATE Estimation
• Convergence issues in Relational Concept Analysis based on AOC-posets
• Beliefs Beyond Posteriors: Local-Consistency Optimisation for Bayesian Neural Networks
• Understanding Diffusion Models via Ratio-Based Function Approximation with SignReLU Networks