atlas

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

The Devil is in the Neurons: Interpreting and Mitigating Social Biases in Language Models

ICLR 2024 · Large language models › Language, safety and interpretability · approach: Bias measurement and mitigation

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 Integrated Gap Gradients (IG^2) to locate social bias neurons in masked language models by attributing the logits gap between demographic pairs, and a training-free Bias Neuron Suppression (BNS) method that zeroes those neurons to reduce bias.

Problem

Social biases in pre-trained language models are detected only via black-box probing and mitigated by costly fine-tuning or pre-training on anti-stereotypical data; the internal units responsible for biased demographic output gaps are unknown.

Approach

Adapt integrated gradients to attribute the logits gap between two demographics (rather than a single output) to FFN neurons, then suppress the top-attributed neurons at inference time without any retraining; also analyze how bias neurons shift layers after debiasing.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Suppress IG^2-identified social bias neuronsFairBERTa · Gender-Nlogits gap: decreases 22.98%Helped
Amplify IG^2-identified social bias neuronsFairBERTa · Gender-Nlogits gap: increases 29.05%Hurt
Suppress the union of neurons identified by vanilla IG for each demographicRoBERTa-Base with Union_IG · StereoSet intrasentencelanguage modeling score (LMS): 91.70 -> 30.61Hurt
Suppress the union of neurons identified by vanilla IG for each demographicRoBERTa-Base with Union_IG · StereoSet intrasentencestereotype score (SS): 53.82Helped
Use BNS to suppress identified social bias neuronsBNS · StereoSet intrasentenceidealized CAT score (ICAT): 84.79Helped
Manipulate randomly selected neurons instead of IG^2-identified neuronsFairBERTa · selected Unfair Targetslogits gapNo clear effect
Pre-train RoBERTa on anti-stereotypical dataRoBERTa · social bias neuron countaverage number of social bias neurons: from 10.35 to 10.06No clear effect
Pre-train RoBERTa on anti-stereotypical dataRoBERTa · social bias neuron distribution across layerspercentage of neurons in the two shallowest layers: from 2.9% (RoBERTa) to 26.0% (FairBERTa)Mixed

Surprising findings

Limitations

Experimental setup

Datasets: StereoSet ParaRel-inspired templates (constructed)

Benchmarks: StereoSet intrasentence

Metrics: LMS SS ICAT logits gap change ratio

Baselines: FairBERTa DPCE AutoDebias Union_IG

Builds on: Integrated Gradients ParaRel StereoSet

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