arXiv:2410.13569cs.LGcs.CV2024-10CVPR被引 7

用树状结构分析模型权重,实现无需文档的零样本模型搜索

Learning on Model Weights using Tree Experts

  • 利用模型间的树状演化关系,减少无关噪声干扰
  • 仅靠单层权重就能准确预测模型训练数据类别
  • 轻量级方法支持文本搜索模型,适合模型检索场景

公开模型数量迅速增长,但多数缺乏文档。用户需先判断模型功能,直接从权重推断文档极具挑战,因权重中存在与功能无关的噪声(称作扰动)。本文发现:大多数公开模型属于少数模型树,所有树内模型均由同一基础模型微调而来;且树内模型间扰动更小。具体而言,跨模型树需复杂架构,而树内仅用单层权重的线性分类器即可有效。尽管如此,此类方法计算开销大。为此,我们提出探针专家(ProbeX),一种理论驱动、轻量级的新方法,首次专为单隐藏层权重设计。实验表明,仅凭权重即可预测模型训练数据类别。令人兴奋的是,ProbeX可将Stable Diffusion权重映射至权重-语言嵌入空间,实现基于文本的模型搜索,即零样本模型分类。

原文摘要 · Abstract (English)

The number of publicly available models is rapidly increasing, yet most remain undocumented. Users looking for suitable models for their tasks must first determine what each model does. Training machine learning models to infer missing documentation directly from model weights is challenging, as these weights often contain significant variation unrelated to model functionality (denoted nuisance). Here, we identify a key property of real-world models: most public models belong to a small set of Model Trees, where all models within a tree are fine-tuned from a common ancestor (e.g., a foundation model). Importantly, we find that within each tree there is less nuisance variation between models. Concretely, while learning across Model Trees requires complex architectures, even a linear classifier trained on a single model layer often works within trees. While effective, these linear classifiers are computationally expensive, especially when dealing with larger models that have many parameters. To address this, we introduce Probing Experts (ProbeX), a theoretically motivated and lightweight method. Notably, ProbeX is the first probing method specifically designed to learn from the weights of a single hidden model layer. We demonstrate the effectiveness of ProbeX by predicting the categories in a model's training dataset based only on its weights. Excitingly, ProbeX can map the weights of Stable Diffusion into a weight-language embedding space, enabling model search via text, i.e., zero-shot model classification.

模型搜索权重分析零样本分类

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