RAPTOR: Ridge-Adaptive Logistic Probes
cs.LG, cs.AI
Submitted: 2026-01-29
Updated: 2026-09-25
Comments: Preprint
License: http://creativecommons.org/licenses/by/4.0/
The gist: Probing studies what information is encoded in a frozen LLM's layer representations by training a lightweight predictor on top of them.
Terminology
Abstract
Probing studies what information is encoded in a frozen LLM's layer representations by training a lightweight predictor on top of them. Beyond analysis, probes are often used operationally in probe-then-steer pipelines: a learned concept vector is extracted from a probe and injected via additive activation steering by adding it to a layer representation during the forward pass. The effectiveness of this pipeline hinges on estimating concept vectors that are accurate, directionally stable under ablation, and inexpensive to obtain. Motivated by these desiderata, we propose RAPTOR (Ridge-Adaptive Logistic Probe), a simple L2-regularized logistic probe whose validation-tuned ridge strength yields concept vectors from normalized weights. Across extensive experiments on instruction-tuned LLMs and human-written concept datasets, RAPTOR matches or exceeds strong baselines in accuracy while achieving competitive directional stability and substantially lower training cost; these quantitative results are supported by qualitative downstream steering demonstrations. Finally, using the Convex Gaussian Min-max Theorem (CGMT), we provide a mechanistic characterization of ridge logistic regression in an idealized Gaussian teacher-student model in the high-dimensional few-shot regime, explaining how penalty strength mediates probe accuracy and concept-vector stability and yielding structural predictions that qualitatively align with trends observed on real LLM embeddings.
Sources
- Understanding intermediate layers using linear classifier probes
- Toward universal steering and monitoring of AI models
- What do Neural Machine Translation Models Learn about Morphology?
- Interpretability Beyond Feature Attribution: Quantitative Testing with Concept Activation Vectors (TCAV)
- Steering Llama 2 via Contrastive Activation Addition
- The Impact of Regularization on High-dimensional Logistic Regression
- Improving Instruction-Following in Language Models through Activation Steering
- Analyzing the Generalization and Reliability of Steering Vectors
- BERT Rediscovers the Classical NLP Pipeline
- Steering Language Models With Activation Engineering
Related papers
- Polynomial-Augmented Neural Networks (PANNs) with Weak Orthogonality Constraints for Enhanced Function and PDE Approximation
- AIRL-S: Unifying Reinforcement Learning and Search-Based Test-Time Scaling via Adversarial Inverse Reinforcement Learning
- Transformers as Bayesian In-Context Experimenters: Smoothness-Adaptive Efficient ATE Estimation
- Convergence issues in Relational Concept Analysis based on AOC-posets
- Beliefs Beyond Posteriors: The Predictive Cost of ELBO Binding
- Understanding Diffusion Models via Ratio-Based Function Approximation with SignReLU Networks