CausalDetox: Causal Head Selection and Intervention for Language Model Detoxification

Yian Wang, Yuen Chen, Agam Goyal, Hari Sundaram


Abstract
Large language models (LLMs) frequently generate toxic content, posing significant risks for safe deployment. Current mitigation strategies often degrade generation quality or require costly human annotation. We propose CausalDetox, a framework that identifies and intervenes on the specific attention heads causally responsible for toxic generation. Using the Probability of Necessity and Sufficiency (PNS), we isolate a minimal set of heads that are necessary and sufficient for toxicity. We utilize these components via two complementary strategies: (1) Local Inference-Time Intervention, which constructs dynamic, input-specific steering vectors for context-aware detoxification, and (2) PNS-Guided Fine-Tuning, which permanently unlearns toxic representations. We also introduceParaTox, a novel benchmark of aligned toxic/non-toxic sentence pairs enabling controlled counterfactual evaluation. Experiments on ToxiGen, ImplicitHate, and ParaDetox show that CausalDetox achieves up to 5.34% greater toxicity reduction compared to baselines while preserving linguistic fluency, and offers a speedup in head selection.
Anthology ID:
2026.findings-acl.577
Volume:
Findings of the Association for Computational Linguistics: ACL 2026
Month:
July
Year:
2026
Address:
San Diego, California, United States
Editors:
Maria Liakata, Viviane P. Moreira, Jiajun Zhang, David Jurgens
Venue:
Findings
SIG:
Publisher:
Association for Computational Linguistics
Note:
Pages:
11893–11914
Language:
URL:
https://preview.aclanthology.org/ingest-acl/2026.findings-acl.577/
DOI:
Bibkey:
Cite (ACL):
Yian Wang, Yuen Chen, Agam Goyal, and Hari Sundaram. 2026. CausalDetox: Causal Head Selection and Intervention for Language Model Detoxification. In Findings of the Association for Computational Linguistics: ACL 2026, pages 11893–11914, San Diego, California, United States. Association for Computational Linguistics.
Cite (Informal):
CausalDetox: Causal Head Selection and Intervention for Language Model Detoxification (Wang et al., Findings 2026)
Copy Citation:
PDF:
https://preview.aclanthology.org/ingest-acl/2026.findings-acl.577.pdf
Checklist:
 2026.findings-acl.577.checklist.pdf