REVIEW 4 cited by
Aligning Large Language Models with Representation Editing: A Control Perspective
Not yet reviewed by Pith; the record is open.
This paper has not been read by Pith yet. Machine review is queued; the pith claim, tier, and objections will appear here once it completes.
SPECIMEN: schema-true, not a live event
T0 review · schema-true
One-sentence machine reading of the paper's core claim.
pith:XXXXXXXX · record.json · timestamp
read the original abstract
Aligning large language models (LLMs) with human objectives is crucial for real-world applications. However, fine-tuning LLMs for alignment often suffers from unstable training and requires substantial computing resources. Test-time alignment techniques, such as prompting and guided decoding, do not modify the underlying model, and their performance remains dependent on the original model's capabilities. To address these challenges, we propose aligning LLMs through representation editing. The core of our method is to view a pre-trained autoregressive LLM as a discrete-time stochastic dynamical system. To achieve alignment for specific objectives, we introduce external control signals into the state space of this language dynamical system. We train a value function directly on the hidden states according to the Bellman equation, enabling gradient-based optimization to obtain the optimal control signals at test time. Our experiments demonstrate that our method outperforms existing test-time alignment techniques while requiring significantly fewer resources compared to fine-tuning methods. Our code is available at https://github.com/Lingkai-Kong/RE-Control.
Forward citations
Cited by 4 Pith papers
-
Steering Robustness into World Action Models via Mechanistic Interpretability and Optimal Control
Contrastive activation directions and a reduced-order LQR (WA-LQR) steer world-action models to recover robustness under camera, gripper, and noise shifts whenever the models' activations are linearly separable for th...
-
Reward Shaping for (Inference-Time) Alignment: A Stackelberg Game Perspective
The optimal reward for KL-regularized LLM alignment is a threshold function—reward B above a prompt-dependent cutoff, 0 below—which can be estimated from base-model samples and integrated into decoding-time alignment.
-
Balancing Stylization and Truth via Disentangled Representation Steering
StyliTruth separates style and truth directions in the activations of selected attention heads, then steers each token along the disentangled subspaces to preserve truthfulness during stylization.
-
LLM-ML Teaming: Integrated Symbolic Decoding and Gradient Search for Valid and Stable Generative Feature Transformation
A product-of-experts decoder that blends a fine-tuned LLM's token probabilities with a gradient-searched sequence decoder produces more valid and stable feature transformations than either alone.
Discussion (0). Sign in to comment.