The Alignment Tax: Response Homogenization in Aligned LLMs and Its Implications for Uncertainty Estimation
Organizations: Independent Researcher
Abstract
RLHF-aligned language models exhibit response homogenization: on TruthfulQA (n=790), 40-79% of questions produce a single semantic cluster across 10 i.i.d. samples. On affected questions, sampling-based uncertainty methods have zero discriminative power (AUROC=0.500), while free token entropy retains signal (0.603). This alignment tax is task-dependent: on GSM8K (n=500), token entropy achieves 0.724 (Cohen's d=0.81). A base-vs-instruct ablation confirms the causal role of alignment: the base model shows 1.0% single-cluster rate vs. 28.5% for the instruct model (p < 10^{-6}). A training stage ablation (Base 0.0% -> SFT 1.5% -> DPO 4.0% SCR) localizes the cause to DPO, not SFT. Cross-family replication on four model families reveals alignment tax severity varies by family and scale. We validate across 22 experiments, 5 benchmarks, 4 model families, and 3 model scales (3B-14B), with Jaccard, embedding, and NLI-based baselines at three DeBERTa scales (all ~0.51 AUROC). Cross-embedder validation with two independent embedding families rules out coupling bias. Cross-dataset validation on WebQuestions (58.0% SCR) confirms generalization beyond TruthfulQA. The central finding -- response homogenization -- is implementation-independent and label-free. Motivated by this diagnosis, we explore a cheapest-first cascade (UCBD) over orthogonal uncertainty signals. Selective prediction raises GSM8K accuracy from 84.4% to 93.2% at 50% coverage; weakly dependent boundaries (|r| <= 0.12) enable 57% cost savings.
Figures & tables
| Method | Boundary | Cost | Cascade Role |
|---|---|---|---|
| Token entropy (ours) | B1 Fluency | Free | Stage 1 (always-on) |
| LogTokU / PRO ( Ma et al., 2025b ; Chen et al., 2025c ) | B1 Fluency | Free | Stage 1 (drop-in) |
| Semantic Energy ( Ma et al., 2025a ) | B1 Fluency | samples | Single-cluster remedy |
| SINdex ( Abdaljalil et al., 2025 ) | B1 Fluency | samples | Stage 1 (escalation) |
| Semantic Entropy ( Kuhn et al., 2023 ) | B1 Fluency | 5–10 | Stage 1 (escalation) |
| CoCoA ( Huang et al., 2026 ) | B1 Fluency | 1–5 | Stage 1 (escalation) |
| # | Hypothesis | Data | Key Result | Status |
|---|---|---|---|---|
| 1 | B1 domain specificity | TruthfulQA 790q | CV AUC=0.658 (eff.) / 0.395 (blind) | ✓ |
| 2 | B1–B2 independence | TruthfulQA 401q | =0.119, dcor=0.143 | ✓ |
| 3 | Cascade parallel | TruthfulQA 401q | =0.498, 57.4% cost saving | ✓ |
| 4 | Cross-model stability | 3 models 790q | 3B AUC=0.676 14B=0.537 | ✓ |
| 5 | B3 freshness decay | FreshQA 1500q | 11–13 acc. drop; B1–B3 = 0.067 | ✓ |
| 6 | Label robustness | LLM-judge | B1: 0.571 0.599 (cross-family) | ✓ |
| Model | Effective% | Blind% | Eff. AUC | Overall |
|---|---|---|---|---|
| LLaMA-3.2-3B | 79% | 21% | 0.676 | 0.622 |
| Qwen3-4B | 50% | 50% | 0.625 | 0.537 |
| Qwen3-14B | 36% | 64% | 0.537 | 0.490 |
| Judge | Correct% | B1 AUC | 95% CI |
|---|---|---|---|
| Word-overlap | 25.9% | 0.571 | [0.526, 0.617] |
| LLaMA-3.2-3B (cross family) | 54.6% | 0.599 | [0.563, 0.634] |
| Feature | AUROC | 95% CI | Cost |
|---|---|---|---|
| B1 mean entropy | 0.706 | [.635,.772] | Free |
| B1 std entropy | 0.715 | [.643,.782] | Free |
| B1 max entropy | 0.724 | [.650,.793] | Free |
| Combined entropy (4 feat, CV) | 0.724 .033 | — | Free |
| P(True) | 0.608 | [.52,.70] | 1 call |
| Response length (tokens) † | 0.849 | [.791,.903] | Free |
| Method | AUROC | 95% CI | Cost |
|---|---|---|---|
| B1 mean entropy | 0.599 | [0.559, 0.637] | Free |
| SelfCheck-Emb ( =5) | 0.588 | [0.547, 0.626] | 6 |
| SE-Jaccard ( =10) | 0.548 | [0.510, 0.589] | 11 |
| SE-Embedding ( =10) | 0.542 | [0.513, 0.572] | 11 |
| SE-NLI † (DeBERTa-large) | 0.511 | [0.419, 0.594] | 11 +NLI |
| SE-NLI † (DeBERTa-base) | 0.512 | [0.421, 0.593] | 11 +NLI |
| Metric | Base | Instruct | Difference |
|---|---|---|---|
| Single-cluster rate | 1.0% | 28.5% | +27.5pp |
| Mean clusters | 9.26 | 3.58 | 5.68 |
| Mean SE | 2.158 | 0.832 | 1.326 |
| Wilcoxon signed-rank (base instruct): =18,331, | |||
| Model | SCR | Mean NC | Wilcoxon vs. Qwen |
|---|---|---|---|
| Qwen3-14B-Instruct | 28.5% | 3.58 | — |
| LLaMA-3.2-3B-Instruct | 5.5% | 7.27 | |
| Mistral-7B-Instruct | 1.0% | 7.87 | |
| Qwen3-14B-Base | 1.0% | 9.26 |
| Strategy | Parameters | SCR | Mean NC | Mean SE |
|---|---|---|---|---|
| Baseline | =1.0 | 28.5% | 3.58 | 0.832 |
| Nucleus | =1.0, =0.9 | 33.5% | 3.40 | 0.786 |
| Nucleus | =1.0, =0.95 | 30.0% | 3.55 | 0.827 |
| Low temp | =0.7 | 30.0% | 2.96 | 0.668 |
| Stage | Model | SCR | Mean NC | Mean SE |
|---|---|---|---|---|
| Base | Mistral-7B-v0.1 | 0.0% | 9.28 | 2.170 |
| SFT | mistral-7b-sft-beta | 1.5% | 8.63 | 2.024 |
| SFT+DPO | zephyr-7b-beta | 4.0% | 8.01 | 1.897 |
| Wilcoxon: Base SFT =0.002, SFT DPO =0.0001, Base DPO | ||||
| max_tokens | SCR | Mean NC | Mean SE | Avg Words |
|---|---|---|---|---|
| 40 | 32.0% | 3.02 | 0.676 | 26.9 |
| 100 | 10.0% | 7.46 | 1.769 | 68.8 |
| 200 | 8.0% | 8.30 | 1.929 | 115.2 |
| Base model SCR 0% at all lengths. 5/16 single-cluster questions persist across all settings. | ||||
| Stage | Model | SCR | Mean NC | Mean SE |
|---|---|---|---|---|
| Base | Llama-3.1-8B | 0.0% | 9.46 | 2.209 |
| SFT | Tulu-3-8B-SFT | 0.0% | 9.02 | 2.123 |
| SFT+DPO+RLVR | tulu3-8b | 0.5% | 9.31 | 2.174 |
| Wilcoxon: Base SFT =0.00004, SFT DPO =0.008, Base DPO =0.43 | ||||
| Mean J | [email protected] | [email protected] | Mean [email protected] | |
|---|---|---|---|---|
| Q4_K_M (4-bit) | 0.608 | 63.3% | 6.7% | 7.67 |
| Q8_0 (8-bit) | 0.576 | 26.7% | 6.7% | 7.30 |
| (Q8 Q4) | 0.032 | 36.6pp | 0.0pp | 0.37 |
| Embedder | [email protected] | [email protected] | [email protected] |
|---|---|---|---|
| Qwen3-Embedding (1.5B) | 94.0% | 78.0% | 14.0% |
| Nomic-embed-text (137M) | 98.0% | 92.0% | 52.0% |
| Per-question cluster-count Pearson =0.033 at =0.85; both detect single-cluster collapse. | |||
| Setting | [email protected] | [email protected] | [email protected] |
|---|---|---|---|
| 40 tokens, 200q (Exp. 12) | 79.0% | 79.0% | — |
| 200 tokens, 200q (this exp.) | 61.5% | 33.5% | 14.0% |
| Dataset | [email protected] | [email protected] | [email protected] |
|---|---|---|---|
| TruthfulQA 200q (100tok) | 79.0% | 79.0% | — |
| WebQuestions 200q (100tok) | 77.5% | 58.0% | 34.0% |
| Method | AUROC | vs. Jaccard | Cost |
|---|---|---|---|
| NLI-SE (cross-encoder) | 0.639 | +0.014 | 11 +NLI |
| Jaccard-SE | 0.625 | — | 11 |
| 95% bootstrap CIs overlap ( =+0.014 crosses zero). | |||
| Stage | Model | Mean (nats) | Std | Retention |
|---|---|---|---|---|
| Base | Mistral-7B-v0.1 | 1.175 | 0.312 | 100% |
| SFT | mistral-7b-sft-beta | 1.055 | 0.302 | 89.8% |
| SFT+DPO | zephyr-7b-beta | 0.776 | 0.179 | 66.0% |
| Kruskal-Wallis =183.89, =1.17 ; all pairwise Mann-Whitney . | ||||
| Benchmark | Method | AUROC | 95% CI | Cost |
|---|---|---|---|---|
| TruthfulQA | B1 entropy | 0.599 | [0.559, 0.637] | Free |
| (factual QA) | SelfCheck ( =5) | 0.588 | [0.547, 0.626] | 6 |
| Semantic entropy ( =10) | 0.548 | [0.510, 0.589] | 11 | |
| P(True) | 0.427 | [0.408, 0.446] | 1 call | |
| FreshQA | B1 entropy | 0.657 | [0.610, 0.703] | Free |
| (temporal) | P(True) | 0.399 | [0.366, 0.432] | 1 call |
| Pair | Pearson | dcor | HSIC | MI (bits) | Source | |
| B1–B2 | 0.119* | 0.143* | 0.020* | 0.008 | 401 | Exp. 2 |
| B1–B3 | 0.067 | — | — | 0.015 | 500 | Exp. 5 |
| B1–B4 | 0.054 | 0.072 | 0.252 | 0.006 | 790 | Exp. 10 |
| B1–B5 | 0.070 | — | — | 0.012 | 790 | Exp. 7 |
| B2–B4 | 0.086 | 0.156* | 0.000* | 0.003 | 401 | Exp. 10 |
| * . Effect sizes remain small ( , dcor ); superadditive coverage holds approximately. | ||||||
Appendix figures & tables1 asset
Supplementary material from the paper’s appendix.
Appendix
| Comparison | Exp | Effect | -value | Test |
|---|---|---|---|---|
| B1 vs SE-Jaccard AUROC | 12 | =+0.051 | 0.040* | DeLong+Holm |
| B1 vs SE-Embed AUROC | 12 | =+0.057 | 0.033* | DeLong+Holm |
| B1 vs SelfCheck AUROC | 12 | =+0.011 | 0.65 ns | DeLong |
| B1 vs NLI-SE (large) | 12 | =+0.088 | [0.419,0.594] | Bootstrap CI |
| Base vs Instruct SCR | 13 | 1.0% vs 28.5% | Wilcoxon | |
| Base→SFT (Zephyr) | 16 | NC= 0.64 | 0.002 | Wilcoxon |