English

An Efficient Hybrid Sparse Attention with CPU-GPU Parallelism for Long-Context Inference

Machine Learning 2026-05-11 v1 Artificial Intelligence Performance

Abstract

Long-context inference increasingly operates over CPU-resident KV caches, either because decoding-time KV states exceed GPU memory capacity or because disaggregated prefill-decode systems place KV data in host memory. Although block-sparse attention reduces attention cost in this setting, sparsity alone is insufficient for end-to-end efficiency. GPU-only designs remain constrained by PCIe bandwidth and metadata memory overhead, while CPU-GPU hybrid designs still suffer from substantial GPU idle time and bottlenecks in CPU-side top-k selection and sparse attention computation. Fluxion is built on three key insights: output-aware KV budgeting, head-specific and granularity-aware sparse configuration, and cross-device coordinated execution for sparse attention over CPU-resident KV caches. Guided by these insights, Fluxion combines a lightweight head-property predictor, a granularity-budget selector, and a priority-based scheduler to jointly optimize budget allocation, sparse configuration, and CPU-GPU execution overlap. This co-design enables hybrid sparse attention to achieve both accuracy and system efficiency in long-context inference. Across 2 models, 3 benchmarks, and 40 tasks, Fluxion preserves quality well -- the worst average degradation is only -0.26 relative to FULL, while delivering 1.5×\times-3.7×\times speedup over the strongest fixed sparse hybrid baseline, whose KV budget is only 0.05.

Keywords

Cite

@article{arxiv.2605.07719,
  title  = {An Efficient Hybrid Sparse Attention with CPU-GPU Parallelism for Long-Context Inference},
  author = {Feiyu Yao and Zhixiong Niu and Xiaqing Li and Yongqiang Xiong and Juan Fang and Qian Wang},
  journal= {arXiv preprint arXiv:2605.07719},
  year   = {2026}
}
R2 v1 2026-07-01T12:57:43.728Z