Reading fine-tuning bias without running outputs

How relative representations of hidden states turn fine-tuning bias shift into a three minute Delta B audit that complements output benchmarks.

Read the original paper
Cover image for 'Reading fine-tuning bias without running outputs'

Reference-Based Bias Detection in LLMs via Relative Representations of Hidden States is a paper published on arXiv under identifier 2609.10060. It claims that encoding hidden states as similarities to a fixed anchor set lets us measure how target groups move toward positive and negative attributes before and after fine-tuning in a single number called Delta B. The original paper is linked under the title.

Hello, this is Patrick.

I keep stopping at the same scene whenever a fine-tuning report lands on my desk. The loss curve slopes down nicely and the target task score goes up, yet a small unease stays behind, as if bias shifted a little somewhere while nobody watched. So how do we check that shift. The usual answer is to run a pile of output benchmarks or attach another judge model to read the answers. That costs time and compute, and repeating it at every checkpoint gets heavy fast. What interests me more in this paper is how it attacks exactly that spot. It proposes to skip waiting for outputs and compare the geometry inside the model directly.

The moment I pause before running another output benchmark

Anyone who has tuned a model has felt this discomfort. You run fine-tuning to polish customer support tone, and then the safety team asks for bias scores again. You run WildGuardMix, then DecodingTrust, then ToxiGen, and half a day is gone. With five checkpoints the cost multiplies by five. Output checks can only start after answers arrive. You pick prompts, sample generations, wait for decoding, then score everything, and that whole chain repeats every time. I found myself doubting that routine. Could we read what already happened inside, without collecting answers outside every time.

People learn in a similar way, if you think about it. Someone tries to fix their tone yet picks up a quiet prejudice along the way. Before their sentences turn rough, their distance toward some group hardens first. The inner distance moves before the outer sentence does. That is what this paper tries to catch. Before and after fine-tuning, it asks whether target groups moved closer to positive words or got pulled toward negative words inside the model. It reads direction one step before outputs appear. So does that idea actually hold. Let us follow that question.

Why you cannot just overlay the model before and after

At first glance the comparison sounds simple. Feed the same sentence into the model before fine-tuning, take the hidden state, feed it into the model after fine-tuning, take the hidden state, and subtract the two vectors. The paper says that direct comparison breaks, and this is its first gate. Fine-tuning twists the representation geometry itself, so absolute hidden states are not directly comparable.

I understood this part through a moved neighborhood. Suppose your old apartment was a ten minute walk from the station, and your new apartment is also a ten minute walk from the station. You would not say the two homes sit at the same location, because the neighborhood itself changed. The hidden space after fine-tuning is like that new neighborhood. The dimensions look the same in numbers, yet what a distance means inside has shifted. If you subtract absolute vectors, you cannot tell whether the gap comes from bias or from the whole space moving. So is there a way to speak that survives the move. The paper answers by borrowing reference points.

Speaking in distances to anchors when the neighborhood moves

The tool here is relative representation. You fix one set of anchor sentences, then you turn every sentence into a list of how similar it is to each anchor. If a sentence under test has cosine similarities like 0.12 and 0.45 to 200 anchors, that row of 200 numbers becomes its new address. You use the same anchor set in the model before fine-tuning and the same anchor set in the model after fine-tuning, so both addresses are written against the same reference frame and can be compared. That is the logic.

I think the more important result is the structure behind the numbers. The anchors act as interpreters. Even when the space bends, the pattern of relations to anchors stays comparatively steady, and that assumption carries the method. Of course the assumption is not free. If the anchor set leans one way, the interpretation leans too. If every anchor comes from IT industry text, medical or legal nuance arrives blurred. That is why the paper also checks whether results wobble when anchors change. I will return to that check later. First we should see how bias gets measured on top of those relative addresses.

How a shift between groups and attributes becomes one number

The paper defines bias as a relation between target groups and attributes. Target groups are the group expressions under test, while attributes are sets of positive words and negative words. On top of the relative space it measures whether target group sentences moved closer to the positive side or got dragged toward the negative side. The change in that closeness before and after fine-tuning is gathered into one value called Representational Bias Shift Delta B.

Picture a concrete scene. It is a Tuesday afternoon and a safety colleague walks over with a fine-tuned support model. She asks whether descriptions of some group now sit a little nearer to negative words than before. The old way would run hundreds of prompts and collect answers. With Delta B you encode the target group sentences plus the positive and negative attribute sentences, then compare their relative positions in the base model and the tuned model. If some group moved clearly toward the negative side, Delta B stretches in that direction. Instead of opening sentences one by one, you read the tilt of the whole space. So does that tilt point the same way as output bias. The paper does not leave the question hanging, and it validates Delta B side by side with output benchmarks.

Putting that number next to output scores in 18 settings

The team combined three model families with three output benchmarks called WildGuardMix, DecodingTrust, and ToxiGen, which gives 18 settings in total. In each setting they measured how often Delta B and output level bias change moved together, using correlation. The outcome was correlation in 15 of the 18 settings. Under full fine-tuning one case reached an absolute correlation of 0.84 with a p value below 0.001. A number that never looked at outputs still pointed the same way as measured bias change fairly often.

I read two things in that outcome at once. One is frequency and the other is strength. Fifteen out of eighteen suggests the method did not get lucky once. A value like 0.84 shows how high it can climb when conditions fit. Still, stopping there would make the reading shallow. Why did it miss in the other three settings. And what happens outside full fine-tuning. The paper does not dodge this, and it reports that results grew more model dependent under parameter efficient adaptation. In models touched lightly, the link between the inner signal and outer scores turned less stable. So I treat this method as a tool that cuts well under matching conditions, rather than a master key. Knowing where it grips and where it slips is the first step to using the paper well.

Why the signal wobbles in lightly adapted models

From a systems perspective, the next question is obvious. How many production teams still run full fine-tuning every time. When my team looks at a new AI architecture we do not stare at benchmark scores alone. We ask what actually changed and what cost structure that change creates in the system. Through the same lens, light adaptation touches few parameters and often only local layers. Then the whole representation does not swing much, and only some pathways get adjusted a little. In that case similarity patterns to anchors wobble less, and Delta B can lose resolution.

Model family differences add to this. Models with different structures and training histories pack information into hidden states in slightly different ways. In one family the distance to anchors may expose the bias direction cleanly, while in another family that signal may hide behind other factors. That matches the paper when it says the parameter efficient case turned more model dependent. So is this method only useful for full fine-tuning audits. I would not narrow it that far. It fits well as a fast first screen in a full fine-tuning line that mints checkpoints often, placed before the heavy benchmarks. Teams that use light adaptation should first calibrate the link between Delta B and output scores on their own model family. If you skip that calibration and trust one number alone, the tool turns into fortune telling.

Can a threshold turn this into checkpoint screening

Correlation alone does not make a tool usable. So the paper adds an experiment that draws a threshold on Delta B and picks out checkpoints with increased bias. Measured by ROC AUC the results ran from 0.65 to 0.99, and on WildGuardMix and DecodingTrust it beat a SEAT based baseline in all three families. The quieter mention of ToxiGen also caught my eye, because each benchmark tracks a different flavor of bias and detection difficulty likely shifts with it.

This is where I would be careful not to over-read the experiment. You need to feel the width of that range directly. A score of 0.99 means almost everything gets caught, while 0.65 sits only a little above chance. When one method spans that widely across conditions, a single fixed threshold will not serve every model. (For context, the SEAT based comparison belongs to a classic family of embedding association checks, and here it serves as a baseline measured without relative spaces.) Where AUC runs high, this approach looks ready to serve as a checkpoint gate. Where it runs low, I would keep it as a queue that decides which checkpoint deserves expensive output checks first. When full checks cost too much yet skipping checks feels unsafe, ranking suspects first is the honest use.

Checking whether the answer holds when anchors and wording change

A method that dances whenever anchors or wording change cannot survive production. The paper therefore reports that Delta B stayed steady when the anchor set, attribute sets, and target templates were varied. That stability note reads short, yet it connects directly to what I would watch next. A breakdown of which anchors hold steady and which ones break would make the user manual for this method much stronger.

Consider short declarative anchors only. Then the similarity values may miss the tone of long conversational sentences. Consider attributes built from strong words only. Then quiet tonal bias may slip past the number. Consider target templates skewed toward one sentence shape. Then the signal from layers sensitive to that shape may get overstated. We need to know the range behind the reported stability to sense what to watch when we port this to our domain. So how should we pick anchors. I would start by spreading domain sentences wide and mixing lengths, shapes, and speech acts so nothing leans one way. Then I would grow the anchor count and check where Delta B converges. How many interpreters you need is a number each deployment has to settle itself.

What a three minute audit really means

Another line that stayed with me is speed and cost. The approach needs no task specific evaluation data and audits a model in about three minutes, using 3 to 50 times less compute than the output benchmarks in the comparison. That is a wide range. A range from 3 to 50 means savings swing a lot with the baseline. Against heavy generative evaluation the savings look large, while against light classification checks they look smaller. So I read those numbers as an operating sense rather than a strict promise.

In operating terms the gap still matters more than it looks. Suppose five fine-tuning jobs run overnight and checkpoints pile up by morning. Full output checks on everything clog the GPU queue and delay release calls. If a three minute inner check nominates suspects first, the costly checks can focus on those candidates. You read what you can read, and you admit what you cannot. One thing needs care here. The felt time behind three minutes shifts with how anchor encoding and similarity work gets handled. Caching anchor embeddings ahead of time feels very different from computing everything from scratch each run. The structure matters more than the number itself, because it turns exhaustive checks into selective checks.

Where I would place this method in a real pipeline

Some readers may wonder whether output checks can go away now. The paper draws that line clearly. It views the method as complementary to output based auditing rather than a replacement for it. Inner distance and outer answers often point together, yet they are not the same thing. I find that modest placement more trustworthy.

From a systems perspective the picture looks like this. Put Delta B in the fine-tuning pipeline as a first gate. Run the anchor based inner check at every checkpoint, and send only those above threshold to output benchmarks. Set the threshold on your own model family and your own domain data. Start low at first, gather pairs with output scores, then tighten gradually. With light adaptation, find the layers and anchor mix where the mapping stays stable before anything else. On axes with a different flavor, like some ToxiGen splits, avoid solo calls and keep the signal as priority only. Used this way, the three minute audit stops sounding like magic and starts working as a practical tool that lets a team survive a morning full of overnight checkpoints.

So what belongs to people in the end. A number alone does not close the case, because someone still has to read which groups and which attributes produced it and decide whether release should stop. The faster we measure distances inside the model, the sharper that reading duty gets. Next time a fine-tuning report arrives, I will put Delta B next to output scores and look at both. Moments when the two disagree will intrigue me more than moments when they agree. Learning to read that mismatch may become the sense that separates audit skill going forward.

References

  1. Reference-Based Bias Detection in LLMs via Relative Representations of Hidden States · arxiv.org

    Reviewed source