atlas

Research map / Large language models / Language, safety and interpretability

A Language Model’s Guide Through Latent Space

ICML 2024 · Large language models › Language, safety and interpretability · approach: Reasoning mechanisms and failure modes · 2 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Extends concept guidance (linear probing plus activation steering) from truthfulness to appropriateness, humor, creativity and quality on Llama-2-7b-chat and Mistral-7B, and proposes a perplexity-normalized effect size metric.

Problem

Concept guidance via linear probes and activation steering has mostly been studied for truthfulness; it is unclear whether detection and guidance transfer to richer concepts like appropriateness, humor, creativity and quality, and how to measure guidance quality jointly with fluency.

Approach

Compare logistic regression, difference-in-means and PCA probes on normalized residual activations of the first t tokens, steer by adding the probe direction with norm preservation across the top-k layers, and score guidance with a perplexity-normalized effect size (PNES) that fits tanh effect and log-quadratic perplexity curves.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Guide along the truthfulness directionLLMs using linear concept guidance · TruthfulPNESHelped
Guide with the appropriateness vectorLLMs using linear concept guidance · CompliancePNES: 1.00Helped
Use the appropriateness vector to elicit appropriatenessLLMs using linear concept guidance · Appropriateappropriateness elicitationHurt
Use the humor concept vector with a tuned guidance configurationLLMs using linear concept guidance · HumorPNES: 0.35Helped
Use the creativity concept vector with a tuned guidance configurationLLMs using linear concept guidance · CreativityPNES: 0.23Helped
Use the quality concept vector with a tuned guidance configurationLLMs using linear concept guidance · QualityPNES: 0.54Helped
Increase context length from a single token to t=16 (ablation)Concept probes · Concept detectionaccuracyMixed
Increase context length for PCA probing (ablation)PCA concept probes · Concept detectionaccuracyHurt
Use logistic regression instead of the other probing techniques (ablation)Concept detection · Appropriateness/harmfulnessaccuracy: approximately 90%Helped
Fine-tune the base models with instruction tuning/alignment (ablation)Llama-2 and Mistral · Truthfulness, appropriateness, and quality detectionaccuracyHelped
Fine-tune the base models with instruction tuning/alignment (ablation)Llama-2 and Mistral · Humor and creativity detectionaccuracyNo clear effect
Select probes based on higher detection accuracy (ablation)Concept guidance configurations · Concept guidancePNESNo clear effect

Surprising findings

Limitations

Experimental setup

Datasets: OpenAssistant (OASST1) ToxicChat TruthfulQA

Benchmarks: TruthfulQA OpenAssistant concept labels (humor, creativity, quality) ToxicChat-based appropriateness/compliance

Metrics: detection accuracy perplexity-normalized effect size (PNES) perplexity concept presence probability

Baselines: Logistic regression probe Difference-in-means probe PCA probe

Builds on: linear representation hypothesis classifier-guided generation inference-time intervention

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy