MLIPS: Multi-Layer Indexer Pool Sharing
cs.LG, cs.AI, cs.PF
Submitted: 2026-10-20
Updated: 2026-10-20
License: http://creativecommons.org/licenses/by/4.0/
The gist: Sparse attention limits the keys used by each query, but finding those keys can still require a full-prefix scan at every layer.
Abstract
Sparse attention limits the keys used by each query, but finding those keys can still require a full-prefix scan at every layer. We introduce MLIPS (Multi-Layer Indexer Pool Sharing), a training-free method that shares candidate search while retaining each consumer's own scoring and selection. A producer publishes a top- M pool larger than the attention budget K; subsequent layers select their own top- K from that pool. Measurements across three sparse-attention model families show substantially higher coverage from broad pools than overlap between final selections. We implement layerwise sharing, persistent decode reuse, and a coarse producer in vLLM for DeepSeek-V4-Flash. Across four long-context suites, three consumer layers per producer give 1.14 -- 1.40 times speedups and 14.8 -- 27.4% lower GPU energy, with observed accuracy changes of-1.49 to +0.29 percentage points in single runs. More extensive sharing exposes task-dependent quality losses. Sampled prefill traces also reveal recurring selected KV positions across query groups and layers. We characterize this shared working-set structure and use it to motivate Top- G, an exploratory bounded pool construction for prefill. Its deployment, task accuracy, and serving benefits remain open.
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: The Predictive Cost of ELBO Binding
- Understanding Diffusion Models via Ratio-Based Function Approximation with SignReLU Networks