Apple · 机器学习研究·· 21 小时前AI 评分35
Apple 机器学习研究提出基于学习支持函数的更快最大内积搜索方法
Faster Retrieval for Better, More Efficient RAG
AI 导读
Apple 机器学习研究团队提出一种基于学习支持函数的改进最大内积搜索(MIPS)方法,通过训练小型神经网络实现更快速和准确的检索。该方法利用凸分析中的支持函数特性,结合回归技术预训练模型,以降低计算成本。研究显示,分别训练用于支持函数和最佳键向量输出的网络能显著提升路由质量,且模型训练基于实际查询分布的监督数据。
来源:Apple · 机器学习研究 · machinelearning.apple.com