AI & ML
FastAFD: When Does Attention-FFN Disaggregation Pay on Rack-Scale NVLink?
arXiv cs.DC0 views
arXiv:2611.07149v1 Announce Type: new Abstract: Attention-FFN disaggregation (AFD) separates attention from the experts of a mixture-of-experts (MoE) model and has been reported to raise decode throughput over colocated serving. Whether this advantage holds on rack-scale NVLink systems such as GB200 NVL72 remains unclear: prior results were obtained on other hardware, against narrower colocated baselines, or with analytical models, whereas production engines serve large MoE models on these systems with wide expert parallelism (wide EP). We build and tune FastAFD, an AFD engine for GB200 NVL72 that overlaps the MoE of one microbatch with the attention of the other, and compare it against the best vLLM wide EP setting. Experiments show that AFD has a throughput advantage of up to 1.17x over wide EP on Qwen3 and 1.16x on Qwen3.5, and matches it on MiniMax-M2.5. We derive an analytical model that predicts AFD throughput within 3% using only profiles of colocated vLLM, telling whether switching to AFD would pay off before deploying it. The model yields a concise insight: AFD's speedup factors into a batch gain (larger batches, since attention GPUs hold no expert weights), an overlap gain (shorter steps, since the MoE runs on the expert GPUs behind attention), and a disaggregation cost (the GPUs given to experts). AFD has the advantage when the two gains outweigh the cost. This view explains which models benefit from AFD, and shows that speculative decoding helps light-expert models most, where the enlarged overlap gain outweighs the larger disaggregation cost and adds a further 8% to AFD's advantage over wide EP. Looking ahead, serving the experts on accelerators that keep their weights in on-chip SRAM, such as the Groq LPX accelerators paired with Vera Rubin NVL72, would remove most of the disaggregation cost, raising the ideal speedup to 2.48x with speculative decoding. FastAFD is open source at https://github.com/hao-ai-lab/FastAFD.
Source:https://arxiv.org/abs/2611.07149PreviousNUMA-Torch: Compiler-Driven NUMA-Aware LLM Prefill on Multi-Socket CPUsNextCapability Gain from ISA to Shared Execution on AMD MI300A and MI250X
Related posts
Palo Alto Networks in talks to buy AI agent security startup Clutch for $500M
0TechCrunch
I built a vocabulary battle game for English learners — and my crowdfunding is still at zero backers
0dev.to
Meta Just Bought Conductor — the Race in AI Coding Is No Longer About the Model
0dev.to
Emergent Generalization by Representation Learning in Artificial Neural Networks
0arXiv math.IT
Principles of Quantum Communication Theory: A Modern Approach
0arXiv math.IT
Quantitative Tiling Stability From Quadratic Discrepancy In Hamming Spaces
0arXiv math.IT