atlas

Research map / Large language models / Reasoning and chain of thought

On Speeding Up Language Model Evaluation

ICLR 2025 · Large language models › Reasoning and chain of thought · approach: Routing, pruning and distillation for reasoning · 0 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes UCB-E and UCB-E-LRF, bandit-based adaptive selection of (method, validation example) pairs with low-rank matrix completition, to identify the best LLM prompt/config using a small fraction of the full evaluation budget.

Problem

Identifying the best LLM-based method (prompt, model, or hyperparameter setting) across a validation set is combinatorially expensive and costly in LLM inference and LLM-judge calls under a fixed evaluation budget.

Approach

Treat method selection as fixed-budget best-arm identification: UCB-E picks the method with the highest upper confidence bound and a random unseen example; UCB-E-LRF additionally fills missing scores with an ensemble low-rank factorization and selects the example with highest predictive uncertainty.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use active selection with UCB-E or UCB-E-LRF instead of non-active baselinesBest-method identification · AlpacaEval (Drop Annotator)evaluation budget to reach precision 1: 8% vs 80-90%Helped
Use UCB-E rather than UCB-E-LRFBest-method identification · AlpacaEvalevaluation budget: 2-3%Helped
Use UCB-E-LRF rather than UCB-EBest-method identification · GSM8K Prompts and PIQA Promptsevaluation budget to achieve higher precision: about 10% in absolute budgetHelped
Remove uncertainty matrix R from UCB-E-LRF (Score Only) (ablation)UCB-E-LRF · GSM8K Promptstop 1 precisionHurt
Reduce batch size from 32 to 1 (ablation)UCB-E-LRF · Harder datasetsperformanceHelped
Increase batch size from 32 to 128 (ablation)UCB-E-LRF · DatasetsperformanceHurt
Increase batch size from 32 to 1 (ablation)UCB-E-LRF · Datasetstime to achieve precision 1: almost 16 timesHurt
Reduce ensemble size to 4 (ablation)UCB-E-LRF · Hard datasetsperformanceHurt
Vary ensemble size (ablation)UCB-E-LRF · AlpacaEval (Drop Annotator)performanceNo clear effect
Use ensemble size larger than 64 (ablation)UCB-E-LRF · DatasetsperformanceNo clear effect
Vary uncertainty scaling from 0.1 to 5 (ablation)UCB-E-LRF · Two evaluated datasetsperformance: 0.1 to 5No clear effect
Reduce warm-up budget to 0.5% (ablation)UCB-E-LRF · Two evaluated datasetsprecision as more data are evaluated: 0.5%Hurt

Surprising findings

Limitations

Experimental setup

Datasets: AlpacaEval GSM8K PIQA

Benchmarks: AlpacaEval GSM8K PIQA

Metrics: top-1 precision (performance gap) top-1 precision (McNemar significance) NDCG@10

Baselines: Full Evaluation Row Mean Imputation Filled Subset LRF

Builds on: UCB-E successive elimination alternating least squares AlpacaEval

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy