arXiv:2410.10112cs.CVcs.CL2024-10ICML被引 6

用矩阵补全预测大模型跨任务表现,省钱省时。

Can We Predict Performance of Large Models across Vision-Language Tasks?

  • 将模型-任务表现建模为稀疏矩阵,用概率分解补全未知分数。
  • 预测准确率高,且能估计不确定性,指导优先测试高风险任务。
  • 适合想快速评估新模型性能的研究者和工程师。

大型视觉语言模型(LVLMs)的评估成本高昂,因计算开销大且任务种类繁多。若已有部分性能数据,可据此推断未知表现。本文提出一种基于概率矩阵分解(PMF)与马尔可夫链蒙特卡洛(MCMC)的框架,将性能预测建模为矩阵补全问题:构建稀疏性能矩阵 $oldsymbol{R}$,其中 $R_{mn}$ 表示第 $m$ 个模型在第 $n$ 个数据集上的得分。通过 MCMC 进行 PMF 推理,实现对缺失分数的预测,并估计预测不确定性。实践者可优先评估不确定性高的未测试任务,显著降低整体预测误差。进一步引入改进策略以应对观测数据稀疏场景。实验表明,该方法预测准确,不确定性估计可靠,且在稀疏条件下仍具有效性。

原文摘要 · Abstract (English)

Evaluating large vision-language models (LVLMs) is very expensive, due to high computational cost and the wide variety of tasks. The good news is that if we already have some observed performance scores, we may be able to infer unknown ones. In this study, we propose a new framework for predicting unknown performance scores based on observed ones from other LVLMs or tasks. We first formulate the performance prediction as a matrix completion task. Specifically, we construct a sparse performance matrix $\boldsymbol{R}$, where each entry $R_{mn}$ represents the performance score of the $m$-th model on the $n$-th dataset. By applying probabilistic matrix factorization (PMF) with Markov chain Monte Carlo (MCMC), we can complete the performance matrix, i.e., predict unknown scores. Additionally, we estimate the uncertainty of performance prediction based on MCMC. Practitioners can evaluate their models on untested tasks with higher uncertainty first, which quickly reduces the prediction errors. We further introduce several improvements to enhance PMF for scenarios with sparse observed performance scores. Our experiments demonstrate the accuracy of PMF in predicting unknown scores, the reliability of uncertainty estimates in ordering evaluations, and the effectiveness of our enhancements for handling sparse data.

模型评估矩阵补全不确定性跨任务预测

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