Origo: causal inference by compression
Abstract
Causal inference from observational data is one of the most fundamental problems in science. In general, the task is to tell whether it is more likely that \(X\) caused \(Y\), or vice versa, given only data over their joint distribution. In this paper we propose a general inference framework based on Kolmogorov complexity, as well as a practical and computable instantiation based on the Minimum Description Length principle. Simply put, we propose causal inference by compression. That is, we infer that \(X\) is a likely cause of \(Y\) if we can better compress the data by first encoding \(X\), and then encoding \(Y\) given \(X\), than in the other direction. To show this works in practice, we propose Origo, an efficient method for inferring the causal direction from binary data. Origo employs the lossless Pack compressor and searches for that set of decision trees that encodes the data most succinctly. Importantly, it works directly on the data and does not require assumptions about neither distributions nor the type of causal relations. To evaluate Origo in practice, we provide extensive experiments on synthetic, benchmark, and realworld data, including three case studies. Altogether, the experiments show that Origo reliably infers the correct causal direction on a wide range of settings.
Keywords
Causal inference Kolmogorov complexity MDL Decision trees Binary data1 Introduction
Causal inference, telling cause from effect, is perhaps one of the most important problems in science. To make absolute statements about cause and effect, carefully designed experiments are necessary, in which we consider representative populations, instrument the cause, and control for everything else [25]. In practice, setting up such an experiment is often very expensive, or simply impossible. The study of the effect of combinations of drugs is good example.
Certain drugs can amplify each other’s effect, and therewith combinations of drugs can turn out to be much more effective, or even only effective, than when the drugs are taken individually. This effect is sometimes positive, for example in combination treatments against HIV and cancer, but sometimes it is also negative, as it can lead to severe up to possibly lethal side effects. For all but the smallest number of drugs, however, there are so many possible combinations that it quickly becomes practically impossible to test these combinations in a controlled manner. This is even when we ignore the ethical aspect of potentially exposing volunteers to lethal side effects, as we need sufficiently many volunteers per combination of drugs, and all of these need to be (as) identical (as reasonably possible) for all other aspects, except the combination of drugs they get. That is, to investigate the combined effects of only 10 drugs, we already need \(2^{10} = 1024\) groups, each of say 100 volunteers, meaning we would need to recruit over 100,000 nearidentical volunteers. Clearly, this is not practically feasible.
We hence consider causal inference from observational data. That is, our goal is to infer the most likely direction of causation from data that has not been obtained in a completely controlled manner but is simply available. In recent years large strides have been made in the theory and practice of discovering causal structure from such data [12, 16, 25]. Most methods, and especially those that defined for pairs of variables, however, can only consider continuousvalued or discrete numeric data [27, 39] and are hence not applicable on binary data such as one would have in the above example.
We propose a general framework for causal inference on observational data, and give a practical instantiation for binary data. We base our inference framework on the solid foundations of Kolmogorov complexity [17, 20] and develop a score for pairs of data objects that identifies not only the direction [12], but also quantifies the strength of causation, without making any assumptions on the distribution nor the type of causal relation between the data objects, and without requiring any parameters to be set.
Kolmogorov complexity is not computable, however, and to be able to put it to practice we derive a practical, computable version based on the Minimum Description Length (MDL) principle [9, 28]. As a proof of concept, we propose Origo,^{1} which is an efficient and parameterfree method for causal inference on binary data. Origo builds on the MDLbased Pack algorithm [36] and compresses data using decision trees. Simply put, it encodes the data one attribute at a time using a decision tree. Such a tree may only split on previously encoded attributes. We use this mechanism to measure how much better we can compress the data of Y given the data of X, simply by (dis)allowing the trees for Y to split on attributes of X, and vice versa. We identify the most likely causal direction as the one with the most succinct description.
Extensive experiments on synthetic, benchmark, and realworld data show that Origo performs well in practice. It is robust to noise, dimensionality, and skew between cardinality of X and Y. It has high statistical power, and outperforms a recent proposal for discrete data by a wide margin. After discretization, Origo performs well on both univariate and multivariate benchmark data. Three case studies confirm that Origo provides intuitive results.

a theoretical framework for causal inference from observational data based on Kolmogorov complexity,

a practical framework for causal inference based on MDL,

a causal inference method for binary data, Origo.

an extensive set of experiments on synthetic and real data.
The remainder of this paper is organised as follows. We introduce notation and preliminaries in Sect. 2. Section 3 explains how to do causal inference based on Algorithmic Information Theory. In Sect. 4 we show how to derive practical, computable, causal indicators using the Minimum Description Length principle. We instantiate this framework for binary data using a decisiontreebased compressor in Sect. 5. Related work is covered in Sect. 6, and we evaluate empirically in Sect. 7. We round up with discussion and conclusions in Sects. 8 and 9, respectively.
All code and data are available for research purposes.^{2}
2 Preliminaries
In this section, we introduce notations and background definitions we will use in subsequent sections.
2.1 Notation
In this work, we consider binary data. We denote a binary string of length n by \(s \in \{0,1\}^n\). A binary dataset D is a binary matrix of size \(n\hbox {by}m\) consisting of n rows, or transactions, and m columns, random variables, or attributes. A row is a binary vector of size m. We write \(\Pr (X=v)\) for the probability of a random variable X assuming value v from the domain \( dom (X)\). We say \(X \rightarrow Y \) to indicate that X causes Y. We will model our data with sets of binary decision trees. The decision tree for \(X_i\) is denoted by \(T_i\).
All logarithms are to base 2, and by convention we use \(0\log 0 = 0\).
2.2 Kolmogorov complexity
To develop our causal inference principle, we need the concept of Kolmogorov complexity [3, 17, 33]. Below we give a brief introduction.
Conditional Kolmogorov complexity, denoted \(K(x \mid y)\), is the length of the shortest binary program \(p^*\) that generates x and halts when y is provided as an input to the program. We have \(K(x) = K(x \mid \epsilon )\), where \(\epsilon \) is the empty string.
Although Kolmogorov complexity is defined over binary strings, we can interchangeably use it over mathematical objects, or data objects in general, as any finite object can be encoded into a string. A data object can be a random variable, sequence of events, a temporal graph, etc.
The amount of algorithmic information contained in y about x is \(I(y:x) = K(y)  K(y \mid x^*)\), where \(x^*\) is the shortest binary program for x. Intuitively, it is the number of bits that can be saved in the description of y when the shortest description of x is already known. Algorithmic information is symmetric, i.e. \(I(y:x) {\mathop {=}\limits ^{+}} I(x:y)\), where \({\mathop {=}\limits ^{+}}\) denotes equality up to an additive constant, and therefore also called algorithmic mutual information [20]. Two strings x and y are algorithmically independent if they have no algorithmic mutual information, i.e. \(I(x:y) {\mathop {=}\limits ^{+}} 0\).
3 Causal inference by Kolmogorov complexity
Suppose we are given data over the joint distribution of two random variables \(X\) and \(Y\) of which we know they are dependent. We are interested in inferring the most likely causal relationship between \(X\) and \(Y\). In other words, we want to infer whether \(X\) causes \(Y\), whether \(Y\) causes \(X\), or whether the two are merely correlated. To do so, we assume causal sufficiency. That is, we assume that there is no confounding variable \(Z\) that is the common cause of both \(X\) and \(Y\).
We base our causal inference method on the following postulate.
Postulate 1
(Independence of input and mechanism [30]) If \(X\) is the cause of \(Y\), \(X \rightarrow Y\), the marginal distribution of the cause \(P \left( X \right) \), and the conditional distribution of the effect given the cause, \(P (Y \mid X)\) are “independent”—\(P \left( X \right) \) contains no information about \(P (Y \mid X)\) and vice versa.
We can think of conditional \(P (Y \mid X)\) as the mechanism that transforms observations of \(X\) into observations of \(Y\), i.e. generates effect \(Y\) for cause \(X\). The postulate is plausible if this mechanism does not care how its input was generated, i.e. it is independent of \(P \left( X \right) \). Importantly, this independence does not hold in the opposite direction as \(P \left( Y \right) \) and \(P (X \mid Y)\) both inherit properties from \(P (Y \mid X)\) and \(P \left( X \right) \) and hence will contain information about each other. This creates an asymmetry between cause and effect.
It is insightful to consider the example of solar power, where it is intuitively clear that the amount of radiation per \( cm ^2\) solar cell (cause) causes the generation of electricity in the cell (effect). It is relatively easy to change \(P \left( cause \right) \) without affecting \(P ( effect \mid cause )\), as we can take actions such as, for example, moving the solar cell to a more sunny or more shady place, and varying its angle to the sun—note that while this will of course change the overall power output of the cell, it does not change the conditional distribution of the effect given the cause. If the same amount of radiation hits the cell, it will generate the same amount of power, after all. Likewise, it is easy to change \(P ( effect \mid cause )\) without affecting \(P \left( cause \right) \). We can do so, for instance, by using more efficient cells—while this may again change the overall power output of the cell, it does not affect the distribution of the incoming radiation. It is surprisingly hard, however, to do the same in the anticausal direction. That is, it is difficult to find actions that only change the distribution of the \( effect \), \(P \left( effect \right) \), while not affecting \(P ( cause \mid effect )\) or vice versa, as through their causal connection these two are intrinsically (more) dependent on each other.
The notion of independence in Postulate 1 is abstract, however. That is, to put the postulate to practice, one needs to choose and formalise an independence score. To this end, different formalisations have been proposed. Janzing et al. [16], for example, define independence in terms of information geometry, Liu and Chan [21] formulate independence in terms of the distance correlation between marginal and conditional empirical distribution, whereas Janzing and Schölkopf [12] formalise independence using algorithmic information theory, and postulate algorithmic independence of \(P \left( X \right) \) and \(P (Y \mid X)\).
Since any physical process can be simulated on a Turing machine [7], it can, in theory, capture all possible dependencies that can be explained with a physical process. As such, the algorithmic model of causality has particularly strong theoretical foundations, and provides a better mathematical formalisation of Postulate 1. Using algorithmic independence, we arrive at the following postulate.
Postulate 2
(Algorithmic independence of Markov kernels [12]) If \(X\) is the cause of \(Y\), \(X \rightarrow Y\), the marginal distribution of the cause \(P \left( X \right) \) and the conditional distribution of the effect given the cause \(P (Y \mid X)\) are algorithmically independent, i.e. \(I\left( P \left( X \right) :P (Y \mid X) \right) {\mathop {=}\limits ^{+}} 0\).
The algorithmic independence between \(P \left( X \right) \) and \(P (Y \mid X)\) implies that the shortest description, in terms of Kolmogorov complexity, of the joint distribution \(P \left( X, Y \right) \) is given by separate descriptions of \(P \left( X \right) \) and \(P (Y \mid X)\) [12]. As a consequence of the algorithmic independence of input and mechanism we have the following theorem.
Theorem 1
That is, if \(X\) causes \(Y\), factorising the joint distribution \(P \left( X, Y \right) \) into \(P \left( X \right) \) and \(P (Y \mid X)\) will lead, in terms of Kolmogorov complexity, to simpler descriptions of the distributions than factorising it into \(P \left( Y \right) \) and \(P (X \mid Y)\). Note that the total complexity of the causal model \(X \rightarrow Y\) is given by the complexity of the marginal distribution of the cause \(P \left( X \right) \) and the complexity of the conditional distribution of the effect given the cause \(P (Y \mid X)\).
With that, we can perform causal inference by simply identifying that direction between \(X\) and \(Y\) where factorization of the joint distribution yields the lowest total Kolmogorov complexity. Although this inference rule has sound theoretical foundations, Kolmogorov complexity is not computable—due to the halting problem. We can approximate Kolmogorov complexity from above, however, through lossless compression [20]. More generally, the Minimum Description Length (MDL) principle [9, 28] provides a statistically sound and computable means for approximating Kolmogorov complexity [9, 37]. Next, we discuss how MDL can be used for causal inference.
4 Causal inference by compression
The Minimum Description Length (MDL) [28] principle is a practical version of the Kolmogorov complexity. Both embrace the slogan Induction by Compression. Instead of all possible programs, MDL considers only those programs for which we know they generate x and halt. That is, lossless compressors. The more powerful the compressor, the closer we are to Kolmogorov complexity. Ideal MDL, which considers all programs that generate x and halt, coincides with Kolmogorov complexity.
The MDL principle has its root in the twopart decomposition of Kolmogorov complexity [20, Ch. 5]. It can be roughly described as follows.
Minimum Description Length Principle

\(L (M)\) is the length, in bits, of the description of the model, and

\(L (D \mid M) \) is the length, in bits, of the description of the data when encoded with \(M\).
Intuitively, \(L (M) \) represents the compressible part of the data, and \(L (D \mid M) \) represents the noise in the data. In general, a model is a probability measure, and the set of models is a parametric collection of such models. Note that MDL requires the compression to be lossless in order to allow for fair comparison between different models \(M \in \mathcal {M} \).
The algorithmic causal inference rule is based on the premise that we have access to the true distribution. In practice, we of course do not know this distribution and we only have observed data. MDL eliminates the need for assuming a distribution, as it instead identifies the model from the class that best describes the data. The total encoded size, which takes into account both how well the model fits the data as well as the complexity of the model, therefore functions as a practical instantiation of \(K(P(\cdot ))\).

If \(L _{X \rightarrow Y} < L _{Y \rightarrow X}\), we infer \(X \rightarrow Y \).

If \(L _{X \rightarrow Y} > L _{Y \rightarrow X}\), we infer \(Y \rightarrow X \).

If \(L _{X \rightarrow Y} = L _{Y \rightarrow X}\), we are undecided.
To use these indicators in practice, we have to define what causal model class \(\mathcal {M}\) we use, how to describe a model \(M \in \mathcal {M} \) in bits, how to encode a dataset \(D \) given a model \(M \), and how to efficiently approximate the optimal \(M ^* \in \mathcal {M} \). We discuss this in the next section.
5 Causal inference by treebased compressors
To apply the MDLbased causal inference rule in practice, we need a class of models suited for causal inference. As such, the model class must allow to causally explain \(Y\) given \(X\) and vice versa. One such model class is that of decision trees. A decision tree allows us to model dependencies on other attributes by splitting, i.e. conditionally describe the data of an attribute \(X_i\) given an attribute \(X_j\). In other words, decision trees can model local dependencies between variables that can identify parts of the data that causally depend on each other. Note that this comes close to the spirit of average treatment effect in randomised experiments [29].
As models we consider sets of decision trees such that we have one decision tree per attribute in the data. The dependencies between variables modelled by these trees induce a directed graph. To ensure lossless decoding, there needs to be an order on the variables in a graph. It is easy to see that there exists an order of the variables if an only if the graph is acyclic. Hence, we enforce that there are no cyclic dependencies between variables across these trees.
Next we instantiate the MDLbased causal inference framework for binary data. As such, we require a compressor for binary data that uses a set of decision trees as its model class. Importantly, the compressor should consider both the complexity of the model and that of the data under the model into account. One such compressor that fits our requirements is Pack [36]. In particular, we build upon Pack to instantiate the MDLbased causal score. Next we briefly explain how Pack works.
5.1 Treebased compressor for binary data
For example, consider a hypothetical binary dataset with three attributes \(X _1\), \(X _2\), and \(X _3\). Pack aims at discovering the set of trees such that we can encode the whole data in as few as possible bits. In Fig. 2a–c we give an example of the trees Pack could discover. As the figure shows, \(X _1\) depends on \(X _2\) and \(X _3\) depends on both \(X _1\) and \(X _2\). These trees identify both local causal dependencies, as well as the global causal DAG shown in Fig. 2d.
5.2 Pack as an information measure
The algorithmic independence of Markov kernels (Postulate 2) links observations to causality: we can reject a causal hypothesis if the algorithmic independence of Markov kernels is violated [12]. The notion of algorithmic independence, however, uses Kolmogorov complexity as an information measure, and is hence incomputable. While we know that MDL provides a wellfounded way to approximate Kolmogorov complexity in general, the question remains whether this also holds for causal inference, and in particular, whether this holds for our Pack score. The answer is yes. Steudel et al. [35] show that independence of Markov kernels is justified when we use a compressor as an information measure, if we restrict ourselves to the class of causal mechanisms that is adapted to the information measure. In general, let \(\mathcal {X}\) be a set of discretevalued random variables and \(\varOmega \) be the powerset of \(\mathcal {X}\), i.e. the set of all subsets of \(\mathcal {X}\). We then have the following definition of an information measure.
Definition 1
 (a)
normalization: \(R(0) = 0\),
 (b)
monotonicity: \(X \le Y\) implies \(R(X) \le R(Y)\) for all \(X,Y \in \varOmega \),
 (c)
submodularity: \(R(X \cup Z)  R(X) \ge R(Y \cup Z)  R(Y)\) for all \(X, Y \in \varOmega \), \(X \subseteq Y\), and for all \(Z \notin Y\).
 (a)
Pack trivially satisfies the normalization property.
 (b)
We examine the monotonicity property under subset restriction. If \(X \subseteq Y\), we can decompose Y into X and Z such that \(Y = X \cup Z\). Then \(L (Y) = L (X \cup Z) = L (X) + L (Z \mid X) \ge L (X) \). This shows that Pack score is monotonic.
 (c)
We have \(L (X \cup Z)  L (X) = L (Z \mid X) \) and \(L (Y, Z)  L (Y) = L (Z \mid Y) \). Since \(X \subseteq Y\), and providing Pack more possibilities to split on can only improve compression, \(L (Z \mid X) \ge L (Z \mid Y) \). Therefore, \(L (X \cup Z)  L (X) \ge L (Y \cup Z)  L (Y) \), which implies that Pack is submodular.
Next we discuss how to compute our MDLbased causal score using Pack.
5.3 Instantiating the MDL score with Pack
To compute \(L (X, M _X) \), we can simply compress X using Pack. However, computing \(L (Y, M _{Y \mid X}\mid X) \) is not straightforward, as Pack does not support conditional compression offtheshelf. Clearly, it does not suffice to simply compress \(X\) and \(Y\) together as this gives us \(L (X Y, M _{X Y}) \) which may use any acyclic dependency between \(X \) and \(Y \) and vice versa. When computing \(L _{X \rightarrow Y}\) or \(L (Y, M _{Y \mid X}) \), however, we do not want the attributes of \(X\) to depend on the attributes of \(Y\). Therefore, we modify line 8 of GreedyPack such that an attribute of \(X\) is only allowed to split on other attributes of \(X\), and an attribute of \(Y\) is allowed to split on both the attributes of \(X\) and the other attributes of \(Y\).
From here onwards, we refer to the Packbased instantiation of the causal score as Origo, which means origin in latin. Although our focus is primarily on binary data, we can infer causal direction from categorical data as well. To this end, we can binarise the categorical data creating a binary feature per value. As the implementation of Pack already provides this feature, we do not have to binarise categorical data ourselves. Moreover, as we will see in the experiments, with a proper discretization, we can even reliably infer causal directions from discretised continuous realvalued data.
5.4 Computational complexity
Next we analyse the computational complexity of Origo. To compute \(L _{X \rightarrow Y}\), we have to run Pack only once. Greedy Pack uses the ID3 algorithm to construct binary decision trees, therewith the computational complexity of Greedy Pack is \(\mathcal {O}(2^m n)\), where n is the number of rows in the data, and m is the total number of attributes in X and Y, i.e. \(m=X+Y\). To infer the causal direction, we have to compute both \(L _{X \rightarrow Y}\) and \(L _{Y \rightarrow X}\). Therefore, in the worst case, the computational complexity of Origo is \(\mathcal {O}(2^m n)\). In practice, Origo is fast and completes within seconds.
6 Related work
Inferring causal direction from observational data is a challenging task if no controlled randomised experiments are available. Due to its importance in practice, however, causal inference has recently seen increased attention [12, 25, 31, 34]. Most proposed causal inference frameworks are limited in practice, however, as they rely on strong assumptions, or have been defined only for either continuous realvalued, or discrete numeric data.
Constraintbased approaches like the conditional independence test [25, 34] require at least three observed random variables. Moreover, these constraintbased approaches cannot distinguish Markov equivalent causal DAGs [38] as the factorization of the joint distribution \(P \left( X, Y \right) \) is the same in both directions, i.e. \(P \left( X \right) P (Y \mid X) = P \left( Y \right) P (X \mid Y) \). Hence, they cannot decide between \(X \rightarrow Y\) and \(Y \rightarrow X\).
There do exist methods that can infer the causal direction from two random variables. Generally, they exploit the sophisticated properties of the joint distribution. The linear trace method [14, 42] infers linear causal relations of the form \(Y = AX\), where A is the structure matrix that maps the cause to the effect, using the linear trace condition which operates on A, and the covariance matrix of X, \(\varSigma _X\). The kernelized trace method [4] can infer nonlinear causal relations, but requires the causal relation to be deterministic, functional, and invertible. In theory, we do not make any assumptions on the causal relation between variables.
One of the key frameworks for causal inference is the Additive Noise Models (ANM) [11, 27, 31, 41]. The ANM assume that the effect is governed by the cause and an additive noise, and the causal inference is done by finding the direction that admits such a model. Peters et al. [26] propose an ANM for discrete numeric data. However, regression is not ideal for modelling nominal variables. Furthermore, it only works with univariate cause–effect pairs.
Algorithmic information theory provides a sound general theoretical foundation for causal inference [12]. As such, causality is defined in terms of the algorithmic similarity between data objects. In particular, for two random variables \(X\) and \(Y\), if \(X\) causes \(Y\), the shortest description of the joint distribution \(P \left( X, Y \right) \) is given by the separate description of the marginal distribution of the cause \(P \left( X \right) \) and the conditional distribution of the effect given the cause \(P (Y \mid X)\) [12]. The algorithmic information theoretic viewpoint of causality is more general in the sense that any physical process can be simulated by a Turing machine. Janzing and Steudel [13] use it to justify the ANMbased causal discovery.
Kolmogorov complexity, however, is not computable. To perform causal inference based on algorithmic information theoretic frameworks therefore requires (efficiently) computable notions of independence or information. The informationgeometric approach [16] defines independence in terms of the orthogonality in information space. Sgouritsa et al. [30] define independence in terms of the accuracy of the estimation of conditional distribution using corresponding marginal distribution. Janzing and Schölkopf [12] sketch how comparing marginal distributions, and resourcebounded computation could be used to infer causal direction, but do not give practical instantiations. Vreeken [39] proposed Ergo, a causal inference framework based on relative conditional complexities, \(K(Y \mid X) / K(Y)\) and \(K(X \mid Y) / K(X)\), that infers the direction with the lowest relative complexity. To apply this method in practice for univariate and multivariate continuous realvalued data, Vreeken instantiates it using cumulative entropy.
All above methods consider numeric data only. Causal inference on observational binary data has seen much less attention. The classic proposal by Silverstein et al. [32] uses conditional independence test, and hence requires an independent variable \(Z\) to tell whether \(X\) and \(Y\) have any causal relation. A very recent proposal by Liu and Chan [21] defines independence in terms of the distance correlation between empirical distributions \(P \left( X \right) \) and \(P (Y \mid X)\) and proposes Dc to infer the causal direction from nominal data. In the experiments, we will compare to Dc directly. In addition, we will compare to the Ergo score [39], instantiating it with Pack as \(L(Y, M_{Y \mid X} \mid X) / L(Y, M_Y)\) and vice versa.
7 Experiments
We implemented Origo in Python and provide the source code for research purposes, along with the used datasets, and synthetic dataset generator.^{3} All experiments were executed singlethreaded on MacBook Pro with 2.5 GHz Intel Core i7 processor and 16 GB memory running Mac OS X. We consider synthetic, benchmark, and realworld data. We compare Origo against the Ergo score [39] instantiated with Pack, and Dc [21].
7.1 Synthetic data
To evaluate Origo on the data with known ground truth, we consider synthetic data. In particular, we generate binary data \(X\) and \(Y\) such that attributes in \(Y\) probabilistically depend on the attributes of \(X\), termed here onwards dependency. Throughout the experiments on synthetic data, we generate \(X\) of size \(5000\hbox {by}k\), and \(Y\) of size \(5000\hbox {by}l\).
To this end, we generate data on a perattribute basis. First, we assume the ordering of attributes—the ordering of attributes in \(X\) followed by the ordering of attributes in \(Y\). Then, for each attribute, we generate a binary decision tree. In doing so, we only consider the attributes preceding it in the ordering as candidate nodes for its decision tree. Then, each row is generated by following the ordering of attributes, and using their corresponding decision trees. Further, we use the split probability to control the depth/size of the tree. We randomly choose weighted probabilities for the presence/absence of leaf attributes.
With the above scheme, with high probability, we generate data with a strong dependency in one direction. In general, we expect this direction to be the true causal direction, i.e. \(X \rightarrow Y\). Although unlikely, it is possible that the model in the reverse direction is superior. Moreover, unless we set the split probability to 1.0, however, it is possible that by chance we generate pairs without dependencies, and hence without a true causal direction. Unless stated otherwise we choose not to control for either case, by which at worst we underestimate the performance of Origo.
All reported values are averaged over 500 samples unless stated otherwise.
7.1.1 Performance
First we examine the effect of dependency on various metrics—the percentage of correct inferences (accuracy), the percentage of indecisive inferences, and the percentage of incorrect inferences. We start with \(k = l = 3\). We fix the split probability to 1.0, and generate trees with the maximum possible height, i.e. \(k + l 1 = 5\). In Fig. 3a, we give the plot showing various metrics at various dependencies for the generated pairs. We see that with the increase in dependency, indecisiveness quickly drops to zero, while accuracy increases sharply towards 90%. Note that at zero dependency, there are no causal edges; hence, Origo is correct in being indecisive.
Next we study the effect of the maximum height \(h\) of the trees on the accuracy of Origo. We set \(k = l = 3\), and the split probability to 1.0. In Fig. 3b, we observe that the accuracy gets higher as \(h\) increases. This is due to the increase in the number of causal edges with the increase in the maximum height of the tree. Although the increase in accuracy is quite large when we move from \(h\) \(=\) 1 to 2, it is almost negligible when we move from \(h\) \(=\) 2 onwards. This shows that Origo already infers the correct causal direction even when there are only few causal dependencies in the generating model.
Next we investigate the accuracy of Origo on cause–effect pairs with asymmetric number of attributes. For that, we fix the split probability to 1.0, and generate trees with the maximum possible height. At every level of dependency, we generate 500 cause–effect pairs, 250 of which with \(k =1,\ l =3\) and remaining 250 with \(k =3,\ l =1\). In particular, we consider those pairs for correctness where there is at least one causal edge from \(X\) to \(Y\). In Fig. 5a, we give the plot comparing the accuracy of Origo against Ergo and Dc. We see that Origo performs much better than the other methods. In particular, the difference in accuracy gets larger as the dependency increases. We also note that the performance of Dc has a striking resemblance to flipping a fair coin.
Next we consider the symmetric case where \(k = l = 3\). We fix the split probability to 1.0, and generate trees with the maximum possible height. As in the asymmetric case, we consider those pairs for correctness where there is at least one causal edge from \(X\) to \(Y\). In Fig. 5b, we show the plot comparing the accuracy of Origo against Ergo, and Dc. We see that both Origo performs as good as or better than other methods. We note that for the pairs without dependency, Dc infers a causal relationship in over 50% of the cases.
7.1.2 Dimensionality
Next we study the robustness against dimensionality. First we consider cause–effect pairs with symmetric number of attributes, i.e. \(k = l \), and vary it between 1 and 10. We fix the dependency to 0.7, the split probability to 1.0, and the maximum height of trees to 5. In particular, we compare Origo against Ergo and Dc. In Fig. 6a, we see that Origo is highly accurate in every setting. With the exception of the univariate case, Ergo also performs well when both \(X\) and \(Y\) have the same cardinality.
7.1.3 Type I error
To evaluate whether Origo infers relevant causal direction, we employ swap randomization [8]. Swap randomization is an approach to producing random datasets by altering the internal structure of the data while preserving its row and column margins. The internal structure of the data is altered by successive swap operations, which correspond to steps in a Markov chain process.
More formally, given a binary data matrix, D, with n rows and m columns, we randomly identify four cells in D characterised by a combination of row indices \(r_1, r_2 \in \{1, 2, \dots , n\}\) and column indices \(c_1, c_2 \in \{1, 2, \dots , m\}\) such that \(D_{r_1, c_1} \ne D_{r_1, c_2}\) and \(D_{r_2, c_1} \ne D_{r_2, c_2}\), but \(D_{r_2, c_1} = D_{r_1, c_2}\) and \(D_{r_1, c_1} = D_{r_2, c_2}\). Then, we swap the values of these four cells either in clockwise or in anticlockwise direction. The swap operation is performed repeatedly until the data mix sufficiently enough to break the internal structure of the data, also called mixing time of a Markov chain. Although there is no optimal theoretical bound for the mixing time of a Markov chain, Gionis et al. [8] empirically suggest the number of swap operations to be in the order of number of 1s in the data.
The key idea behind significance testing with swap randomization is to create several random datasets with the same row and column margins as the original data, run the data mining algorithm on those data, and see if the results differ significantly between the original data and random datasets.
7.1.4 Type II error
To assess whether Origo identifies causal relationship when causal relationship really exists, we test its statistical power. The null hypothesis is that there is no causal relationship between cause–effect pairs. To determine the cutoff for testing the null hypothesis, we first generate 250 cause–effect pairs with no causal relationship. Then we compute their \(\varDelta \) values and set the cutoff \(\varDelta \) value at a significance level of 0.05. Next we generate new 250 cause–effect pairs with causal relationship. The statistical power is the proportion of the 250 new cause–effect pairs whose \(\varDelta \) value exceeds the cutoff delta value.
We set \(k = l = 3\), and the split probability to 1.0 and generate trees with the maximum possible height. We show the results in Fig. 7b. The lines corresponding to Origo and Ergo overlap as both have the same high statistical power, outperforming Dc in every setting.
Last but not least, we observe that for all the above experiments inferring the causal direction for one pair typically takes only up to a few seconds. Next we evaluate Origo on realworld data.
7.2 Realworld data
Next, we evaluate Origo on realworld data.
7.2.1 Univariate pairs
First we evaluate Origo on benchmark cause–effect pairs with known ground truth [23]. In particular, we here consider the 95 univariate pairs. So far, there does not exist a discretisation strategy that provably preserves the causal relationship between variables. To complicate matters further we do not know the underlying domain of the data, and each cause–effect pair is from a different domain. Hence, for exposition we enforce one discretisation strategy over all the pairs.
We considered various discretisation strategies—including equifrequency and equiwidth binning, MDLbased histogram density estimation [19], and parameterfree unsupervised interactionpreserving discretisation (Ipd) [24]. Overall, we obtained the best results using Ipd using its default parameters and will report these below.
Next we investigate the accuracy of Origo against the fraction of decisions Origo is forced to make. To this end, we sort the pairs by their absolute score difference \(\varDelta \) in two directions in descending order. Then we compute the accuracy over top\(k\%\) pairs. The decision rate is the fraction of top cause–effect pairs that we consider. Alternatively, it is also the fraction of cause–effect pairs whose \(\varDelta \) is greater than some threshold \(\varDelta _t\). For undecided pairs, we flip a coin. For other methods, we follow the similar procedure with their respective absolute score difference.
7.2.2 Multivariate pairs
Next we evaluate Origo quantitatively on realworld data with multivariate pairs. For that we consider four cause–effect pairs with known ground truth taken from [23]. The Chemnitz dataset is taken from Janzing et al. [15], whereas the Car dataset is from the UCI repository.^{4} We again use Ipd to discretise the data. We give the base statistics in Table 1. For each pair, we report the number of rows, the number of attributes in \(X\), the number of attributes in \(Y\), and the ground truth. Furthermore, we report the results of Origo, Ergo, and Dc.
Results on Tübingen multivariate cause–effect pairs [23]
Dataset  #rows  \(X \)  \(Y \)  Truth  Origo  Ergo  Dc 

Weather forecast  10,226  4  4  \(Y \rightarrow X\)  \(\)  \(\checkmark \)  \(\) 
Ozone  989  1  3  \(Y \rightarrow X\)  \(\checkmark \)  \(\checkmark \)  \(\times \) 
AutoMpg  392  3  2  \(X \rightarrow Y\)  \(\checkmark \)  \(\checkmark \)  \(\times \) 
Radiation  72  16  16  \(Y \rightarrow X\)  \(\times \)  \(\times \)  \(\times \) 
Chemnitz  1440  3  7  \(X \rightarrow Y\)  \(\checkmark \)  \(\times \)  \(\checkmark \) 
Car  1728  6  1  \(X \rightarrow Y\)  \(\checkmark \)  \(\checkmark \)  \(\checkmark \) 
7.3 Qualitative results
Last, we consider whether Origo provides results that agree with intuition. To this end we consider three case studies.
7.3.1 Acute inflammation
The acute inflammation dataset is taken from the UCI repository (see footnote 4). It consists of the presumptive diagnosis of two diseases of urinary system for 120 potential patients. There are 6 symptoms—temperature of the patient (\(X _1\)), occurrence of nausea (\(X _2\)), lumber pain (\(X _3\)), urine pushing (\(X _4\)), micturition pains (\(X _5\)), and burning of urethra, itch, swelling of urethra outlet (\(X _6\)). All the symptoms are binary but the temperature of the patient, which takes a real value between 35–42 \(^{\circ }\)C. The two diseases for diagnosis are inflammation of urinary bladder (\(Y _1\)) and nephritis of renal pelvis origin (\(Y _2\)).
We discretise the temperature into two bins using Ipd. This results in two binary attributes \(X _{11}\) and \(X _{12}\). We then run Origo on the pair \(X,Y \), where \(X =\{X _{11}, X _{12}, X _3, X _4, X _5, X _6\}\) and \(Y =\{Y _1, Y _2\}\). We find that \(Y \rightarrow X\). That is, Origo infers that the diseases cause the symptoms, which is in agreement with intuition.
7.3.2 ICDM abstracts
Next we consider the ICDM abstracts dataset, which is available from the authors of [6]. This dataset consists of abstracts—stemmed and stop words removed—of 859 papers published at the ICDM conference until the year 2007. Each abstract is represented by a row, and words are the attributes.
We use Opus Miner on the ICDM abstracts dataset to discover top 100 selfsufficient itemsets [40]. Then, we apply Origo on those 100 selfsufficient itemsets. We sort the discovered causal directions by their \(\varDelta \) value in descending order. In Table 2, we give 8 highly characteristic and nonredundant results along with their \(\varDelta \) values taken from top 17 causal directions. We expect the causal directions having higher \(\varDelta \) values to show clear causal connection, and indeed, we see that this is the case.
Results of Origo on ICDM. We give 8 characteristic and nonredundant exemplars drawn from top 17 causal directions
Discovered causal direction  \(\varDelta \) (bits) 

frequent itemset \(\rightarrow \) mining  4.809964 
fp \(\rightarrow \) tree  0.880654 
drift \(\rightarrow \) concept  0.869090 
anomaly \(\rightarrow \) detection  0.804479 
lda \(\rightarrow \) linear  0.772805 
neural \(\rightarrow \) network  0.748579 
walk \(\rightarrow \) random  0.701649 
social \(\rightarrow \) network  0.694999 
7.3.3 Census
The Adult dataset is taken from the UCI repository and consists of 48 832 records from the census database of the USA in 1994. Out of 14 attributes, we consider only four—workclass, education, occupation, and income. In particular, we binarise workclass attribute into four attributes as private, selfemployed, publicservant, and unemployed. We binarise education attribute into seven attributes as dropout, associates, bachelors, doctorate, hsgraduate, masters, and profschool. Further, we binarise occupation attribute into eight attributes as admin, armedforce, bluecollar, whitecollar, service, sales, professional, and otheroccupation. Lastly, we binarise income attribute into two attributes as \(>50K\) and \(\le 50K\).
Results of Origo on Adult
Discovered causal direction  \(\varDelta \) (bits) 

publicservant admin hsgraduate \(\rightarrow \le 50K\)  9.917098 
publicservant professional doctorate \(\rightarrow >50K\)  8.053542 
bachelors selfemployed whitecollar \(\rightarrow >50K\)  7.719200 
publicservant professional masters \(\rightarrow >50K\)  7.583210 
hsgraduate bluecollar \(\rightarrow \le 50K\)  5.209738 
These case studies show that Origo discovers sensible causal directions from realworld data.
8 Discussion
The experiments show that Origo works well in practice. Origo reliably identifies true causal structure regardless of cardinality and skew, with high statistical power, even at low level of causal dependencies. On benchmark data it performs very well, despite information loss through discretization. Moreover, the qualitative case studies show that the results are sensible.
Although these results show the strength of our framework, and of Origo in particular, we see many possibilities to further improve. For instance, Pack does not work directly on categorical data. By binarizing the categorical data, it can introduce undue dependencies. This presents an inherent need for a lossless compressor that works directly on categorical data which is likely to improve the results.
Further, we rely on discretization strategies to discretise continuous realvalued data. We observe different results on continuous realvalued data depending on the discretization strategy we pick. It would make an engaging future work to devise a discretization strategy for continuous realvalued data that preserves causal dependencies. Alternatively, it will be interesting to instantiate the framework using regression trees to directly consider realvalued data. This is not trivial, as it requires both a encoding scheme for this model class and efficient algorithms to infer good sets of trees.
Our framework is based on causal sufficiency assumption. Extending Origo to include confounders is another avenue of future work. Moreover, our inference principle is defined over data in general, yet we restricted our analysis to binary, categorical, and continuous realvalued data. It would be interesting to apply our inference principle on time series data. To instantiate our MDL framework the only thing we need is a lossless compressor that can capture directed relations on multivariate time series data.
9 Conclusion
We considered causal inference from observational data. We proposed a framework for causal inference based on Kolmogorov complexity, and gave a generally applicable and computable framework based on the minimum description length (MDL) principle.
To apply the framework in practice, we proposed Origo, an efficient method for inferring the causal direction from binary data. Origo uses decision trees to encode data, works directly on the data, and does not require assumptions about either distributions or the type of causal relations. Extensive evaluation on synthetic, benchmark, and realworld data showed that Origo discovers meaningful causal relations, and outperforms the state of the art.
Footnotes
Notes
Acknowledgements
Kailash Budhathoki is supported by the International Max Planck Research School for Computer Science (IMPRSCS). The authors are supported by the Cluster of Excellence “Multimodal Computing and Interaction” within the Excellence Initiative of the German Federal Government. Open access funding provided by Max Planck Society.
References
 1.Breiman L (1996) Bagging predictors. Mach Learn 24(2):123–140MathSciNetMATHGoogle Scholar
 2.Budhathoki K, Vreeken J (2016) Causal inference by compression. In: Proceedings of the 16th IEEE international conference on data mining (ICDM), Barcelona, Spain, IEEEGoogle Scholar
 3.Chaitin GJ (1969) On the simplicity and speed of programs for computing infinite sets of natural numbers. J ACM 16(3):407–422MathSciNetCrossRefMATHGoogle Scholar
 4.Chen Z, Zhang K, Chan L (2013) Nonlinear causal discovery for high dimensional data: a kernelized trace method. In: Proceedings of the 13th IEEE international conference on data mining (ICDM), Dallas, TX, pp 1003–1008Google Scholar
 5.Cover TM, Thomas JA (2006) Elements of information theory. WileyInterscience, New YorkMATHGoogle Scholar
 6.De Bie T (2011) Maximum entropy models and subjective interestingness: an application to tiles in binary databases. Data Min Knowl Discov 23(3):407–446MathSciNetCrossRefMATHGoogle Scholar
 7.Deutsch D (1985) Quantum theory, the Church–Turing principle and the universal quantum computer. Proc R Soc A (Math Phys Eng Sci) 400(1818):97–117MathSciNetCrossRefMATHGoogle Scholar
 8.Gionis A, Mannila H, Mielikäinen T, Tsaparas P (2007) Assessing data mining results via swap randomization. ACM Trans Knowl Discov Data 1(3):167–176CrossRefGoogle Scholar
 9.Grünwald P (2007) The minimum description length principle. MIT Press, CambridgeGoogle Scholar
 10.Grünwald PD, Vitányi PMB (2008) Algorithmic information theory. CoRR arXiv:0809.2754
 11.Hoyer P, Janzing D, Mooij J, Peters J, Schölkopf B (2009) Nonlinear causal discovery with additive noise models. In: Proceedings of the 22nd annual conference on neural information processing systems (NIPS), pp 689–696Google Scholar
 12.Janzing D, Schölkopf B (2010) Causal inference using the algorithmic Markov condition. IEEE Trans Inf Technol 56(10):5168–5194MathSciNetCrossRefMATHGoogle Scholar
 13.Janzing D, Steudel B (2010) Justifying additive noise modelbased causal discovery via algorithmic information theory. Open Syst Inf Dyn 17(2):189–212MathSciNetCrossRefMATHGoogle Scholar
 14.Janzing D, Hoyer P, Schölkopf B (2010a) Telling cause from effect based on highdimensional observations. In: Proceedings of the 27th international conference on machine learning (ICML), Haifa, Israel, pp 479–486Google Scholar
 15.Janzing D, Hoyer P, Schölkopf B (2010b) Telling cause from effect based on highdimensional observations. In: Proceedings of the 27th international conference on machine learning, International Machine Learning Society, pp 479–486Google Scholar
 16.Janzing D, Mooij J, Zhang K, Lemeire J, Zscheischler J, Daniušis P, Steudel B, Schölkopf B (2012) Informationgeometric approach to inferring causal directions. Artif Intell 182–183:1–31MathSciNetCrossRefMATHGoogle Scholar
 17.Kolmogorov AN (1965) Three approaches to the quantitative definition of information. Probl Inf Transm 1:1–7Google Scholar
 18.Kontkanen P, Myllymäki P (2007) A lineartime algorithm for computing the multinomial stochastic complexity. Inf Process Lett 103(6):227–233MathSciNetCrossRefMATHGoogle Scholar
 19.Kontkanen P, Myllymäki P (2007) MDL histogram density estimation. In: Proceedings of the eleventh international conference on artificial intelligence and statistics (AISTATS), San Juan, Puerto RicoGoogle Scholar
 20.Li M, Vitányi P (1993) An introduction to Kolmogorov complexity and its applications. Springer, BerlinCrossRefMATHGoogle Scholar
 21.Liu F, Chan L (2016) Causal inference on discrete data via estimating distance correlations. Neural Comput 28(5):801–814CrossRefGoogle Scholar
 22.Mooij JM, Stegle O, Janzing D, Zhang K, Schölkopf B (2010) Probabilistic latent variable models for distinguishing between cause and effect. In: Proceedings of the 23rd annual conference on neural information processing systems (NIPS), Vancouver, BC, Curran, pp 1687–1695Google Scholar
 23.Mooij JM, Peters J, Janzing D, Zscheischler J, Schölkopf B (2016) Distinguishing cause from effect using observational data: methods and benchmarks. J Mach Learn Res 17(32):1–102MathSciNetMATHGoogle Scholar
 24.Nguyen HV, Müller E, Vreeken J, Böhm K (2014) Unsupervised interactionpreserving discretization of multivariate data. Data Min Knowl Discov 28(5–6):1366–1397MathSciNetCrossRefMATHGoogle Scholar
 25.Pearl J (2000) Causality: models, reasoning, and inference. Cambridge University Press, New YorkMATHGoogle Scholar
 26.Peters J, Janzing D, Schölkopf B (2010) Identifying cause and effect on discrete data using additive noise models. In: Proceedings of the international conference on artificial intelligence and statistics (AISTATS), pp 597–604Google Scholar
 27.Peters J, Mooij J, Janzing D, Schölkopf B (2014) Causal discovery with continuous additive noise models. J Mach Learn Res 15:2009–2053MathSciNetMATHGoogle Scholar
 28.Rissanen J (1978) Modeling by shortest data description. Automatica 14(1):465–471CrossRefMATHGoogle Scholar
 29.Rubin DB (1974) Estimating causal effects of treatments in randomized and nonrandomized studies. J Educ Psychol 66(5):688–701CrossRefGoogle Scholar
 30.Sgouritsa E, Janzing D, Hennig P, Schölkopf B (2015) Inference of cause and effect with unsupervised inverse regression. In: Proceedings of the international conference on artificial intelligence and statistics (AISTATS), Journal of Machine Learning Research, pp 847–855Google Scholar
 31.Shimizu S, Hoyer PO, Hyvärinen A, Kerminen A (2006) A linear nonGaussian acyclic model for causal discovery. J Mach Learn Res 7:2003–2030MathSciNetMATHGoogle Scholar
 32.Silverstein C, Brin S, Motwani R, Ullman J (2000) Scalable techniques for mining causal structures. Data Min Knowl Discov 4(2):163–192CrossRefGoogle Scholar
 33.Solomonoff RJ (1964) A formal theory of inductive inference. Part I, II. Inf Control 7:1–22MathSciNetCrossRefMATHGoogle Scholar
 34.Spirtes P, Glymour C, Scheines R (2000) Causation, prediction, and search. MIT Press, CambridgeMATHGoogle Scholar
 35.Steudel B, Janzing D, Schölkopf B (2010) Causal markov condition for submodular information measures. In: Proceedings of the 23rd annual conference on learning theory. OmniPress, pp 464–476Google Scholar
 36.Tatti N, Vreeken J (2008) Finding good itemsets by packing data. In: Proceedings of the 8th IEEE international conference on data mining (ICDM), Pisa, Italy, pp 588–597Google Scholar
 37.Vereshchagin N, Vitanyi P (2004) Kolmogorov’s structure functions and model selection. IEEE Trans Inf Technol 50(12):3265–3290MathSciNetCrossRefMATHGoogle Scholar
 38.Verma T, Pearl J (1991) Equivalence and synthesis of causal models. In: Proceedings of the 6th international conference on uncertainty in artificial intelligence (UAI), pp 255–270Google Scholar
 39.Vreeken J (2015) Causal inference by direction of information. In: Proceedings of the SIAM international conference on data mining (SDM), Vancouver, Canada, pp 909–917Google Scholar
 40.Webb G (2011) Filteredtopk association discovery. Wiley Interdiscip Rev Data Min Knowl Discov 1(3):183–192CrossRefGoogle Scholar
 41.Zhang K, Hyvärinen A (2009) On the identifiability of the postnonlinear causal model. In: Proceedings of the 25th international conference on uncertainty in artificial intelligence (UAI), pp 647–655Google Scholar
 42.Zscheischler J, Janzing D, Zhang K (2011) Testing whether linear equations are causal: a free probability theory approach. In: Proceedings of the 27nd international conference on uncertainty in artificial intelligence (UAI). AUAI Press, pp 839–847Google Scholar
Copyright information
Open AccessThis article is distributed under the terms of the Creative Commons Attribution 4.0 International License (http://creativecommons.org/licenses/by/4.0/), which permits unrestricted use, distribution, and reproduction in any medium, provided you give appropriate credit to the original author(s) and the source, provide a link to the Creative Commons license, and indicate if changes were made.