MoME: Mixture-of-Memory Embeddings for Context-Aware Sparse Lookup
MoME:上下文感知稀疏查找的混合记忆嵌入
🧭 上下文感知稀疏记忆嵌入
University of British Columbia · Vector Institute for AI · Canada CIFAR AI Chair · NSERC CRC Chair
MoME 用上下文门控选择 M 个记忆槽的混合,取代单一行记忆,在 iso 参数/FLOP 下优于现有稀疏记忆基线,并改善 scaling 趋势。
📌 工作参考价值(4 点)
- 在 LLM4Rec / Query 推荐等场景,可将类目、品牌、高频 query 模板等静态知识注入 token 级条件记忆:用 MoME 式混合门控取代固定 embedding lookup,按上下文选择多个 slot,能区分“苹果”手机与水果等多义 token,提升下游意图理解。
- 在线 serving 需要低延迟稀疏查找时,条件记忆比 MoE 更便宜;MoME 的推理开销接近简单 embedding lookup,适合在推荐/搜索模型里增加轻量领域记忆,而无需承担 expert 路由与全激活成本。
- 对于搜索 query 生成/改写,可借鉴其多槽分离机制,将同一 surface token 的多种语义分开存储,由 learned gate 自动识别意图;定性结果显示路由具有可解释性,有助于后续分析或人工干预。
- 工程实现上注意 slot 数 M 与训练 FLOP 的平衡:MoME 在 sub-billion 规模下 memory-size scaling 趋势更优,说明在中小规模模型上增加记忆容量比扩大 dense 参数更具性价比。