arXiv:2410.02223cs.CLcs.AI2024-10ICLR被引 48

用紧凑向量表征大模型,实现高效模型路由与性能预测。

EmbedLLM: Learning Compact Representations of Large Language Models

  • 设计编码器-解码器框架,学习大模型的紧凑向量表示。
  • 在模型路由任务中准确率与延迟均优于现有方法。
  • 无需额外推理开销即可预测模型在多个基准的表现。

如今 Hugging Face 上已有数十万种语言模型,高效评估和利用这些模型在各类下游任务中的表现变得愈发关键。现有许多方法需为每个任务重复学习特定表示,造成时间和计算资源浪费。为此,我们提出 EmbedLLM 框架,旨在学习大语言模型(LLMs)的紧凑向量表示,以支持涉及大量模型的下游应用,如模型路由。该框架采用编码器-解码器结构学习嵌入,并建立系统化评估体系。实验表明,EmbedLLM 在模型路由任务中不仅准确率更高,且延迟更低。此外,我们的方法可在不增加额外推理成本的前提下,预测模型在多个基准上的表现。广泛探针实验验证,所学嵌入能捕捉模型关键特性(如是否专精编程任务),即使未显式训练于此。我们已开源数据集、代码与嵌入器,以推动进一步研究与应用。

原文摘要 · Abstract (English)

With hundreds of thousands of language models available on Huggingface today, efficiently evaluating and utilizing these models across various downstream, tasks has become increasingly critical. Many existing methods repeatedly learn task-specific representations of Large Language Models (LLMs), which leads to inefficiencies in both time and computational resources. To address this, we propose EmbedLLM, a framework designed to learn compact vector representations, of LLMs that facilitate downstream applications involving many models, such as model routing. We introduce an encoder-decoder approach for learning such embeddings, along with a systematic framework to evaluate their effectiveness. Empirical results show that EmbedLLM outperforms prior methods in model routing both in accuracy and latency. Additionally, we demonstrate that our method can forecast a model's performance on multiple benchmarks, without incurring additional inference cost. Extensive probing experiments validate that the learned embeddings capture key model characteristics, e.g. whether the model is specialized for coding tasks, even without being explicitly trained on them. We open source our dataset, code and embedder to facilitate further research and application.

大模型表征模型路由嵌入学习

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。