跳到正文
arXiv cs.AI· Mohammad Zangooei, Peyvand Teymoori, Mohammad Robati Shirzad, Soroosh Shahtalebi·· 1 天前AI 评分31

CPA:反事实成对增强提升小型 pointwise LLM 评判器

Counterfactual Pairwise Augmentation Improves Small Pointwise LLM Judges

AI 导读

CPA(Counterfactual Pairwise Augmentation)通过为每个 pointwise 样本合成并审核同查询下标签相反的反事实回答,将接受对框定为成对任务,联合微调后仅以 pointwise 模式部署。

正文 · 原文

View PDF HTML (experimental)

Abstract:Agentic systems increasingly gate individual steps with a pointwise judge that scores a response against a criterion. Small language models are attractive for this role because they can be served and specialized efficiently. Such a judge must locate the criterion's decision boundary, but pointwise labels under-determine that boundary in fine-tuning: labeled responses to different queries differ along many dimensions at once, so they do not reveal which features decide the label. We introduce CPA (Counterfactual Pairwise Augmentation), which makes the deciding difference explicit in training. For each pointwise item, CPA synthesizes and audits a counterfactual response to the same query intended to carry the opposite label. It frames the accepted pair as a pairwise task asking which of the two responses satisfies the criterion. The judge is fine-tuned jointly on both pointwise and pairwise datasets and deployed in pointwise mode only. A second-order analysis shows when this should outperform the prevailing strategy of merging separately trained pointwise and pairwise experts: merging moves every parameter direction toward the pairwise expert by the same fraction, whereas joint training can hold the pointwise optimum wherever the pairwise loss is flat and absorb its curvature where it is sharp. This is the regime that CPA's same-source, counterfactually minimal pairwise view is designed to produce. On separate Qwen3-8B LoRA adapters for six binary criteria, CPA improves criterion-averaged pointwise macro-F1 by 2.5 points over pointwise-only training and outperforms model merging on all six criteria at every tested merge weight. It also surpasses the strongest zero-shot frontier judge we evaluate, Gemini 3.6 Flash, by 11.8 points.
Comments: Accepted at Neurips'26 workshop of small language models for agentic systems
Subjects: Artificial Intelligence (cs.AI)
Cite as: arXiv:2610.41149 [cs.AI]
  (or arXiv:2610.41149v1 [cs.AI] for this version)
  https://doi.org/10.48550/arXiv.2610.41149

arXiv-issued DOI via DataCite (pending registration)

Submission history

From: Mohammad Zangooei [view email]
[v1] Fri, 30 Oct 2026 02:12:30 UTC (48 KB)

来源:arXiv cs.AI · arxiv.org