AI & ML
Decode Token Parallelism: Memory-Efficient Distributed Attention Parallelism for Scalable LLM Decoding
arXiv cs.DC0 views
arXiv:2611.13293v1 Announce Type: new Abstract: Serving large language models at high concurrency with long contexts requires KV-cache capacity and attention bandwidth that grow with the decode pool. Existing parallelization strategies trade one against the other: data parallelism replicates model weights, limiting KV capacity; tensor parallelism stops scaling attention at the model's KV-head count and pushes dense collectives across nodes; and decode context parallelism (DCP) recovers capacity by sharding the sequence but adds cross-node aggregation of partial attention results. We observe that during decode, dense layers and attention scale differently: dense layers amortize weight reads across the batch and run efficiently within a single NVLink domain, whereas attention cost grows with every active request's KV cache. We present Decode Token Parallelism (TKNP), a decode-pool architecture that decouples weight placement from attention ownership. TKNP distributes a single model copy across nodes in balanced layer blocks. The active node holding a block executes its projections and MLP for the full batch using intra-node tensor parallelism, while every node computes attention for its own requests against locally resident KV cache. This extends request-local attention across nodes without replicating weights: only current-token activations cross the network, dense all-reduces remain within NVLink, and attention requires no cross-node merging of partial results. Implemented in vLLM and evaluated on Llama 3.1 405B and Nemotron-4 340B on up to 64 H100 GPUs, TKNP improves completion throughput for long-context, high-concurrency workloads by up to 1.7$\times$ over DCP and by up to 2.06$\times$ over data parallelism at matched GPU count.
Source:https://arxiv.org/abs/2611.13293PreviousOne-Sided Expert-Parallel All-to-All for Mixture-of-Experts Inference over NVLinkNextPegasusAI Studio: Legible Cyberinfrastructure and Accountable AI Agents for Scientific Workflows
Related posts
Voice layer for any website – does it worth?
0Hacker News New
Brazil's AI Development Strategy Needs an Edge Layer – TechPolicy.Press
0Hacker News New
Generative AI and the temporary upskilling of knowledge workers
0Hacker News New
Anthropic Unveils Claude Whiteboards, Claude Sheets, and Claude Data
0Hacker News New
AI Cracks a Quantum Computing Bottleneck
0Hacker News New
Jack and Jill Cut Candidate Screening Costs by 88% with Jev
0Hacker News New