Collaborative Performance Prediction for Large Language Models

Qiyuan Zhang, Fuyuan Lyu, Xue Liu, Chen Ma
7/1/2024
cs.CLcs.AIcs.LG

No official code found. Best implementation: Don-Joey/CPP_LLM (3 stars, last commit 2024-07-04).

Abstract

Comprehensively understanding and accurately predicting the performance of large language models across diverse downstream tasks has emerged as a pivotal challenge in NLP research. The pioneering scaling law on downstream works demonstrated intrinsic similarities within model families and utilized such similarities for performance prediction. However, they tend to overlook the similarities between model families and only consider design factors listed in the original scaling law. To overcome these limitations, we introduce a novel framework, Collaborative Performance Prediction (CPP), which significantly enhances prediction accuracy by leveraging the historical performance of various models on downstream tasks and other design factors for both model and task. We also collect a collaborative data sourced from online platforms containing both historical performance and additional design factors. With the support of the collaborative data, CPP not only surpasses traditional scaling laws in predicting the performance of scaled LLMs but also facilitates a detailed analysis of factor importance, an area previously overlooked.

View on arXivView PDF

Code Implementations(1)

Don-Joey/CPP_LLMCommunity100%
30Python, Jupyter NotebookMar 3, 20242 years ago

Cite this paper

@article{zhang2024collaborative,
  title  = {Collaborative Performance Prediction for Large Language Models},
  author = {Qiyuan Zhang and Fuyuan Lyu and Xue Liu and Chen Ma},
  year   = {2024},
  eprint = {2407.01300},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2407.01300v2}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2407.01300.svg)](https://codeofpaper.com/paper/2407.01300)

Discussion