ELDR: 专家局部性感知的解码路由用于PD分离的MoE服务

Hugging Face Daily Papers 论文

摘要

ELDR是一种专家局部性感知的解码路由器,用于预填充-解码分离的混合专家(MoE)服务。它通过预填充签名预测专家激活,并路由请求以最小化延迟。该路由器已在vLLM中实现,在多达40个GPU上实现了中位数TPOT降低5.9%至13.9%。

在预填充-解码(PD)分离的LLM服务中,每个请求在预填充后被分配给一个解码工作节点。现有的解码路由器仅平衡负载;对于混合专家(MoE)模型,这是不完整的:负载相等的工作节点可能在延迟上有所不同,因为每个解码步骤会加载其批次激活的每个不同专家的权重。我们提出了ELDR,一种用于PD分离的MoE服务的专家局部性感知的解码路由器。从请求的预填充专家激活中,ELDR构建一个专家签名,预测其在生成过程中将激活的专家。在离线阶段,平衡的K-means将签名空间分割到各个解码工作节点;在线阶段,局部性带路由将每个请求发送到与其签名最匹配且负载最低的工作节点。一个签名缓存,与KV缓存以KV块粒度共同索引,在前缀缓存下保持签名精确。在vLLM中实现,并在多达40个GPU的部署上进行评估,ELDR在三个MoE模型和两个工作负载上,将中位数TPOT降低了5.9%至13.9%,优于四个负载均衡基线中最强的,且模型输出不变。
查看原文
查看缓存全文

缓存时间: 2026/07/02 07:47

论文页面 - ELDR:面向预填-解码分离式MoE服务的专家局部性感知路由

来源: https://huggingface.co/papers/2607.00466

摘要

ELDR 是一个面向预填-解码分离式混合专家(MoE)服务的专家局部性感知解码路由方案,通过预测专家激活模式并据此进行路由分配来提升性能。

在预填-解码(https://huggingface.co/papers?q=prefill-decode)分离式大语言模型服务(https://huggingface.co/papers?q=disaggregated%20LLM%20serving)中,每个请求在完成预填后被分配给一个解码工作节点。现有的解码路由(https://huggingface.co/papers?q=decode%20router)方案仅平衡负载;对于混合专家(https://huggingface.co/papers?q=mixture-of-experts)(MoE)模型而言,这种做法并不完善:负载相等的工作节点可能产生不同的延迟,因为每个解码步骤都需要加载其批次中激活的每个不同专家的权重。我们提出了 ELDR,一种面向预填-解码分离式 MoE 服务的专家局部性感知解码路由方案。从请求的预填专家激活信息中,ELDR 构建一个专家签名,用以预测该请求在生成阶段将激活的专家。在离线阶段,平衡K均值(https://huggingface.co/papers?q=K-means)将签名空间划分为多个解码工作节点;在线阶段,局部性带路由(https://huggingface.co/papers?q=locality-band%20routing)将每个请求发送到与其签名最匹配且负载最低的工作节点。一个签名缓存(https://huggingface.co/papers?q=signature%20cache),与KV缓存(https://huggingface.co/papers?q=KV%20cache)以KV块粒度联合索引,使得签名在缓存前缀时保持精确。在 vLLM(https://huggingface.co/papers?q=vLLM)中实现,并在多达40个GPU的部署上进行评估,ELDR 将中位数TPOT(https://huggingface.co/papers?q=TPOT)降低了5.9%-13.9%,超过四种负载均衡基线中最优的方案,覆盖三个 MoE 模型和两个工作负载,且模型输出保持不变。

查看 arXiv 页面 (https://arxiv.org/abs/2607.00466)查看 PDF (https://arxiv.org/pdf/2607.00466)添加至收藏 (https://huggingface.co/login?next=%2Fpapers%2F2607.00466)

在您的智能体中获取此论文:

hf papers read 2607.00466

没有最新的 CLI?curl \-LsSf https://hf\.co/cli/install\.sh \| bash

引用此论文的模型0

没有关联此论文的模型

请在模型 README.md 中引用 arxiv.org/abs/2607.00466 以从本页面链接该模型。

引用此论文的数据集0

没有关联此论文的数据集

请在数据集 README.md 中引用 arxiv.org/abs/2607.00466 以从本页面链接该数据集。

引用此论文的 Spaces0

没有关联此论文的 Space

请在 Space README.md 中引用 arxiv.org/abs/2607.00466 以从本页面链接该 Space。

包含此论文的收藏集0

没有包含此论文的收藏集

请将此论文添加至一个收藏集 (https://huggingface.co/new-collection) 以从本页面链接该收藏集。

相似文章

dMoE: 具有可学习块专家的扩散大语言模型

arXiv cs.CL

dMoE 提出了用于扩散大语言模型的块级专家路由,将唯一激活的专家数量从 69.5 降至 14.6,同时保留了 99.11% 的性能,并实现了 76-80% 的内存减少和 1.14-1.66 倍的加速。

Director: 通过在线主动专家放置加速分布式MoE服务

arXiv cs.LG

本文介绍了Director,一个分布式MoE服务系统,通过预测驱动的在线主动专家放置来最小化端到端延迟。它采用轻量级预测器和基于松弛法的优化器,对Mistral、DeepSeek和Qwen等模型可实现高达55%的延迟降低。