Correctness-Optimized Residual Activation Lens (CORAL): Transferrable and Calibration-Aware Inference-Time Steering
Not on the map. Why?
Abstract (arXiv)
Large language models (LLMs) exhibit persistent miscalibration, especially after instruction tuning and preference alignment. Modified training objectives can improve calibration, but retraining is expensive. Inference-time steering offers a lightweight alternative, yet most existing methods optimize proxies for correctness rather than correctness itself. We introduce CORAL (Correctness-Optimized Residual Activation Lens), a regularized inference-time steering method that captures distributed correctness signals from model internal activations using weight-decay MLP probes. We evaluate CORAL across three 7B-parameter models and find that it consistently improves accuracy by 10% and expected calibration error (ECE) by 50% on average. We additionally demonstrate that these gains transfer without retraining to the complete published test sets of four held-out benchmarks (ARC-Challenge, HellaSwag, Math-MC, OpenBookQA), averaging 14% accuracy improvements and 49% ECE improvements. Our results support the hypothesis that distributed information in model internals can be extracted using regularized probes when individual neurons are insufficient. CORAL thus provides a compute-efficient, transferable, and calibration-aware approach to improve MCQA performance during inference.
Key facts. Cited by 2 papers in the map (data through 7 Oct 2026). Builds on 23 of its 33 references that are in the map. First posted on arXiv 5 Feb 2026.
Builds on
23 of 33 references.
References · 23
- 2022Training language models to follow instructions with human feedbackcited by 15,712
- 2020Measuring Massive Multitask Language Understandingcited by 7,768
- 2017On Calibration of Modern Neural Networkscited by 4,860
- 2018Think you have Solved Question Answering? Try ARC, the AI2 Reasoning Challengecited by 4,710
- 2018CommonsenseQA: A Question Answering Challenge Targeting Commonsense Knowledgecited by 1,793
- 2020Transformer Feed-Forward Layers Are Key-Value Memoriescited by 1,307
- 2016Understanding intermediate layers using linear classifier probescited by 1,279
- 2023Inference-Time Intervention: Eliciting Truthful Answers from a Language Modelcited by 1,259
- 2023Steering Llama 2 via Contrastive Activation Additioncited by 1,009
- 2017RACE: Large-scale ReAding Comprehension Dataset From Examinationscited by 851
- 2023Towards Automated Circuit Discovery for Mechanistic Interpretabilitycited by 700
- 2019MathQA: Towards Interpretable Math Word Problem Solving with Operation-Based Formalismscited by 653
11 more
- 2022Transformer Feed-Forward Layers Build Predictions by Promoting Concepts in the Vocabulary Spacecited by 522
- 2020How Can We Know When Language Models Know? On the Calibration of Language Models for Question Answeringcited by 439
- 2024Taming Overconfidence in LLMs: Reward Calibration in RLHFcited by 105
- 2024Confidence Regulation Neurons in Language Modelscited by 57
- 2024pyvene: A Library for Understanding and Improving PyTorch Models via Interventionscited by 53
- 2019Careful Selection of Knowledge to solve Open Book Question Answeringcited by 35
- 2025No Answer Needed: Predicting LLM Answer Accuracy from Question-Only Linear Probescited by 35
- 2025Calibrating LLM Confidence by Probing Perturbed Representation Stabilitycited by 23
- 2025SteerConf: Steering LLMs for Confidence Elicitationcited by 20
- 2025CorrSteer: Generation-Time LLM Steering via Correlated Sparse Autoencoder Featurescited by 10
- 2025Calibration Across Layers: Understanding Calibration Evolution in LLMscited by 9
Built on by
Citing papers per year: 2026: 2.
2 papers.