atlas

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

Prometheus: Inducing Fine-Grained Evaluation Capability in Language Models

ICLR 2024 · Large language models › Language, safety and interpretability · approach: Statistical frameworks for LLM evaluation · 14 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.

Introduces the Feedback Collection dataset (1K customized score rubrics, 100K GPT-4 feedback/score pairs) and fine-tunes Llama-2-Chat into Prometheus, an open-source 13B fine-grained evaluator that matches GPT-4 correlation with humans.

Problem

Proprietary LLMs like GPT-4 are the de facto evaluators for long-form text, but they are closed-source, have uncontrolled versioning, and are costly; open models lack fine-grained, customized rubric-based evaluation ability.

Approach

Train an open evaluator LM on a large dataset of GPT-4-generated feedback conditioned on thousands of customized score rubrics, reference answers, and balanced 1-5 scores, so that reference materials (rubric plus reference answer) enable GPT-4-level absolute grading.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Fine-tune Llama2-Chat 13B on customized-rubric feedback dataLlama2-Chat 13B · Feedback Bench (seen and unseen rubrics)Pearson correlation with GPT-4-0613: +0.420 and +0.397Helped
Fine-tune Llama2-Chat 13B on customized-rubric feedback dataLlama2-Chat 13B · Vicuna Bench, MT Bench, and Flask EvalPearson correlation with GPT-4-0613: +0.255, +0.493, and +0.202Helped
Fine-tune Llama2-Chat 13B on fine-grained feedback instead of coarse-grained rubricsLlama2-Chat 13B · HHH Alignment and MT Bench Human Judgmentaccuracy: +5.43% and +5.38%Helped
Add the score rubric (ablation)Prometheus 7B · Vicuna BenchPearson correlation: 0.355 -> 0.457Helped
Add the score rubric (ablation)Prometheus 7B · Feedback Bench (seen and unseen score rubrics)Pearson correlation: 0.837 -> 0.860; 0.745 -> 0.847Helped
Add feedback distillation (ablation)Prometheus 7B · Feedback Bench (seen and unseen score rubrics)Pearson correlation: 0.668 -> 0.860; 0.673 -> 0.847Helped
Add feedback distillation (ablation)Prometheus 7B · Vicuna BenchPearson correlation: 0.413 -> 0.457Helped
Add the reference answer (ablation)Prometheus 7B · Feedback Bench (seen and unseen score rubrics)Pearson correlation: 0.642 -> 0.860; 0.626 -> 0.847Helped
Add the reference answer (ablation)Prometheus 7B · Vicuna BenchPearson correlation: 0.349 -> 0.457Helped
Train on coarse-grained Flask rubrics instead of customized rubrics (ablation)Llama2-Chat 13B · Flask EvalPearson correlation with GPT-4-0613: 0.467 -> 0.517Helped
Train on coarse-grained rubrics (ablation)Llama2-Chat 13B · HHH Alignment and MT Bench Human Judgmentaccuracy: 73.76 -> 67.42; 52.34 -> 46.89Hurt
Use Llama-2-Chat instead of Llama-2 as the starting model (ablation)Prometheus 7B · Feedback Bench (seen and unseen score rubrics)Pearson correlation: 0.839 -> 0.860; 0.818 -> 0.847No clear effect

Surprising findings

Limitations

Experimental setup

Datasets: Feedback Collection FLASK

Benchmarks: Feedback Bench Vicuna Bench MT Bench FLASK Eval MT Bench Human Judgments HHH Alignment

Metrics: Pearson Kendall-Tau Spearman accuracy pairwise win rate

Baselines: GPT-4-0613 GPT-3.5-Turbo-0613 Llama-2-Chat 70B StanfordNLP Reward Model ALMOST Reward Model Llama2-Chat 13B + Coarse

Builds on: Llama-2-Chat chain-of-thought fine-tuning FLASK

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