atlas

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

Whispering Experts: Neural Interventions for Toxicity Mitigation in Language Models

ICML 2024 · Large language models › Language, safety and interpretability · approach: Faithfulness of LLM explanations · 0 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.

Proposes AurA, a hyperparameter-free intervention that dampens toxicity expert neurons (ranked by AUROC) in pre-trained LLMs, reducing RealToxicityPrompts toxicity up to 2.2x with small perplexity increase.

Problem

LLMs generate toxic language; existing mitigation via fine-tuning or pre-/post-processing is costly, jailbreakable, or degrades perplexity. Prior expert-neuron zeroing is sensitive to the number of intervened neurons.

Approach

Identify toxicity expert neurons by the AUROC of their max activation for toxic vs non-toxic sentences, and dampen each expert's activation by a factor 1 - Gini (= 2 - 2·AUROC), so stronger experts are scaled down more; this is hyperparameter-free and can be folded into weights.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Apply AurA expert-neuron dampeningGPT2-XL · RealToxicityPromptstoxicity: 0.382 -> 0.289Helped
Apply AurA expert-neuron dampeningFalcon-7B · RealToxicityPromptstoxicity: 0.382 -> 0.180Helped
Apply AurA expert-neuron dampeningFalcon-40B · RealToxicityPromptstoxicity: 0.395 -> 0.243Helped
Apply AurA expert-neuron dampeningMPT-7B · RealToxicityPromptstoxicity: 0.333 -> 0.187Helped
Apply AurA expert-neuron dampeningMPT-30B · RealToxicityPromptstoxicity: 0.392 -> 0.240Helped
Apply AurA expert-neuron dampeningLlama-v2 · RealToxicityPromptstoxicity: 0.379 -> 0.218Helped
Apply AurA expert-neuron dampeningMistral-7B · RealToxicityPromptstoxicity: 0.380 -> 0.173Helped
Add AurA to favorable pre-promptingFalcon-7B-instruct · Favorable pre-promptstoxicity reduction: 1.28× -> 2.35×Helped
Add AurA during adversarial pre-promptingFalcon-7B-instruct · Adversarial pre-promptstoxicity reduction: 2×Helped
Use AurA instead of no interventionFalcon-7B · PIQAaccuracy: 79.5 -> 78.7Hurt
Use AurA instead of no interventionFalcon-7B · TriviaQAexact match (%): 38.2 -> 22.7Hurt
Use AurA instead of no interventionLlama-v2 · average zero-shot common-sense reasoningaverage: 52.92 -> 53.12Helped

Surprising findings

Limitations

Experimental setup

Datasets: Jigsaw 2018 toxic category (500 toxic, 2000 non-toxic sentences for expert identification) Wikipedia (perplexity)

Benchmarks: RealToxicityPrompts HONEST Jigsaw 2018 WikiText perplexity (PPL_WIK) PIQA SIQA TriviaQA TruthfulQA Hellaswag

Metrics: toxicity probability perplexity zero-shot accuracy exact match inter-annotator agreement (Cohen's kappa)

Baselines: DExperts CTRL Det_zero (zeroing experts) Damp (constant dampening) pre-prompting

Builds on: expert neurons (Suau et al. 2023 concept induction) DExperts CTRL PPLM

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