一种稀疏 MoE 模型推理架构:通过分层与线性衰减增加激活参数量,无需训练或微调即可实现精简推理 [p]

Reddit r/MachineLearning 工具

摘要

一种 llama.cpp 实现,可在推理阶段通过自适应阈值与分层线性衰减将 MoE 专家路由扩展至原生 top-K 之外,无需重新训练或微调。

我将 MoE 专家扩展移植到了 llama.cpp 🚀 运行 MoE 模型时可路由比原生 top-K 更多的专家(8→x),支持自适应阈值、99→50% 影响力衰减及层范围设定。纯运行时实现,兼容所有后端。已在 Qwen 3.6 35B A4B+ 上测试 https://github.com/vagrillo/llama.cpp/blob/moe-expansion/docs/moe-expansion.md
查看原文

相似文章

多层级MoE缓存

Reddit r/LocalLLaMA

讨论MoE模型的多层级缓存策略,通过将频繁激活的专家保留在GPU上来提升推理速度,参考了PowerInfer和llama.cpp分支等现有实现。

粘性路由:训练MoE模型以实现内存高效推理

arXiv cs.LG

StickyMoE提出了一种可微的路由一致性损失函数,鼓励相邻token在MoE模型中激活相同的专家,从而在边缘设备推理过程中将专家交换开销和缓存未命中率降低高达3.92倍,同时改善困惑度。