arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2601.21082cs.LGcs.AI
收藏

LOCUS:低维模型嵌入用于高效模型探索、比较和选择

LOCUS: Low-Dimensional Model Embeddings for Efficient Model Exploration, Comparison, and Selection

Carnegie Mellon University(卡内基梅隆大学)

Shivam Patel, William Cocke, Gauri Joshi

AI总结:

LOCUS通过低维嵌入高效实现模型探索、比较和选择,减少查询样本需求并提升模型相似性判断能力。

摘要:

随着大型语言模型(LLMs)生态系统迅速扩展,有效管理和利用庞大的动态模型池变得越来越具有挑战性。我们提出了LOCUS,一种生成低维向量嵌入的方法,能够紧凑地表示语言模型在查询上的能力。LOCUS是一种基于注意力的方法,通过确定性的前向传递查询编码和评估分数,利用编码器模型生成嵌入,使新模型能够无缝地融入池中,并在不重新训练的情况下改进现有模型嵌入。我们还训练了一个正确性预测器,利用模型嵌入和查询编码,在未见过的查询上实现最先进的路由准确性。实验显示,LOCUS需要比基线少多达4.8倍的查询评估样本即可生成信息丰富且稳健的嵌入。此外,学习到的嵌入空间具有几何意义:接近性反映模型相似性,使一系列下游应用成为可能,包括模型比较和聚类、模型组合选择以及不可用模型的稳健代理。

英文摘要:

The rapidly growing ecosystem of Large Language Models (LLMs) makes it increasingly challenging to manage and utilize the vast and dynamic pool of models effectively. We propose LOCUS, a method that produces low-dimensional vector embeddings that compactly represent a language model's capabilities across queries. LOCUS is an attention-based approach that generates embeddings by a deterministic forward pass over query encodings and evaluation scores via an encoder model, enabling seamless incorporation of new models to the pool and refinement of existing model embeddings without having to perform any retraining. We additionally train a correctness predictor that uses model embeddings and query encodings to achieve state-of-the-art routing accuracy on unseen queries. Experiments show that LOCUS needs up to 4.8x fewer query evaluation samples than baselines to produce informative and robust embeddings. Moreover, the learned embedding space is geometrically meaningful: proximity reflects model similarity, enabling a range of downstream applications including model comparison and clustering, model portfolio selection, and resilient proxies of unavailable models.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑