Use a small assistant LLM to remove teacher-model favoritism from proxy judge training

May 25, 20257 min

Overview

Decision SnapshotNeeds Validation

The approach uses public assistant models and standard fine-tuning, so it's implementable with moderate engineering effort; evidence is several benchmarks and ablations reported in the paper.

Citations0

Evidence Strength0.70

Confidence0.80

Risk Signals9

Trust Signals

Findings with numeric evidence: 4/4

Findings with evidence refs: 4/4

Results with explicit delta: 3/3

Reproducibility

Status: Code + data available

Open source: Partial

At A Glance

Cost impact: 60%

Production readiness: 60%

Novelty: 50%

Authors

Zhuo Liu, Moxin Li, Xun Deng, Qifan Wang, Fuli Feng

Links

Abstract / PDF / Code

Why It Matters For Business

If you use proxy judge models to compare or benchmark LLMs, training them on a single strong teacher (like GPT-4) can skew results toward that teacher. AGDe-Judge offers a practical, low-cost way to reduce that skew while keeping evaluation accuracy.

Who Should Care

Summary TLDR

Proxy judge models (small open LLMs fine-tuned to evaluate other LLMs) inherit a 'teacher preference bias' when trained on evaluations produced by a powerful teacher model (e.g., GPT-4). The paper introduces AGDe-Judge: a three-stage, data-centric pipeline that (1) filters ambiguous teacher labels using an assistant model and an implicit reward margin, (2) uses an assistant to identify severe flaws and rewrite biased teacher feedback, and (3) fine-tunes the proxy judge on the cleaned data. Experiments across six benchmarks show AGDe-Judge greatly reduces favoritism toward teacher outputs while keeping or improving standard evaluation accuracy. Code is available.

Problem Statement

Training proxy judge models on evaluations produced by a single strong teacher model embeds that teacher's self-preference into the training data. The result is a proxy judge that unfairly favors responses produced by the teacher regardless of true quality. The paper asks: how to remove this teacher preference bias from both labels and textual feedback without human relabeling?

Main Contribution

Define and document teacher preference bias: proxies learn to favor teacher-model outputs from teacher-generated training data.

Propose AGDe-Judge, a three-stage data-centric debiasing pipeline using an additional assistant LLM to filter labels and refine feedback.

Key Findings

Proxy judges trained on GPT-4 labeled data show strong favoritism to GPT-4 outputs on bias tests.

NumbersOffsetBias accuracy: GPT4-trained proxy 0.182 vs GPT3.5-trained 0.269 (Table 2)

Practical UseIf you fine-tune a judge only on GPT-4 outputs, expect systematic over-ranking of GPT-4 responses; don't trust teacher-only training for fair comparisons.

Evidence RefTable 2, Section 3

AGDe-Judge substantially reduces teacher favoritism on the dedicated bias test.

NumbersOffsetBias accuracy: AGDe-Judge 0.391 vs GPT4-trained proxy 0.182 (Table 2)

Practical UseApply AGDe-Judge (assistant-guided label filtering + feedback revision) to cut teacher favoritism roughly two-fold on the tested bias set.

Evidence RefTable 2, Section 5

Results

MetricValueBaselineDeltaSplit / DatasetEvidenceEvidence Ref
OffsetBias (lower bias = higher score)AGDe-Judge 0.391GPT4-trained proxy 0.182+0.209OffsetBias (filtered)AGDe-Judge bested GPT4-trained proxy; numbers from Table 2Table 2
AccuracyAGDe-Judge 0.832GPT4-trained proxy 0.721+0.111Preference-BenchTable 2 shows AGDe-Judge improved accuracy on in-domain preference benchmarkTable 2

What To Try In 7 Days

Run a bias check: compute 'win rates' on a subset comparing teacher outputs vs others.

Add a smaller assistant LLM (e.g., GPT-3.5-Turbo) and filter teacher labels where assistant disagrees.

Use the paper's Stage 2 prompts to let the assistant flag severe flaws and rewrite teacher feedback before finetuning.

Optimization Features

Training Optimization
label filtering via implicit reward marginfeedback refinement with assistant-generated critiques

Reproducibility

Code AvailableYes
Data AvailableYes
Open Source StatusPartial
LicenseUnknown

Risks & Boundaries

Limitations

All experiments used GPT-4 as the teacher; behavior with other teacher models is untested.

AGDe-Judge focuses on data fixes; model-architecture or training-level solutions are not explored.

When Not To Use

If you cannot run or afford a separate assistant LLM for labels and critique.

If your training data is already human-labeled and free of teacher-generated bias.

Failure Modes

Filtering threshold too high may remove too much data and hurt training (paper shows sensitivity).

Assistant model could introduce its own blind spots if it holds subtle biases.

Core Entities

Models

GPT-4GPT-3.5-TurboMistral-7B-Instruct-v0.3Llama-2-7b-hfQwen2.5-7B-Instruct

Metrics

Accuracywin ratioimplicit reward margin

Datasets

AutoJ-pairwisePreference CollectionOffsetBias (filtered)MT-Bench Human Judge (filtered)Preference-BenchReward-BenchArena-HumanUltraFeedback BinarizedJudgeLM

Benchmarks

Preference-BenchMT-BenchReward-BenchArena-HumanUltraFeedbackJudgeLMOffsetBias