SeqMoE:预测式与图兼容 MoE 卸载实现近满载性能
🧭 MoE 推理卸载 · 预测式专家预取
University of Science and Technology of China · Suzhou Institute for Advanced Research, University of Science and Technology of China
通过序列建模预测专家激活,结合联合预取调度与图兼容运行时,实现 45% 常留下 96.97% 命中率和 80.22% 满载性能
📌 工作参考价值(4 点)
- 若业务使用 MoE 架构 LLM(如 Mixtral、DeepSeek-MoE)做 query 改写、文案生成或 Agent 工具编排,可参考 SeqMoE 的专家激活序列预测:只让约 45% 的 expert 常驻显存,仍能获得约 80% 的 full-load 性能,适合在显存受限 GPU 上部署大规模 MoE 推理服务。
- 将“下一步需要加载哪些 expert”建模为序列预测问题,类似推荐中的 next-item prediction,可以用轻量 decoder 输出多步、多层的 expert ID 序列,比按层独立 LRU 更精准,能直接迁移到 LLM serving 的预取模块。
- 联合预取调度可形式化为 Job Sequencing with Deadlines:把 PCIe/NVLink 带宽作为约束、各 token 的 expert 需求作为 deadline,统一调度一批请求的 expert 加载,能显著提升带宽利用率;在 Continuous Batching 场景下可以复用这一思想。
- 图兼容 runtime 的做法值得工程实现借鉴:将 expert 放置与计算解耦,避免预取引入依赖同步,从而能用 CUDA Graph 捕获端到端执行路径,减少 kernel launch overhead;不过动态 MoE routing 会加大图捕获复杂度,需在 serving 框架里做静态化或部分捕获。
HyQuant:面向 LLM 注意力的混合精度量化框架
🧭 LLM 长上下文推理 · 混合精度量化 ★ 深度解读
Shanghai Jiao Tong University · Xi’an Jiaotong University · Tencent Penglai Lab · Xiamen University
保留 vertical-line token 与局部窗口全精度、其余 KV 低比特,实现长上下文推理的近无损量化加速
📌 工作参考价值(4 点)
- 长上下文/Agent 场景做 KV cache 量化时,不要全局等精度:按列注意力质量选择 top-5% vertical-line token + 最近 W=128 窗口保留 FP16/BF16,其余 4-bit,可在 LongBench/GSM8K 上接近全精度;该模式可直接迁移到电商导购 Agent 的长期对话历史、RAG 长 prompt 缓存。
- 工程实现上,把 dequantize 融合进 attention kernel 的 online softmax 扫描,避免先材料化全精度 KV;在 H100 上 32K 前缀 decode kernel 相对 FA2 达 3.58×、端到端 1.17×;对广告/搜索的 LLM serving 能降低显存带宽压力,显存不足时也可以跑更大 batch。
- Vertical-line 识别只需累加 attention 概率列和,额外 runtime 约 3-5%;用 tail-query proxy 且每 64 tokens 更新,避免逐 token 全局打分,可作为轻量重要性评估器与稀疏/eviction 策略配合。
- 局限:短上下文收益有限,且未在 coding/agentic 任务评估;接入长提示或 long CoT 前应先在自家任务验证 vertical-line 稳定性。