Can Large Language Models Learn Independent Causal Mechanisms?

Gael Gendron, Bao Nguyen, Alex Peng, Michael Witbrock, Gillian Dobbie


Abstract
Despite impressive performance on language modelling and complex reasoning tasks, Large Language Models (LLMs) fall short on the same tasks in uncommon settings or with distribution shifts, exhibiting a lack of generalisation ability. By contrast, systems such as causal models, that learn abstract variables and causal relationships, can demonstrate increased robustness against changes in the distribution. One reason for this success is the existence and use of Independent Causal Mechanisms (ICMs) representing high-level concepts that only sparsely interact. In this work, we apply two concepts from causality to learn ICMs within LLMs. We develop a new LLM architecture composed of multiple sparsely interacting language modelling modules. We show that such causal constraints can improve out-of-distribution performance on abstract and causal reasoning tasks. We also investigate the level of independence and domain specialisation and show that LLMs rely on pre-trained partially domain-invariant mechanisms resilient to fine-tuning.
Anthology ID:
2024.emnlp-main.381
Volume:
Proceedings of the 2024 Conference on Empirical Methods in Natural Language Processing
Month:
November
Year:
2024
Address:
Miami, Florida, USA
Editors:
Yaser Al-Onaizan, Mohit Bansal, Yun-Nung Chen
Venue:
EMNLP
SIG:
Publisher:
Association for Computational Linguistics
Note:
Pages:
6678–6701
Language:
URL:
https://aclanthology.org/2024.emnlp-main.381
DOI:
Bibkey:
Cite (ACL):
Gael Gendron, Bao Nguyen, Alex Peng, Michael Witbrock, and Gillian Dobbie. 2024. Can Large Language Models Learn Independent Causal Mechanisms?. In Proceedings of the 2024 Conference on Empirical Methods in Natural Language Processing, pages 6678–6701, Miami, Florida, USA. Association for Computational Linguistics.
Cite (Informal):
Can Large Language Models Learn Independent Causal Mechanisms? (Gendron et al., EMNLP 2024)
Copy Citation:
PDF:
https://aclanthology.org/2024.emnlp-main.381.pdf