English

Analytical Provisioning for Attention-FFN Disaggregated LLM Serving under Stochastic Workloads

Machine Learning 2026-05-13 v3 Artificial Intelligence

Abstract

Attentio-FFN disaggregation (AFD) is an emerging architecture for LLM decoding that separates state-heavy, KV-cache-dominated Attention computation from stateless, compute-intensive FFN computation, connected by per-step communication. While AFD enables independent scaling of memory and compute resources, its performance is highly sensitive to the Attention/FFN provisioning ratio: mis-sizing induces step-level blocking and costly device idle time. We develop an analytical provisioning framework for AFD bundles in an rrA--11F topology under stochastic workloads. Two sources of randomness shape the problem: per-slot Attention workload evolves as KV caches grow and completed requests are replenished with random prompt and decode lengths, and synchronized execution across Attention workers introduces a barrier governed by the slowest worker. We address both via a renewal-reward characterization of the per-slot stationary token load, identifying a single workload statistic θ\theta that governs provisioning under arbitrary prefill-decode distributions and admits a nonparametric estimator from request traces. The analysis yields a closed-form mean-field rule for the optimal A/F ratio decomposing into Attention-, communication-, and FFN-bottleneck regimes, together with a Gaussian barrier-aware refinement that quantifies cross-worker synchronization overhead. A trace-calibrated AFD simulator supports the framework across workloads: the predicted optimal ratio matches the simulation-optimal within 10%. Together, these results provide a compact, calibratable account of how stochastic workload structure determines provisioning in disaggregated LLM serving.

Keywords

Cite

@article{arxiv.2601.21351,
  title  = {Analytical Provisioning for Attention-FFN Disaggregated LLM Serving under Stochastic Workloads},
  author = {Chendong Song and Meixuan Wang and Hang Zhou and Hong Liang and Yuan Lyu and Zixi Chen and Yuwei Fan and Zijie Zhou},
  journal= {arXiv preprint arXiv:2601.21351},
  year   = {2026}
}

Comments

Submitted to Neurips 2026

R2 v1 2026-07-01T09:25:10.012Z