Probing Causality Manipulation of Large Language Models

Chenyang Zhang,Haibo Tong,Bin Zhang,Dongyu Zhang
2024-08-27
Abstract:Large language models (LLMs) have shown various ability on natural language processing, including problems about causality. It is not intuitive for LLMs to command causality, since pretrained models usually work on statistical associations, and do not focus on causes and effects in sentences. So that probing internal manipulation of causality is necessary for LLMs. This paper proposes a novel approach to probe causality manipulation hierarchically, by providing different shortcuts to models and observe behaviors. We exploit retrieval augmented generation (RAG) and in-context learning (ICL) for models on a designed causality classification task. We conduct experiments on mainstream LLMs, including GPT-4 and some smaller and domain-specific models. Our results suggest that LLMs can detect entities related to causality and recognize direct causal relationships. However, LLMs lack specialized cognition for causality, merely treating them as part of the global semantic of the sentence.
Computation and Language,Artificial Intelligence
What problem does this paper attempt to address?
The problem that this paper attempts to solve is to explore the internal mechanisms of large - language models (LLMs) in handling causal relationships. Specifically, although large - language models perform excellently in natural - language - processing tasks, including some causal - reasoning tasks, how they intrinsically handle causal relationships remains unclear. Due to the complex structures of these models and the large number of parameters, the cost of directly reconstructing or in - depth studying their internal mechanisms is extremely high, and some advanced architectures such as Mixture - of - Experts (MoE) make detailed exploration even more difficult. In addition, the technical details of some existing models are not made public, which further increases the difficulty of research. To meet this challenge, this paper proposes an innovative method to explore the internal mechanisms of LLMs in handling causal relationships. By constructing a classification dataset to detect causal entities in sentences and their relationships, and by providing "shortcuts" hierarchically (such as Retrieval - Augmented Generation (RAG) and In - Context Learning (ICL)) to guide model behavior, and finally by observing the performance changes under different "shortcuts" to explore the internal way that LLMs handle causal relationships. The main contributions of the paper include: 1. Constructing a classification dataset specifically for exploring causal - relationship handling. 2. Proposing a method of providing "shortcuts" hierarchically to observe and analyze the performance of LLMs under different conditions. 3. The experimental results show that although LLMs can identify causal entities and understand causal relationships to a certain extent, they lack specific cognition of causal relationships and mainly rely on the global semantics of sentences to make judgments. Through these studies, the author hopes to reveal the deficiencies of LLMs in handling causal relationships and provide references for further optimizing model training.