跳到主要内容
AI News HubLIVE
站内改写2 分钟阅读

MoEZip:面向稀疏混合专家大模型的感知路由 KV 缓存压缩

文章摘要

这篇被 NeurIPS 2026 接收的论文提出 MoEZip,一种专门为稀疏混合专家(MoE)大语言模型设计的 KV 缓存压缩方法。现有压缩技术在稠密 Transformer 上效果不错,却忽略了 MoE 特有的问题:驱逐 KV 条目会改变生成答案时被激活的专家,从而损害生成质量。MoEZip 利用基于路由分布 Fisher 信息矩阵的新指标衡量专家路由对扰动的敏感度,并与注意力权重结合进行感知路由的打分,同时组合针对稠密与稀疏 MoE 架构的驱逐策略。在仅保留 10% KV 缓存的激进压缩下,它在 LongBench 和 RULER 上平均较当前最优基线分别提升 +24 和 +20 分,相当于 2.3 倍和 4.9 倍的分值。

来源arXiv Machine Learning作者: Minsang Kim, Seungjun Baek
MoEZip:面向稀疏混合专家大模型的感知路由 KV 缓存压缩
报告错误

纠错通道尚未开通,可先复制下方文章信息留存。

查看更正说明
直接读正文

混合专家(MoE)语言模型通过为每个 token 仅激活一小部分参数,实现了容量的高效扩展。然而在长上下文推理中,内存仍然是主要瓶颈,因为 KV 缓存会随序列长度线性增长。现有的 KV 缓存压缩方法主要为稠密 Transformer 设计,降低这类模型的开销,但它们忽略了一个 MoE 特有的问题:KV 逐出(eviction)可能在答案生成阶段改变被选中的专家,从而损害生成质量。针对这一问题,Minsang Kim 等作者提出 MoEZip——一种面向稀疏 MoE 语言模型的路由感知 KV 缓存压缩方法。该论文已被神经信息处理系统大会(NeurIPS)2026 接收。

论文的核心思路是:在评估上下文片段的重要性时,不仅要考虑注意力权重,还要考虑该片段对专家路由决策的影响。为估计专家路由对扰动的敏感程度,作者提出了一种基于路由分布 Fisher 信息矩阵的新度量。具体而言,路由分布刻画了每个 token 在各专家上的选择概率,而 Fisher 信息矩阵能够衡量该分布在受到扰动时的变化幅度;敏感度高的上下文片段一旦被逐出,就更容易导致路由发生偏移,因此应被赋予更高的保留优先级。MoEZip 将这一度量与注意力权重结合,形成路由感知的上下文评分机制。

此外,作者还构建了一套互补的评分体系,把面向稠密架构的逐出方法与面向稀疏 MoE 架构的逐出方法组合起来使用,从而在两类结构上都能兼顾传统注意力重要性与路由稳定性。

实验结果表明,MoEZip 在多种基准上、在激进的缓存压缩条件下仍能保持稳健性能。在 KV 保留比例为 10% 时,MoEZip 在 LongBench 和 RULER 上相较当前最优(SoTA)基线平均分别提升 +24 分和 +20 分,相当于得分分别提高 2.3 倍和 4.9 倍。

论文信息方面,该预印本于 2026 年 10 月 21 日提交至 arXiv(v1 版本提交时间为当日 13:06:30 UTC,全文约 5,660 KB),编号 arXiv:2610.30206,DOI 为 https://doi.org/10.48550/arXiv.2610.30206(经 DataCite 颁发的 DOI 尚待注册),主分类为机器学习(cs.LG),同时交叉归类于人工智能(cs.AI)。论文标注为 NeurIPS 2026 接收论文,除 PDF 外还提供实验性 HTML 版本与 TeX 源码。值得一提的是,KV 缓存随序列长度线性增长是长上下文推理中普遍存在的内存开销来源,而 MoE 模型由于路由机制的存在,使逐出决策的影响超出了单纯的注意力层面,这正是 MoEZip 试图填补的空白。

展开要点与分析

文章情报

工程师进阶

要点

  • 针对 MoE 模型指出 KV 驱逐会改变专家选择、降低生成质量这一被忽视的问题
  • 提出基于路由分布 Fisher 信息矩阵的专家路由敏感度指标,并与注意力权重结合打分
  • 在 10% KV 保留率下,LongBench 与 RULER 平均分别比当前最优基线高 24 分和 20 分
  • 论文已被 NeurIPS 2026 接收,编号 arXiv:2610.30206

要点与分析由自动化流程生成,可能有误,请结合原始来源核实。