Search papers, labs, and topics across Lattice.
This paper introduces ELDR, an expert-locality-aware decode routing method designed for prefill-decode disaggregated serving of mixture-of-experts (MoE) models. By leveraging expert signatures derived from prefill activations, ELDR optimally routes requests to the least-loaded decode workers that best match the predicted expert activations, significantly improving latency. Evaluated across multiple MoE models and workloads, ELDR achieves a median throughput improvement of 5.9-13.9% compared to traditional load-balancing approaches without compromising output quality.
ELDR slashes median latency by up to 13.9% for MoE models by intelligently routing requests based on expert activation signatures.
In prefill-decode (PD) disaggregated LLM serving, each request is assigned to a decode worker after prefill. Existing decode routers balance only load; for mixture-of-experts (MoE) models this is incomplete: equally loaded workers can differ in latency, since each decode step loads the weights of every distinct expert its batch activates. We present ELDR, an expert-locality-aware decode router for PD-disaggregated MoE serving. From a request's prefill expert activations, ELDR builds an expert signature predicting the experts it will activate during generation. Offline, balanced K-means partitions signature space across decode workers; online, locality-band routing sends each request to the least-loaded worker among those best matching its signature. A signature cache, co-indexed with the KV cache at KV-block granularity, keeps signatures exact under prefix caching. Implemented in vLLM and evaluated on deployments of up to 40 GPUs, ELDR reduces median TPOT by 5.9-13.9% over the strongest of four load-balancing baselines across three MoE models and two workloads, with model outputs unchanged.
Microsoft Research