atlas

Research map / Architectures and efficiency / RNNs and normalization

Debiasing Algorithm through Model Adaptation

ICLR 2024 · Architectures and efficiency › RNNs and normalization · approach: Layer-wise analysis of language models · 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.

Causal tracing locates mid-upper MLP layers carrying gender bias in LLaMA models, and DAMA projects their output weights onto the orthogonal complement of a stereotype subspace to reduce bias while preserving perplexity and downstream accuracy.

Problem

Large language models encode gender stereotypes from training data (e.g., associating professions with a gender), causing biased pronoun predictions and uneven coreference performance; the goal is to detect which components carry this bias and edit weights to reduce it without hurting general perfor

Approach

Use causal tracing with a linear fit of pronoun probability differences against annotated stereotypical and factual gender scores to locate bias-mediating MLP layers, then edit them by multiplying the FF output matrix by a PLS-derived orthogonal projection that removes the stereotype subspace.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Project selected MLP weights onto the subspace orthogonal to stereotypical directionsLLaMA 7B · Gendered promptsa_s: 0.235 -> -0.005Helped
Project selected MLP weights onto the subspace orthogonal to stereotypical directionsLLaMA 7B · WinoBiasΔS: 40.3% -> 31.5%Helped
Project selected MLP weights onto the subspace orthogonal to stereotypical directionsLLaMA 7B · StereoSet gender intrasentenceICAT: 53.7 -> 58.5Helped
Project selected MLP weights onto the subspace orthogonal to stereotypical directionsLLaMA 7B · Wikipedia 103PPL: 26.1 -> 28.9Hurt
Project selected MLP weights onto the subspace orthogonal to stereotypical directionsLLaMA 7B · ARC-Caccuracy: 42.2 -> 41.8No clear effect
Project selected MLP weights onto the subspace orthogonal to stereotypical directionsLLaMA 7B · ARC-Eaccuracy: 69.1 -> 68.3Hurt
Project selected MLP weights onto the subspace orthogonal to stereotypical directionsLLaMA 7B · MMLUaccuracy: 30.3 -> 30.8No clear effect
Project selected MLP weights onto the subspace orthogonal to stereotypical directionsLLaMA 13B, LLaMA 30B, and LLaMA 65B · Gendered promptsa_s: 0.270 -> 0.148; 0.265 -> 0.105; 0.249 -> 0.185Helped
Project selected MLP weights onto the subspace orthogonal to stereotypical directionsLLaMA 13B, LLaMA 30B, and LLaMA 65B · WinoBiasΔS: 35.7% -> 31.1%; 36.0% -> 26.7%; 35.7% -> 27.2%Helped
Expand the number of edited layers and projection dimensions (ablation)DAMA on LLaMA 7B · Gendered promptsa_sHelped
Expand the number of edited layers and projection dimensions (ablation)DAMA on LLaMA 7B · Wikipedia 103PPLHurt
Apply DAMA to the two last model layers (ablation)DAMA · Gender bias evaluationa_sHurt

Surprising findings

Limitations

Experimental setup

Datasets: Bolukbasi et al. professions list (factual and stereotypical scores) WinoBias StereoSet Wikitext-103 OpenBookQA ARC MMLU

Benchmarks: WinoBias StereoSet Wikitext-103 perplexity OpenBookQA ARC-Challenge ARC-Easy MMLU

Metrics: linear coefficients a_s, a_f, b, R^2 WinoBias accuracy, delta S, delta G StereoSet lms, ss, ICAT perplexity accuracy

Baselines: MEMIT LoRA FT LLaMA 7B

Builds on: MEMIT causal tracing LoRA

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