Causal ATE Mitigates Unintended Bias in Controlled Text Generation

Rahul Madhavan, Kahini Wadhawan


Abstract
We study attribute control in language models through the method of Causal Average Treatment Effect (Causal ATE). Existing methodsfor the attribute control task in Language Models(LMs) check for the co-occurrence of words in a sentence with the attribute of interest, and control for them. However, spurious correlation of the words with the attribute in the training dataset, can cause models to hallucinate the presence of the attribute when presented with the spurious correlate during inference. We show that the simple perturbation-based method of Causal ATE removes this unintended effect. Specifically, we ground it in the problem of toxicity mitigation, where a significant challenge lies in the inadvertent bias that often emerges towards protected groups post detoxification. We show that this unintended bias can be solved by the use of the Causal ATE metric. We provide experimental validations for our claims and release our code (anonymously) here: [github.com/causalate-mitigates-bias](https://github.com/causalate-mitigates-bias/causal-ate-mitigates-bias).
Anthology ID:
2024.conll-1.11
Volume:
Proceedings of the 28th Conference on Computational Natural Language Learning
Month:
November
Year:
2024
Address:
Miami, FL, USA
Editors:
Libby Barak, Malihe Alikhani
Venue:
CoNLL
SIG:
Publisher:
Association for Computational Linguistics
Note:
Pages:
130–142
Language:
URL:
https://aclanthology.org/2024.conll-1.11
DOI:
Bibkey:
Cite (ACL):
Rahul Madhavan and Kahini Wadhawan. 2024. Causal ATE Mitigates Unintended Bias in Controlled Text Generation. In Proceedings of the 28th Conference on Computational Natural Language Learning, pages 130–142, Miami, FL, USA. Association for Computational Linguistics.
Cite (Informal):
Causal ATE Mitigates Unintended Bias in Controlled Text Generation (Madhavan & Wadhawan, CoNLL 2024)
Copy Citation:
PDF:
https://aclanthology.org/2024.conll-1.11.pdf