Isolating a Language Model's Judgment with a Label-Free Subspace
Organizations: Korea Advanced Institute of Science and Technology (KAIST)
Abstract
Language models are often asked to make judgments: whether a proposed answer is correct, whether one statement follows logically from another, or whether a response is safe. Yet changing the instructions that specify how the judgment should be expressed can make the final answer less reliable, even for the same input. We ask whether this reflects a change in internal judgment or a failure to express it. Comparing hidden-state vectors for the same input under different instructions, we find that their differences concentrate in a low-dimensional expression subspace. We identify this subspace without benchmark labels and remove its contribution from a linear predictor to obtain an isolated judgment. Across correctness, entailment, and safety, the isolated judgment remains reliable under instructions not used to identify the subspace, even when the model's final answer and the original predictor degrade. When the judgment question is negated, isolation raises the predictor's AUROC from 0.504 to 0.825 for entailment and from 0.480 to 0.891 for safety. We also find that judgment is readable in earlier layers, before the expression subspace begins to disrupt the predictor. Together, these results provide a way to separate judgment from expression and recover internal judgments even when the model's final answers are unreliable.
Figures & tables
| Component | Setting |
| Judgments | Correctness, entailment, safety |
| Datasets | GSM8K, MMLU, ARC-Challenge, TriviaQA, ANLI, BeaverTails |
| Models | Qwen2.5-7B, Llama-3.1-8B, Mistral-7B, Qwen2.5-14B, Gemma-2-9B |
| Response formats | Yes/No, True/False, A/B, B/A, 1/0, Correct/Incorrect, sentence |
| Held-out conditions | No/Yes, False/True, 0/1, negated question |
| Position | AUROC |
| Before judgment question | 0.741 |
| After judgment question | 0.812 |
| Before verdict | 0.818 |
| Condition | Verdict | Readout | Isolated | Condition-fitted |
| No means correct | 0.617 | 0.640 | 0.781 | 0.805 |
| False means correct | 0.593 | 0.665 | 0.787 | 0.803 |
| 0 means correct | 0.769 | 0.787 | 0.809 | 0.810 |
| Negated question | 0.665 | 0.650 | 0.802 | 0.805 |
Appendix figures & tables22 assets
Supplementary material from the paper’s appendix.
Appendix
| cell | Yes/No own | frozen A/B | A/B fitted | frozen B/A | B/A fitted | frozen sentence | sentence fitted |
| Llama-3.1-8B ARC | 0.737 | 0.703 | 0.695 | 0.714 | 0.702 | 0.730 | 0.731 |
| Llama-3.1-8B GSM8K | 0.966 | 0.971 | 0.970 | 0.972 | 0.972 | 0.952 | 0.961 |
| Llama-3.1-8B MMLU | 0.719 | 0.679 | 0.672 | 0.688 | 0.678 | 0.708 | 0.699 |
| Llama-3.1-8B TriviaQA | 0.842 | 0.846 | 0.843 | 0.845 | 0.842 | 0.844 | 0.846 |
| Mistral-7B ARC | 0.730 | 0.662 | 0.696 | 0.649 | 0.670 | 0.685 | 0.690 |
| Mistral-7B GSM8K | 0.826 | 0.814 | 0.822 | 0.812 | 0.821 | 0.829 | 0.827 |
| cell | stem | response position |
| Llama-3.1-8B ARC | 0.691 | 0.746 |
| Llama-3.1-8B GSM8K | 0.967 | 0.971 |
| Llama-3.1-8B MMLU | 0.683 | 0.710 |
| Llama-3.1-8B TriviaQA | 0.810 | 0.827 |
| Mistral-7B ARC | 0.694 | 0.717 |
| Mistral-7B GSM8K | 0.776 | 0.788 |
| cell | rank | direction norm in subspace | readout, B/A | isolated, B/A | readout, negated | isolated, negated |
| Llama-3.1-8B ARC | 7 | 0.524 | 0.714 | 0.729 | 0.718 | 0.732 |
| Llama-3.1-8B GSM8K | 9 | 0.473 | 0.972 | 0.972 | 0.944 | 0.965 |
| Llama-3.1-8B MMLU | 6 | 0.277 | 0.688 | 0.692 | 0.664 | 0.703 |
| Llama-3.1-8B TriviaQA | 8 | 0.504 | 0.845 | 0.846 | 0.726 | 0.841 |
| Mistral-7B ARC | 8 | 0.569 | 0.649 | 0.695 | 0.646 | 0.687 |
| Mistral-7B GSM8K | 6 | 0.444 | 0.812 | 0.823 | 0.744 | 0.817 |
| condition | verdict | readout | pooled | isolated | condition-fitted |
| No means correct | 0.617 | 0.640 | 0.789 | 0.781 | 0.805 |
| False means correct | 0.593 | 0.665 | 0.771 | 0.787 | 0.803 |
| 0 means correct | 0.769 | 0.787 | 0.804 | 0.809 | 0.810 |
| negated question | 0.665 | 0.650 | 0.794 | 0.802 | 0.805 |
| target | verdict | readout | pooled | tuned probe | isolated | condition-fitted |
| Yes/No | 0.788 | 0.815 | 0.809 | 0.824 | 0.818 | 0.815 |
| True/False | 0.779 | 0.815 | 0.812 | 0.811 | 0.814 | 0.814 |
| A/B | 0.744 | 0.798 | 0.803 | 0.774 | 0.808 | 0.804 |
| sentence | – | 0.807 | 0.801 | 0.776 | 0.806 | 0.809 |
| 1/0 | 0.794 | 0.815 | 0.806 | 0.784 | 0.810 | 0.814 |
| correct/incorrect | 0.779 | 0.816 | 0.810 | 0.800 | 0.815 | 0.810 |
| target | verdict | readout | pooled | tuned probe | isolated | condition-fitted |
| Yes/No | 0.801 | 0.812 | 0.802 | 0.812 | 0.812 | 0.812 |
| True/False | 0.800 | 0.812 | 0.809 | 0.794 | 0.804 | 0.812 |
| A/B | 0.779 | 0.812 | 0.807 | 0.787 | 0.815 | 0.811 |
| sentence | – | 0.802 | 0.793 | 0.762 | 0.804 | 0.807 |
| 1/0 | 0.809 | 0.804 | 0.805 | 0.789 | 0.810 | 0.813 |
| correct/incorrect | 0.794 | 0.807 | 0.803 | 0.802 | 0.808 | 0.807 |
| depth | Yes/No | B/A | No means correct | negated question | ||||
| readout | isolated | readout | isolated | readout | isolated | readout | isolated | |
| 0.25 | 0.699 | 0.702 | 0.699 | 0.700 | 0.700 | 0.701 | 0.699 | 0.702 |
| 0.50 | 0.786 | 0.788 | 0.780 | 0.785 | 0.781 | 0.782 | 0.783 | 0.784 |
| 0.75 | 0.818 | 0.824 | 0.800 | 0.813 | 0.756 | 0.808 | 0.756 | 0.818 |
| 0.90 | 0.813 | 0.820 | 0.794 | 0.808 | 0.648 | 0.799 | 0.632 | 0.811 |
| 1.00 | 0.815 | 0.818 | 0.792 | 0.805 | 0.640 | 0.781 | 0.650 | 0.802 |
| condition | readout | five formats only | negation pairs only | both | condition-fitted |
| No means correct | 0.640 | 0.722 | 0.777 | 0.781 | 0.805 |
| False means correct | 0.665 | 0.754 | 0.784 | 0.787 | 0.803 |
| 0 means correct | 0.787 | 0.805 | 0.810 | 0.809 | 0.810 |
| negated question | 0.650 | 0.733 | 0.801 | 0.802 | 0.805 |
| condition | readout | rank 1 | rank 3 | rank 8 | condition-fitted |
| Yes/No | 0.820 | 0.819 | 0.820 | 0.822 | 0.820 |
| True/False | 0.819 | 0.819 | 0.820 | 0.821 | 0.819 |
| A/B | 0.805 | 0.812 | 0.811 | 0.812 | 0.815 |
| sentence | 0.819 | 0.808 | 0.817 | 0.817 | 0.816 |
| 1/0 | 0.809 | 0.811 | 0.814 | 0.812 | 0.815 |
| correct/incorrect | 0.821 | 0.813 | 0.818 | 0.820 | 0.816 |
| condition | readout | after LEACE | after INLP | isolated |
| No means correct | 0.690 | 0.689 | 0.694 | 0.772 |
| False means correct | 0.675 | 0.674 | 0.688 | 0.783 |
| 0 means correct | 0.766 | 0.760 | 0.743 | 0.815 |
| negated question | 0.694 | 0.695 | 0.689 | 0.805 |
| condition | readout | identical-verdict pairs | verdict sign regressed out | as in Table 7 |
| No means correct | 0.690 | 0.772 | 0.771 | 0.772 |
| False means correct | 0.675 | 0.763 | 0.783 | 0.783 |
| 0 means correct | 0.766 | 0.820 | 0.818 | 0.815 |
| negated question | 0.694 | 0.784 | 0.782 | 0.805 |
| estimation evaluation | condition | readout | other wordings | same wording | fitted |
| P01,P03,P05 P07,P09 | No means correct | 0.680 | 0.745 | 0.760 | 0.804 |
| P01,P03,P05 P07,P09 | False means correct | 0.674 | 0.763 | 0.763 | 0.810 |
| P01,P03,P05 P07,P09 | 0 means correct | 0.770 | 0.814 | 0.816 | 0.818 |
| P01,P03,P05 P07,P09 | negated question | 0.727 | 0.788 | 0.811 | 0.809 |
| P07,P09 P01,P03,P05 | No means correct | 0.704 | 0.760 | 0.780 | 0.801 |
| P07,P09 P01,P03,P05 | False means correct | 0.676 | 0.764 | 0.795 | 0.808 |
| condition | readout | isolated judgment | fitted on the new family |
| Yes/No | 0.810 (0.819) | 0.813 (0.816) | 0.814 (0.820) |
| True/False | 0.810 (0.815) | 0.807 (0.815) | 0.812 (0.818) |
| A/B | 0.795 (0.814) | 0.802 (0.816) | 0.800 (0.813) |
| B/A | 0.778 (0.779) | 0.795 (0.809) | 0.797 (0.812) |
| sentence | 0.806 (0.817) | 0.804 (0.814) | 0.804 (0.815) |
| negated question | 0.652 (0.695) | 0.792 (0.793) | 0.804 (0.801) |
| entailment | safety | |||||||
| condition | readout | isolated | fitted | verdict | readout | isolated | fitted | verdict |
| Yes/No | 0.866 | 0.866 | 0.866 | 0.864 | 0.903 | 0.902 | 0.903 | 0.894 |
| True/False | 0.866 | 0.860 | 0.866 | 0.865 | 0.905 | 0.904 | 0.907 | 0.891 |
| A/B | 0.848 | 0.846 | 0.846 | 0.812 | 0.888 | 0.896 | 0.901 | 0.889 |
| B/A | 0.851 | 0.848 | 0.846 | 0.767 | 0.878 | 0.894 | 0.899 | 0.829 |
| sentence | 0.845 | 0.846 | 0.838 | – | 0.874 | 0.895 | 0.896 | – |
| model | Yes/No readout | frozen vs fitted | fall under negation | isolated vs fitted |
| Qwen2.5-32B-Instruct | 0.804 | 0.016 | 0.380 | 0.007 |
| Gemma-2-27B-IT | 0.750 | 0.016 | 0.140 | 0.006 |
| TriviaQA | MMLU | |||||||
| model | Yes/No | frozen vs fitted | fall | isolated vs fitted | Yes/No | frozen vs fitted | fall | isolated vs fitted |
| Mistral-Nemo-12B-Instruct | 0.776 | 0.007 | 0.151 | 0.002 | 0.756 | 0.007 | 0.033 | 0.002 |
| Ministral-8B-Instruct | 0.807 | 0.007 | 0.141 | 0.007 | 0.788 | 0.018 | 0.062 | 0.008 |
| OLMo-2-7B-Instruct | 0.764 | 0.006 | 0.014 | 0.005 | 0.737 | 0.012 | 0.030 | 0.002 |
| judge reads candidate | negatives | own-fitted | fitted here |
| Qwen2.5-7B reads Llama-3.1-8B, TriviaQA | 115 | 0.801 | 0.800 |
| Qwen2.5-7B reads Llama-3.1-8B, MMLU | 128 | 0.810 | 0.810 |
| Llama-3.1-8B reads Mistral-7B, TriviaQA | 114 | 0.828 | 0.825 |
| Llama-3.1-8B reads Mistral-7B, MMLU | 160 | 0.763 | 0.771 |
| Mistral-7B reads Qwen2.5-14B, TriviaQA | 145 | 0.822 | 0.822 |
| Mistral-7B reads Qwen2.5-14B, MMLU (descr.) | 82 | 0.706 | 0.716 |
| condition | readout | isolated | fitted | verdict |
| Yes/No | 0.885 (0.890) | 0.880 (0.884) | 0.885 (0.890) | 0.881 (0.888) |
| True/False | 0.881 (0.888) | 0.868 (0.870) | 0.881 (0.888) | 0.876 (0.887) |
| A/B | 0.865 (0.883) | 0.863 (0.882) | 0.862 (0.877) | 0.793 (0.800) |
| B/A | 0.848 (0.875) | 0.855 (0.880) | 0.854 (0.873) | 0.753 (0.740) |
| sentence | 0.867 (0.873) | 0.865 (0.877) | 0.875 (0.888) | – |
| negated question | 0.518 (0.502) | 0.827 (0.849) | 0.854 (0.861) | 0.574 (0.613) |
| cell | verdict, correct? | verdict, incorrect? | readout, correct? | readout, incorrect? | isolated, incorrect? | fitted, incorrect? |
| Llama-3.1-8B ARC | 0.698 | 0.528 | 0.737 | 0.718 | 0.732 | 0.727 |
| Llama-3.1-8B GSM8K | 0.852 | 0.821 | 0.966 | 0.944 | 0.965 | 0.968 |
| Llama-3.1-8B MMLU | 0.670 | 0.581 | 0.719 | 0.664 | 0.703 | 0.715 |
| Llama-3.1-8B TriviaQA | 0.832 | 0.798 | 0.842 | 0.726 | 0.841 | 0.839 |
| Mistral-7B ARC | 0.708 | 0.557 | 0.730 | 0.646 | 0.687 | 0.663 |
| Mistral-7B GSM8K | 0.792 | 0.761 | 0.826 | 0.744 | 0.817 | 0.822 |
| condition | readout | isolated judgment |
| Yes/No | 0.996 (0.98–1.00) | 0.988 (0.94–1.00) |
| True/False | 0.991 (0.94–1.00) | 0.974 (0.89–1.00) |
| A/B | 0.977 (0.87–1.00) | 0.969 (0.86–0.99) |
| sentence | 0.970 (0.86–0.99) | 0.947 (0.84–0.99) |
| 1/0 | 0.988 (0.93–1.00) | 0.970 (0.86–1.00) |
| correct/incorrect | 0.986 (0.90–1.00) | 0.977 (0.89–1.00) |
| cell | negated | No means correct | False means correct | |||
| plain | isolated | plain | isolated | plain | isolated | |
| Llama-3.1-8B ARC | 0.427 | 0.762 | 0.757 | 0.860 | 0.829 | 0.883 |
| Llama-3.1-8B GSM8K | 0.425 | 0.935 | 0.845 | 0.922 | 0.817 | 0.910 |
| Llama-3.1-8B MMLU | 0.488 | 0.708 | 0.797 | 0.790 | 0.798 | 0.806 |
| Llama-3.1-8B TriviaQA | 0.226 | 0.905 | 0.610 | 0.930 | 0.817 | 0.920 |
| Mistral-7B GSM8K | 0.204 | 0.857 | 0.829 | 0.936 | 0.856 | 0.983 |
| cell | verdicts agree | vs Yes/No verdict | vs negated verdict | vs Yes/No, disagreeing rows |
| Llama-3.1-8B ARC | 0.219 | 0.762 | 0.287 | 0.967 |
| Llama-3.1-8B GSM8K | 0.682 | 0.935 | 0.692 | 0.963 |
| Llama-3.1-8B MMLU | 0.251 | 0.708 | 0.394 | 0.855 |
| Llama-3.1-8B TriviaQA | 0.687 | 0.905 | 0.732 | 0.760 |
| Mistral-7B GSM8K | 0.801 | 0.857 | 0.783 | 0.877 |
| Mistral-7B MMLU | 0.476 | 0.820 | 0.403 | 0.817 |
LLM Interpretability
Large language model (LLM) interpretability studies how internal representations and computations produce outputs, and how they can be explained or controlled.
See all 781 papers