返回全部动态

UniMoMo:基于专家合并的推荐系统MoE加速方法

原标题:UniMoMo: Expert Merging-Based MoE Acceleration for Large Recommendation Models

Hugging Face Daily Papers一手来源研究质量 83

AI 摘要

UniMoMo 是一种针对大型推荐系统中混合专家(MoE)模型的训练后压缩框架,通过功能相似性分组和层自适应保护,将训练好的 MoE 模型压缩为更小的标准 MoE 检查点,在保持精度的同时加速推理。实验表明,在 Amazon Beauty、KuaiRec 和 TenRec 数据集上,将专家数从 8 压缩到 4 时,NDCG@10 保持 99.92%–102.30%,A100 推理速度提升 1.28–1.63 倍;更激进的 8 到 2 专家设置下,速度提升可达 2.21 倍。该方法无需额外的在线模块,压缩后的模型仍为标准 top-k MoE,便于部署。

以上摘要由 AI 生成,可能存在误差。事实请以原文为准。

正文节选

UniMoMo: Expert Merging-Based MoE Acceleration for Large Recommendation Models Abstract UniMoMo compresses trained recommendation mixture-of-experts models into smaller standard MoE checkpoints via functional similarity grouping and layer-adaptive protection, preserving accuracy while accelerating inference. Sparse mixture-of-experts (MoE) layers expand recommendation capacity through conditional computation, yet a trained checkpoint still stores and routes over its full expert bank. We study a


发布时间:—
抓取时间:2026-08-12 15:24
来源机构:Hugging Face
阅读原文huggingface.co