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

May 25, 20257 min

Overview

Production Readiness

0.6

Novelty Score

0.5

Cost Impact Score

0.6

Citation Count

0

Authors

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

Links

Abstract / PDF

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.

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.

Empirically show AGDe-Judge reduces teacher favoritism and preserves or improves evaluation accuracy on six benchmarks.

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)

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)

AGDe-Judge preserves or improves standard benchmark accuracy while debiasing.

NumbersPreference-Bench accuracy: AGDe-Judge 0.832 vs GPT4-trained proxy 0.721; MT-Bench: 0.715 vs 0.700 (Table 2)

Both label filtering and feedback refinement are necessary for best results.

NumbersOffsetBias: Label-only 0.274, Feedback-only 0.335, AGDe-Judge 0.391 (Table 3)

Results

OffsetBias (lower bias = higher score)

ValueAGDe-Judge 0.391

BaselineGPT4-trained proxy 0.182

Accuracy

ValueAGDe-Judge 0.832

BaselineGPT4-trained proxy 0.721

Accuracy

ValueAGDe-Judge 0.715

BaselineGPT4-trained proxy 0.700

Who Should Care

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 margin
  • feedback refinement with assistant-generated critiques

Reproducibility

Code Available

Data Available

Open Source Status

  • partial

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.
  • The root causes of teacher self-preference bias are not fully analyzed in the paper.

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.
  • When teacher models other than GPT-4 dominate and their bias patterns differ (untested).

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.
  • Prompting choices for feedback revision can change outcomes; naive concatenation underperforms.

Core Entities

Models

  • GPT-4
  • GPT-3.5-Turbo
  • Mistral-7B-Instruct-v0.3
  • Llama-2-7b-hf
  • Qwen2.5-7B-Instruct

Metrics

  • Accuracy
  • win ratio
  • implicit reward margin

Datasets

  • AutoJ-pairwise
  • Preference Collection
  • OffsetBias (filtered)
  • MT-Bench Human Judge (filtered)
  • Preference-Bench
  • Reward-Bench
  • Arena-Human
  • UltraFeedback Binarized
  • JudgeLM

Benchmarks

  • Preference-Bench
  • MT-Bench
  • Reward-Bench
  • Arena-Human
  • UltraFeedback
  • JudgeLM
  • OffsetBias