ELDR: 专家局部性感知的解码路由用于PD分离的MoE服务
摘要
ELDR是一种专家局部性感知的解码路由器,用于预填充-解码分离的混合专家(MoE)服务。它通过预填充签名预测专家激活,并路由请求以最小化延迟。该路由器已在vLLM中实现,在多达40个GPU上实现了中位数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: 具有可学习块专家的扩散大语言模型
dMoE 提出了用于扩散大语言模型的块级专家路由,将唯一激活的专家数量从 69.5 降至 14.6,同时保留了 99.11% 的性能,并实现了 76-80% 的内存减少和 1.14-1.66 倍的加速。
BEAM:用于MoE动态路由的二值专家激活掩码
BEAM通过二值专家激活掩码实现混合专家大语言模型的动态路由,在最小化性能损失的情况下将FLOPs减少高达85%,解码速度提升2.5倍。
Director: 通过在线主动专家放置加速分布式MoE服务
本文介绍了Director,一个分布式MoE服务系统,通过预测驱动的在线主动专家放置来最小化端到端延迟。它采用轻量级预测器和基于松弛法的优化器,对Mistral、DeepSeek和Qwen等模型可实现高达55%的延迟降低。
@robertnishihara: 关于PD分离的一些直觉——PD不会加速预填充,实际上可能损害TTFT——PD的真正…
这篇来自Anyscale的博客文章解释了LLM服务中Prefill-Decode(PD)分离的直觉,展示了如何将预填充和解码阶段分配到专用GPU上,在使用Ray和vLLM的AMD MI325X上实现高达2.7倍的有效吞吐量提升和67%的成本节省,同时也讨论了PD分离何时没有帮助。
DECO:端侧设备上媲美稠密性能的稀疏混合专家模型
DECO是一种稀疏MoE架构,仅激活20%的专家并配合3倍加速内核,即可达到与稠密Transformer相当的性能。该架构采用了基于ReLU的路由机制、可学习缩放因子以及NormSiLU激活函数。