atlas

Research map / Large language models / Language, safety and interpretability

Smoothie: Label Free Language Model Routing

NeurIPS 2024 · Large language models › Language, safety and interpretability · approach: Uncertainty quantification for LLMs · 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.

Smoothie performs label-free LLM routing by estimating sample-conditional LLM quality scores from embedding distances between LLM outputs, using a Gaussian weak-supervision model with kernel smoothing over nearest neighbors.

Problem

Choosing which LLM to route each input sample to when no labeled validation data is available, in multi-task settings where LLM quality varies across samples.

Approach

Adapts a weak-supervision multivariate Gaussian latent-variable model over SentenceBERT embeddings of (input, LLM output) pairs; closed-form estimates of each LLM's quality score come from pairwise embedding distances, and nearest-neighbor kernel smoothing makes the scores sample-conditional (Smoothie-Local).

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use Smoothie global rather than random model selectionLLM selection · GSM8Kaccuracy: 28.4 -> 37.5Helped
Use Smoothie global rather than random selectionLLM selection · MixInstructChatGPT-rank: 5.96 -> 3.91Helped
Use sample-conditional local scoring instead of global scoringSmoothie routing · Multi-task distributionsaccuracy and ROUGE-2Helped
Estimate routing scores from held-out training examples rather than test-set generations (ablation)Smoothie global · NLG taskstask performanceMixed
Estimate routing scores from held-out training examples, with neighborhood size 20 (ablation)Smoothie local · Multi-task distributionsaccuracy and ROUGE-2Hurt
Increase neighborhood size above 1 (ablation)Smoothie local · Multi-task distributionsaccuracy and ROUGE-2Hurt
Replace SentenceBERT embeddings with BGE-small embeddings (ablation)Smoothie local · Multi-task distributionsaccuracy and ROUGE-2: 58.7 -> 59.3; 20.2 -> 19.7; 75.0 -> 74.6; 26.9 -> 27.1Mixed
Use Smoothie global to select the best model from varied ensemblesSmoothie global · 350 NLG ensemble-task settingsbest-model identification rate: 211 of them (60.2%)Helped
Use Smoothie local rather than the best individual model on averageMulti-task routing · Mix-task distributionsgeneration qualityHelped

Surprising findings

Limitations

Experimental setup

Datasets: CNN/DailyMail XSum SQuAD TriviaQA E2E WebNLG LegalBench Definition Extraction AlpacaEval MixInstruct GSM8K

Benchmarks: CNN/DailyMail XSum SQuAD TriviaQA E2E WebNLG LegalBench Definition Extraction AlpacaEval MixInstruct GSM8K

Metrics: Rouge2 accuracy win-rate length-controlled win-rate ChatGPT-rank Spearman correlation solve rate

Baselines: Random Best-on-Val Labeled-KNN PairRM Best individual model

Builds on: weak supervision (Gaussian model) SentenceBERT

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