Apple Machine Learning Research·· 21 小时前AI 评分22
Apple 提出用学习型支撑函数加速 MIPS,提升 RAG 检索效率
Faster Retrieval for Better, More Efficient RAG
AI 导读
Apple 在 ICML 2026 论文中提出用小型神经网络学习 MIPS 的支撑函数,离线训练后可在推理时更快、更准确地把查询路由到最佳匹配。方法包含学习支撑函数的 SupportNet 和直接输出最佳匹配 key 的 KeyNet,训练样本来自真实查询分布并用穷举搜索标注。该方案以一次性摊销成本换取推理阶段更高效的检索。
来源:Apple Machine Learning Research · machinelearning.apple.com