跳至正文
Apple Machine Learning Research·· 9 天前AI 評分38

Apple 的學習式支撐函數加速 RAG 檢索

Faster Retrieval for Better, More Efficient RAG

AI 導讀

Apple 研究人員在 ICML 2026 發表了「Amortizing Maximum Inner Product Search with Learned Support Functions」,透過離線訓練小型神經網路,將查詢更快地導向最匹配的鍵。SupportNet 透過輸入凸神經網路(ICNNs)學習支撐函數,以改善叢集路由;KeyNet 則直接輸出最匹配的鍵向量,並藉由歐拉定理得出的一致性損失,使其在幾何上保持有據。兩者皆在實際查詢分布上訓練,以預先攤提的成本換取推論時更快、更準確的檢索。

來源:Apple Machine Learning Research · machinelearning.apple.com