Skip to content

Reference-Based Bias Detection in LLMs via Relative Representations of Hidden States

Sep 2026 · 0 citations · 35 references
Computer Science

TL;DR

A reference-based method that audits bias in hidden-state representations across related model variants across related model variants, for example before and after fine-tuning, which is complementary to output-based auditing rather than a replacement for it.

Abstract

Existing bias auditing methods typically rely on model outputs, requiring costly benchmarks or judge models and potentially missing internal shifts that never appear in generated text. We propose a reference-based method that audits bias in hidden-state representations across related model variants, for example before and after fine-tuning. Because fine-tuning reshapes representation geometry, absolute hidden states are not directly comparable, so we encode each sentence by its similarities to a fixed set of anchor sentences, yielding relative representations in a shared comparison space. There we measure how target groups shift in their association with positive and negative attributes, a quantity we call the Representational Bias Shift $\Delta B$. Across three model families and the WildGuardMix, DecodingTrust and ToxiGen benchmarks, $\Delta B$ correlates with output-level bias change in 15 of the 18 settings we test, reaching $|r| = 0.84$ ($p<0.001$) under full fine-tuning and becoming more model-dependent under parameter-efficient adaptation. Thresholding $\Delta B$ detects checkpoints whose bias increased with ROC AUC between $0.65$ and $0.99$, and on WildGuardMix and DecodingTrust it separates them better than a SEAT-based baseline for all three families. $\Delta B$ is also stable under changes to the anchor set, attribute sets and target templates. Our method requires no task-specific evaluation data and audits a model in about three minutes, using $3$-$50\times$ less compute than the output-level benchmarks considered here. We view it as complementary to output-based auditing rather than a replacement for it.

View source

Similar papers

Preprint Sep 2026

Reduced latent leakage does not reliably predict lower likelihood bias in collider inference

Reusable collider representations can be evaluated through downstream discrimination and probes of retained information, but neither quantity directly tests the behaviour of score templates in a profiled likelihood. We test a specific prediction in a controlled two-channel routing protocol: if reduced physics-label rea...

Tong-Sheng Pan · 0 citations
#natural language process... Preprint Oct 2026

HeadEdit: Calibrating Language Model Behavior Through the Frozen Unembedding Matrix

Alignment does not eliminate behavioral errors in language models. Models may still refuse benign requests, call unnecessary tools, or yield to false user claims. Current methods mitigate such errors as a computation problem, and rarely explore if the desired behavior is already encoded in the model's representation. M...

Zirui He, Hai-Yan Zhao, Jing-Yu Hu et al. · 0 citations
#machine learning Preprint Sep 2026

Towards Identifying the Dataset Biases Causing Phantom Transfer

This work embeds a dataset's completions with Sentence-BERT, subtract the embeddings of clean reference completions, and compares the result to an open vocabulary of candidate topics, observing that different teacher models appear to express the same bias through different vocabulary.

Jonas Jürß, Pietro Lió · 0 citations
#small language model Preprint Aug 2026

Phantom Gains: Auditing Self-Improvement Against a Measured Null

Auditing three rounds of rank-$32$ LoRA self-training on Qwen3-8B against a frozen control pushed through the identical pipeline, this work identifies seven measurement failures, each of which inverts a reported finding when its control is absent.

Cheng Xu, Nan Yan, Li-Ming Chen et al. · 4 citations
#artificial intelligence Preprint Sep 2026

Pinocchio: Fast Uncertainty Estimates for Black-Box Language Models

Pinocchio is introduced, an external calibrator that estimates the correctness of responses from black-box API models that needs only a single forward pass to generate an uncertainty estimate and requires no access to the target model's logits, weights, or internal states.

K. D. Hayes, Arka Pal, Hao-Song Zhang et al. · 1 citation

Related blog posts

We use cookies to run the site and, with your consent, for analytics and to show ads. See our Cookie Policy.