<?xml-model href='http://www.tei-c.org/release/xml/tei/custom/schema/relaxng/tei_all.rng' schematypens='http://relaxng.org/ns/structure/1.0'?><TEI xmlns="http://www.tei-c.org/ns/1.0">
	<teiHeader>
		<fileDesc>
			<titleStmt><title level='a'>Sample Complexity of Distinguishing Cause from Effect</title></titleStmt>
			<publicationStmt>
				<publisher>Proceedings of Machine Learning Research</publisher>
				<date>04/14/2023</date>
			</publicationStmt>
			<sourceDesc>
				<bibl> 
					<idno type="par_id">10477204</idno>
					<idno type="doi"></idno>
					<title level='j'>Proceedings of Machine Learning Research</title>
<idno>2640-3498</idno>
<biblScope unit="volume">206</biblScope>
<biblScope unit="issue"></biblScope>					

					<author>Jayadev Acharya</author><author>Sourbh Bhadane</author><author>Arnab Bhattacharyya</author><author>Saravanan Kandasamy</author><author>Ziteng Sun</author><author>Francisco Ruiz</author><author>Jennifer Dy</author><author>Jan-Willem van de Meent</author>
				</bibl>
			</sourceDesc>
		</fileDesc>
		<profileDesc>
			<abstract><ab><![CDATA[We study the sample complexity of causal structure learning on a two-variable system with observational and experimental data. Specifically, for two variables X and Y, we consider the classical scenario where either X causes Y , Y causes X, or there is an unmeasured confounder between X and Y. We show that if X and Y are over a finite domain of size k and are significantly correlated, the minimum number of interventional samples needed is sublinear in k. We give a tight characterization of the tradeoff between observational and interventional data when the number of observational samples is sufficiently large. We build upon techniques for closeness testing and for non-parametric density estimation in different regimes of observational data. Our hardness results are based on carefully constructing causal models whose marginal and interventional distributions form hard instances of canonical results on property testing.]]></ab></abstract>
		</profileDesc>
	</teiHeader>
	<text><body xmlns="http://www.tei-c.org/ns/1.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xmlns:xlink="http://www.w3.org/1999/xlink">
<div xmlns="http://www.tei-c.org/ns/1.0"><head n="1">Introduction</head><p>Reichenbach's Common Cause Principle states that if two variables X and Y are correlated, then either X causes Y , or Y causes X, or there is a common hidden variable U that causes both. We focus on the three situations depicted in Figure <ref type="figure">1</ref>, with the goal being to discover which of the three alternatives is true.  If the causal structure is X -&#8594; Y , then (X, Y ) is generated by the assignments: X := N X and Y := f Y (X, N Y ) where N X , N Y are independent random variables and f Y is some (deterministic) function. The roles of X and Y are interchanged if the structure is X &#8592;-Y . If the causal structure is X &#8592;-U -&#8594; Y , then (X, Y ) is generated as: U := N U , X := f X (U , N X ), and Y := f Y (U , N Y ), where f X , f Y are functions and N X , N Y , N U are independent random variables. In this formalism, an intervention corresponds to setting one of the variables to a fixed value and examining the distribution of the other. For example, if the structure is X -&#8594; Y and the intervention of fixing X to x is performed, then Y is generated as f Y (x, N Y ). We denote this intervention by do(X = x).</p><p>How would one distinguish between the three possibilities shown in Figure <ref type="figure">1</ref>? Observationally, they are impossible to distinguish as the joint distribution of (X, Y ) may be exactly the same in all three cases. But a fundamental insight of <ref type="bibr">Fisher (1925)</ref> was that they can be distinguished if interventions are allowed. For example, if the true causal structure was X -&#8594; Y , then intervening on X should have an effect on Y , while intervening on Y should have no effect on X. The situation is vice versa for X &#8592;-Y . On the other hand, if the causal structure was X &#8592;-U -&#8594; Y , intervention on neither X nor Y would affect the other.</p><p>In this work, we revisit the problem of recovering the correct causal structure from a quantitative point of view. While it is clear that a nonzero number of samples from inter-ventional distributions are necessary, what is the minimum number of such samples needed? More precisely, given m 1 , what is the minimum m 2 such that m 1 observations and m 2 samples from interventions suffice to distinguish between the possibilities in Figure <ref type="figure">1</ref>? While there is a long line of work on causal structure learning while minimizing the number of experiments (e.g., <ref type="bibr">Eberhardt (2007</ref><ref type="bibr">Eberhardt ( , 2008))</ref>; <ref type="bibr">Hauser and B&#252;hlmann (2012)</ref>; <ref type="bibr">Shanmugam et al. (2015)</ref>; <ref type="bibr">Kocaoglu et al. (2017)</ref>; <ref type="bibr">Greenewald et al. (2019)</ref>; <ref type="bibr">Squires et al. (2020)</ref>), most of these works ignore the issue of finite sample complexity that this work addresses.</p><p>We uncover a non-trivial tradeoff between m 1 and m 2 that shows that as the number of observations increases, we need fewer and fewer (but of course, positive) number of samples from interventions. Our study holds in the setting where X and Y are random variables over a finite domain of size k and are known to be "significantly correlated". For example, we show that if m 1 &#8764; k c for 2/3 c 1, then m 2 &#8764; k 1-c/2 samples are sufficient, while if m 1 is sufficiently large, m 2 can be completely independent of k. Furthermore, the tradeoffs we establish are nearly tight, in several interesting parameter regimes.</p><p>Organization We will define our problem statement in Section 1.1 and state our results precisely in Section 1.2. Related work will be discussed in Section 1.3. We present an overview of our technique in Section 1.4. Due to the page limit, we focus on the regime where m 1 = O(k) in the main paper and present the algorithm and lower bound in Section 2 and Section 3 respectively. We present detailed analysis of other regimes in the supplementary material.</p></div>
<div xmlns="http://www.tei-c.org/ns/1.0"><head n="1.1">Problem formulation</head><p>We will work in the semantic framework of structural causal models (SCMs) <ref type="bibr">(Pearl, 2009)</ref>. Below we provide a complete formulation of the problem statement by tailoring SCMs to our setting.</p><p>Our goal is to test causal relationships between two correlated discrete random variables X and Y over a domain &#931; of size k. We use a distance from independence as a notion of correlation below.</p><p>Definition 1 (TV-correlation). For discrete random variables X and Y define their total variation correlation as</p><p>where d TV (., .) is the total variation distance defined for any discrete distributions p and q over a domain &#931; of size k as</p><p>Let X and Y be two correlated random variables with &#961; TV (X, Y ) &#949; with a causal structure given in Figure <ref type="figure">1</ref>. We are given access to two types of samples:</p><p>-Observational samples. These are draws from the joint distribution P [X, Y ].</p><p>-Interventional samples. We can intervene by setting X = x for some x &#8712; &#931; (resp. Y = y) and observe samples of Y (resp. X) under the intervention. We denote the interventional distribution as</p><p>and resp. as</p><p>Note the distinction between "intervention" and "interventional samples"; the former corresponds to fixing, say X, to a certain value x whereas the latter corresponds to drawing multiple samples from the interventional distribution</p><p>In practice, an intervention corresponds to setting up a certain medical trial and the number of interventional samples correspond to the number of people participating in the trial.</p><p>As discussed in the introduction, suppose we intervene do (X = x), if we are in Figure <ref type="figure">1</ref>(a), the samples of Y we obtain satisfies</p><p>whereas if we are in Figure <ref type="figure">1</ref>(b) or Figure <ref type="figure">1</ref>(c) then we just obtain samples from P [Y ] as there is no causal influence from X to Y .</p><p>We now define our Causal Structure Identification problem.</p><p>Definition 2 (Causal Structure Identification). Suppose an SCM on two observable random variables X, Y supported over &#931; of size k satisfies &#961; TV (X, Y ) &#949;. Given m 1 observational and m 2 interventional samples, an algorithm solves Causal Structure Identification problem (CSI(k, &#949;)) if with probability at least 2/3, the algorithm outputs:</p><p>Note that while the above definition is for the case of constant probability, we can boost the success of our algorithms to 1 -&#948; for an arbitrary &#948; &gt; 0 by the median trick, i.e., repeating a testing algorithm log (1/&#948;) times and outputting the median response which incurs only a logarithmic increase in the sample complexity.</p></div>
<div xmlns="http://www.tei-c.org/ns/1.0"><head>Number of observational samples Number of interventional samples</head><p>Figure <ref type="figure">2</ref>: An illustration of the tradeoffs between m 1 and m 2 for CSI(k, &#949;) for &#949; &gt; 1/k 1/4 . The curves in bold are tight up to logarithmic factors. For &#949; &lt; 1/k 1/4 , the curve is flat at</p></div>
<div xmlns="http://www.tei-c.org/ns/1.0"><head>&#8730;</head><p>k/&#949; 2 until m 1 approaches k 2 /&#949; 2 and then becomes O 1/&#949; 2 . The x-axis is scaled such that all regimes of interest appear equal in length.</p></div>
<div xmlns="http://www.tei-c.org/ns/1.0"><head n="1.2">Our results</head><p>We provide sample complexity bounds for solving CSI(k, &#949;) with the number of interventional samples sublinear in k. The upper bounds we obtain have interesting phase transitions on the trade-offs between the number of interventional and observational samples in different regimes. We also show that the obtained upper bounds are tight when m 1 = O(k) and m 1 &gt; k 2 /&#949; 2 . While our main tradeoffs are in terms of the interventional samples, we view it as a positive that the number of distinct interventions that our algorithms need is independent of the domain size.</p><p>Based on the number of observational samples available, we present our results in the following four regimes: (1) Zero (few) observational samples:</p><p>The bounds we obtain are shown in Figure <ref type="figure">2</ref>. The above regimes are separated when &#949; 1/k 1/4 , which is the regime where interesting phase transitions happen. We note that all described results hold for all regimes of &#949; but regimes (1)-(3) will overlap and lead to the same sample for m 2 when &#949; &lt; 1/k 1/4 . Zero (few) observational samples: m 1 = O(k 2/3 /&#949; 4/3 ).</p><p>Here we discuss the case when the number of observational samples is small. The next theorem shows that we can solve CSI(k, &#949;) with m 2 = O(k 2/3 /&#949; 4/3 ) even when the number of observational samples m 1 is zero.</p><p>Theorem 1.1. There exists an algorithm that uses zero observational samples and</p><p>Interestingly, as we will see in Theorem 1.3, the requirement on m 2 cannot be improved even when m 1 = &#920;(k 2/3 /&#949; 4/3 ). This shows that the above interventional complexity is optimal.</p><p>Sublinear observational samples: m 1 = O(k). In this case, we show that the tradeoff between observational samples and interventional samples largely resembles the tradeoff for asymmetric closeness testing (see Definition 3) <ref type="bibr">(Acharya et al., 2014a;</ref><ref type="bibr">Bhattacharya and Valiant, 2015;</ref><ref type="bibr">Diakonikolas and Kane, 2016;</ref><ref type="bibr">Diakonikolas et al., 2021)</ref>. Theorem 1.2. When m 1 = &#8486;(k 2/3 /&#949; 4/3 ), there exists an algorithm that takes m 1 observational samples and</p><p>The algorithm we use relies on asymmetric closeness testing between conditional distributions P [Y | x] (or P [X | y]) and interventional distributions P x [Y ] (or P y [X]). While asymmetric closeness testing only deals with two distributions, the causal structure identification problem involves k conditional distributions and interventional distributions, which makes it trickier to handle. To resolve this, we use Levin's investment strategy <ref type="bibr">(Levin, 1985;</ref><ref type="bibr">Goldreich, 2014)</ref> to select a sequence of conditional and interventional distributions to conduct closeness tests. See Section 1.4 for an overview of the technique.</p><p>We also prove a lower bound showing that the result is tight in the sublinear regime m 1 = O(k).</p><p>Theorem 1.3. For m 1 = &#8486;(k 2/3 /&#949; 4/3 ) and m 1 = O(k), any algorithm that takes m 1 observational samples must take m 2 = &#8486;(max(k/(</p><p>Superlinear observational samples: m 1 = &#8486;(k) and m 1 = O(k 2 /&#949; 2 ). In this regime, we obtain a sample complexity upper bound of O( &#8730; k/&#949; 2 ), which doesn't improve as m 1 increases. The result follows immediately from Theorem 1.2 as the upper bound in Theorem 1.2 doesn't improve when m 1 &gt; k. As we will see when m 1 is sufficiently large &#8486;(k 2 /&#949; 2 ), the interventional complexity can be further reduced. We leave improving the interventional complexity in the superlinear regime or proving hardness results as future work.</p><p>Sufficient observational samples: m 1 = &#937;(k 2 /&#949; 2 ). In this regime, we have enough observational samples to get near-"perfect" estimates of P [Y ] and P</p><p>Hence the problem can be solved using simple hypothesis testing between</p><p>, there exists an algorithm that takes m 1 observational samples and m 2 = O(1/&#949; 2 ) interventional samples and solves CSI(k, &#949;). Moreover, the algorithm only needs to make one distinct intervention.</p><p>We also show that this simple hypothesis testing approach is optimal with the following lower bound, showing that increasing m 1 beyond k 2 /&#949; 2 does not help reducing the intervention complexity up to logarithmic factors.</p><p>Theorem 1.5. Any algorithm that solves CSI(k, &#949;) requires m 2 = &#8486;(1/&#949; 2 ) interventional samples.</p></div>
<div xmlns="http://www.tei-c.org/ns/1.0"><head n="1.3">Related Work</head><p>Causal discovery from observational and experimental data has been subject to intense study, both from the potential outcomes <ref type="bibr">(Rubin, 1974;</ref><ref type="bibr">Rosenbaum and Rubin, 1983</ref>) and the graphical model <ref type="bibr">(Pearl, 2009</ref>) schools of causality. For the particular case of two variables, there is also a newer line of research <ref type="bibr">(Peters et al., 2017)</ref> that constrain the mechanisms underlying parent-child relationships in the causal model, allowing the causal direction to be identifiable solely from observations. A high-level account of different approaches to learn the causal direction from observations can be found in <ref type="bibr">Guyon et al. (2019)</ref>. The effect of sample size on causal structure discovery has been empirically studied in several contexts <ref type="bibr">Mooij et al. (2016)</ref>. <ref type="bibr">Compton et al. (2022)</ref> obtain finite-sample results for the two-variable system under the assumption of causal sufficiency and an assumption on the entropy of the exogenous variable. <ref type="bibr">Wadhwa and Dong (2021)</ref> study the sample complexity of causal discovery with multiple nodes by applying finite-sample conditional independence testers <ref type="bibr">Canonne et al. (2018)</ref> to the inferred causation algorithm <ref type="bibr">Pearl and Verma (1991)</ref>. <ref type="bibr">Bello and Honorio (2018)</ref> study the sample complexity of causal discovery for discrete causal Bayesian networks but have a negative dependence on a parameter quantifying the minimal causal effect that can be arbitrarily small for our setting. With access to interventions <ref type="bibr">Eberhardt et al. (2010)</ref> and <ref type="bibr">Yang et al. (2018)</ref> experimentally demonstrated how the sample complexity affects structure learning. Both these studies compared perfect interventions (the notion used here) with soft interventions; in the future, we hope to extend our theory also to soft interventions.</p><p>Some of the techniques we use were developed in the context of distribution property testing; see <ref type="bibr">Canonne (2020b)</ref> for an excellent survey. Specifically, we rely on existing work for the asymmetric closeness testing problem, where given sample access to two distributions p and q, the question is for a given m 1 number of samples from p, how many samples m 2 are required from q such that the hypothesis p = q can be distinguished from d TV (p, q) &gt; &#949; with probability at least 2/3. The problem interpolates between the case when p is known (identity testing) and when p is not known (closeness testing). Sample complexity bounds for asymmetric closeness testing were first studied by <ref type="bibr">Acharya et al. (2014b)</ref> who showed that it is sufficient to have</p><p>samples from q, where k is the size of the support of p and q. The relation between m 1 and m 2 was made tight by the work of <ref type="bibr">Bhattacharya and Valiant (2015)</ref>; <ref type="bibr">Diakonikolas et al. (2021)</ref>, where they showed that it is sufficient to have</p><p>, and that in fact, this is optimal.</p><p>Our setting is also related to the problem of testing against a collection of distributions <ref type="bibr">Levi et al. (2013)</ref>; <ref type="bibr">Diakonikolas and Kane (2016)</ref>, where given sample access to a collection of distributions p 1 , p 2 , . . . , p s , the goal is to test whether they are identical or there doesn't exist a distribution p such that (1/s)</p><p>Our algorithm for the zero observation case (Theorem 1.1) can be viewed as a modified version the algorithm proposed in <ref type="bibr">Diakonikolas and Kane (2016)</ref> in the query model in the setting where the distribution mixture is not necessarily uniform.</p></div>
<div xmlns="http://www.tei-c.org/ns/1.0"><head n="1.4">Our technique</head><p>In this section, we give an overview of our algorithms and the lower bound constructions. We will focus mostly on the sublinear regime where m 2 = O(k) as it exhibits the main intuition on how sublinear interventional complexity can be achieved. We briefly elaborate on the techniques employed in the zero(few) and sufficient observational samples cases and refer the reader to the supplementary material for details.</p></div>
<div xmlns="http://www.tei-c.org/ns/1.0"><head n="1.4.1">Sublinear Observational Samples</head><p>Our algorithm. Consider the testing problem of whether X -&#8594; Y , which corresponds to Figure <ref type="figure">1</ref>(a), or X -&#8594; Y , which corresponds to Figure <ref type="figure">1(b)</ref> and Figure <ref type="figure">1(c)</ref>. By symmetry, whether Y -&#8594; X or Y -&#8594; X can be distinguished similarly. By definition, the testing problem reduces to distinguishing the following two cases.</p><p>(1)</p><p>A naive algorithm would be to intervene on all x &#8712; &#931; and use existing techniques on asymmetric closeness testing to test whether</p><p>Clearly, this would result in a sample complexity at least linear in k. We resolve this issue by using different methods in two regimes: (1) </p><p>or not using existing algorithms on closeness testing. This shows that if &#964; is large, we see an informative element less often and it takes less samples to test. While if &#964; is small, we see an informative element more often but it also takes more samples to test. However, without knowing &#964; , it is hard to "invest" the right amount of samples to test P x [Y ] for each x. In the general case, we resolve this by using Levin's investment strategy (Lemma 1) which shows that it is sufficient to design a collection of &#964; 's that form a geometric sequence and do as well as if &#964; is fixed and known . The upper bound we obtain recovers the O(max{k/&#949; 2 &#8730; m 1 , &#8730; k/&#949; 2 }) rate similar to asymmetric closeness testing. See details in Section 2.</p><p>Lower bound construction. Our construction for m 1 = O(k) uses the lower bound construction of <ref type="bibr">Bhattacharya and Valiant (2015)</ref> for asymmetric closeness testing as a primitive. They showed that there exist distributions p and q such that given access to O(m 1 ) samples from p any asymmetric closeness tester, requires &#8486; min</p><p>samples from q to distinguish p = q versus d TV (p, q). We construct q -, a slight modification of q, such that a uniform mixture of q and q -is p.</p><p>Using p, q, and q -we construct SCMs with marginal and conditional distributions as follows. For simplicity, we take X as a binary random variable while Y takes values from &#931;. The marginal probabilities P [X = 0] = P [X = 1] are 1/2. The conditional distributions are P [Y | X = 0] = q and P [Y | X = 1] = q -, thus obtaining P [X, Y ]. Note that the marginal distribution P [Y ] is p by construction. Note that there exists two SCMs, under Figure <ref type="figure">1(a)</ref> and Figure <ref type="figure">1(c</ref>) that could generate P [X, Y ]. We show that it is impossible to distinguish these two figures using the available</p><p>To do this, we extend the wishful thinking theorem <ref type="bibr">Valiant (2011)</ref> to distinguishing a collection of four distributions and show that their fourth-order moments are close. See Section 3 for details.</p></div>
<div xmlns="http://www.tei-c.org/ns/1.0"><head n="1.4.2">Zero(few) observational samples</head><p>Again consider testing X -&#8594; Y versus X -&#8594; Y . We use a similar strategy as that of the sublinear observational samples algorithm with a modification. Instead of finding symbols to intervene by sampling x from P [X] to test whether </p><p>While we do not have access to the observational marginal P [X], if X -&#8594; Y , we can instead simulate P [X] by sampling from P y [X] for an arbitrary y. Note that if X -&#8594; Y , then the interventional distributions P x1 [Y ] and P x2 [Y ] are identical.</p></div>
<div xmlns="http://www.tei-c.org/ns/1.0"><head n="1.4.3">Sufficient observational samples</head><p>We show that a single intervention is sufficient when we observe &#937;(k 2 /&#949; 2 ) observational samples. The crux of the algorithm in this regime is to identify the symbol to intervene upon. Assume that we want to test if X -&#8594; Y or X -&#8594; Y . By a similar reasoning as Section 1.4.1, we want to find a x &#8712; &#931; such that d</p><p>A folklore result states that for a discrete distribution of domain size k, the optimal sample complexity of estimating the distribution up to total variation distance &#949; is &#952; k &#949; 2 Canonne (2020a). Therefore, given &#937;(k 2 /&#949; 2 ) samples from the joint distribution P [X, Y ], empirical estimates of the marginals are O(&#949;)-close in total variation distance. However, not all conditional distributions are guaranteed to be &#949;-close since it is possible that if P [X] is small enough, the number of samples to empirically estimate P [Y | X] might not be sufficient. We claim that there exists a symbol x * &#8712; &#931; such that with a high probability there are sufficient samples to obtain &#949;-close estimates of P [Y | X = x * ] and P [Y ] and ensure</p><p>Therefore, a simple hypothesis test with O 1/&#949; 2 samples from</p><p>2 Sublinear Observational Samples: Algorithm</p><p>Here we discuss trade-offs between m 1 and m 2 when m 1 is sublinear in k. We show Theorem 1.2 in this section, where we present an almost optimal algorithm that solves CSI(k, &#949;)</p><p>, a modification of this algorithm combined with further analysis results in Theorem 1.1, which is included in supplementary material.</p><p>The following lemma is critical to our analysis.</p><p>Lemma 1. <ref type="bibr">Levin (1985)</ref>; <ref type="bibr">Goldreich (2014)</ref> Let D be a probability distribution, q : supp(D) &#8594; [0, 1], and &#949; &#8712; (0, 1]. Suppose that E[q(s)] &gt; &#949;, and let = log 2 (2/&#949;) .</p><p>Then, there exists j</p><p>Our algorithm uses the asymmetric closeness tester in Diakonikolas et al. ( <ref type="formula">2021</ref>) as a primitive.</p><p>Definition 3 (Closeness Testing). Given sample access to unknown discrete distributions p and q over domain [k], a closeness tester CT (m 1 , m 2 , &#949;, &#948;) draws m 1 samples from p, m 2 samples from q and outputs "YES" if p = q and "NO" if d TV (p, q) &gt; &#949; with probability atleast 1 -&#948; where &#949;, &#948; &gt; 0.</p><p>Lemma 2 (Theorem 1.5 in <ref type="bibr">Diakonikolas et al. (2021)</ref>). For discrete distributions p and q over [k], there exists a computationally efficient closeness tester CT (m 1 , m 2 , &#949;, &#948;) where m 1</p><p>Theorem 1.2. When m 1 = &#8486;(k 2/3 /&#949; 4/3 ), there exists an algorithm that takes m 1 observational samples and</p><p>)) interventional samples and solves CSI(k, &#949;). The number of distinct interventions the algorithm makes is O (1/&#949;) log 2 (1/&#949;) .</p><p>Proof. We analyze Algorithm 1 in two parts to prove the theorem. In the first part, we test whether X -&#8594; Y or X -&#8594; Y . If this test doesn't return X -&#8594; Y , we move to the second part and use essentially the same steps to test Algorithm 1:</p><p>Input :&#949; &gt; 0, sample access to</p><p>We apply Levin's investment strategy (Lemma 1), for the above choice of q. Let j := ( + 5 -j) 2 and s j := 2 j &#949; / ( j ) 2 . Lemma 1 guarantees the existence of j * &#8712; [ ] such that:</p><p>Therefore, in 20/s j * samples from P [X], by Chernoff bound, with probability at least 1 -e -10 , there exists a sample x i that satisfies d</p><p>If X -&#8594; Y , there exists j * &#8712; [ ] and a sample x i that satisfies d</p><p>Consider the following test: for every j &#8712; [ ], Algorithm 1 samples x i , 20/s j times from P [X] to distinguish</p><p>For n j 1 = &#8486; k 2/3 /2 -4j/3 and n j 2 = O(k/ n j 1 2 -2j ), each closeness test requires n j 1 samples from P [Y ] and n j 2 log(1/&#948; j ) samples from P x [Y ] to succeed with probability 1 -&#948; j . If any of the tests output "NO", then the algorithm returns X -&#8594; Y .</p><p>Test whether Y -&#8594; X or Y -&#8594; X. If all tests return "YES" then with a high probability, X -&#8594; Y and the algorithm proceeds to distinguish Y -&#8594; X and Y -&#8594; X using the same steps as before. Similar to the previous part, each individual test requires n j 1 samples from P [X] and n j 2 log(1/&#948; j ) samples from P y [X] to succeed with probability 1 -&#948; j . The algorithm returns Y -&#8594; X if one of the tests outputs "NO". If all tests return "YES", then the algorithm returns</p><p>, and hence all of the previous closeness tests are "YES" instances.</p></div>
<div xmlns="http://www.tei-c.org/ns/1.0"><head>Sample complexity. The number of samples we take</head><p>Similarly, the total number of interventional samples taken by the algorithm in the first stage is</p><p>Error analysis. Now we analyze the error probability. The total number of tests performed is at most O j=1 (20/s j ) . Hence by union bound the probability of failure of these tests is at most</p><p>When X -&#8594; Y , the probability of the algorithm failing to find a sample x satisfying d TV (P x [Y ], P [Y ]) &gt; 2 -j is at most 1/300. The analysis is the same for graph X &#8592;-Y . Hence the algorithm returns the correct graph with error probability at most 1/150.</p></div>
<div xmlns="http://www.tei-c.org/ns/1.0"><head>Number of interventions. The number of interventions</head><p>taken by the algorithm is upper bounded by the number of x i 's and y i 's drawn, which is:</p><p>3 Sublinear Observational Samples: Hardness</p><p>We now prove Theorem 1.3, which establishes an almost optimal lower bound on the tradeoff between m 1 and m 2 when m 1 is O (k), through a reduction to canonical results on property testing. We construct causal models under different structures (see Figure <ref type="figure">1</ref>) with the same observational distribution. We base our construction on the hard instance for asymmetric closeness testing in <ref type="bibr">Bhattacharya and Valiant (2015)</ref>.</p><p>For testing causal models, extra care is needed to prove hardness since the adversary has sample access to multiple interventional distributions. To handle this, we extend <ref type="bibr">Valiant (2008</ref><ref type="bibr">Valiant ( , 2011))</ref>'s wishful thinking theorem (see Theorem 4.6.9 in <ref type="bibr">Valiant (2008)</ref>) which distinguishes two distribution pairs to the case of distinguishing two distribution quadruplets. While the extension is immediate, we remark that the constants become much larger for the quadruplet case. For completeness, we state the extension below and defer the proofs to the supplementary material.</p></div>
<div xmlns="http://www.tei-c.org/ns/1.0"><head n="3.1">Wishful thinking for quadruplets</head><p>For the rest of this section we denote the sequence x 1 , x 2 , x 3 , x 4 by x 1:4 . Definition 4. For integers n 1:4 &gt; 0, the (n 1:4 )-based moment m (a 1:4 ) of the distribution quadruplet (p 1:4 ) is defined as m(a 1:4 ) :=</p><p>Proposition 1. Given integers n 1:4 &gt; 0, and two distribution quadruplets p + 1:4 and p - 1:4 , where p + i , p - i have frequencies at most 1 cni , for constant c &gt; 0. If m + and m - are the (n 1:4 )-based moments of p - 1:4 and p - 1:4 respectively that satisfy</p><p>for some constant c &gt; 0, then p - 1:4 cannot be distinguished from p - 1:4 with probability greater than 0.5 using a tester that takes Poi(n i ) samples from p + i , p - i for each i &#8712; [4].</p></div>
<div xmlns="http://www.tei-c.org/ns/1.0"><head n="3.2">Lower Bound</head><p>Theorem 1.3. For m 1 = &#8486;(k 2/3 /&#949; 4/3 ) and m 1 = O(k), any algorithm that takes m 1 observational samples must take m 2 = &#8486;(max(k/(</p><p>Proof Sketch. We present a proof sketch and defer the details to the supplementary material. Some definitions. Let &#960; be a permutation of [k] chosen uniformly at random. Let A = {&#960;(1), &#960;(2), . . . , &#960;(a)} and B = {&#960;(a + 1), &#960;(a + 2), . . . , &#960;(a + b)} be disjoint subsets of <ref type="bibr">[k]</ref> of size a = (3/4)m 1 and b = k/C, where C is a constant. For S &#8838; [k], let 1 S be the indicator function of set S. For permutation &#960;, define distributions p, q and q - over [k] as following:</p><p>Construction. Consider an SCM on variables X and Y over support {0, 1} and [k] resp. with 1. X &#8764; Bernoulli (0.5)</p><p>Note that &#961; TV (X, Y ) = &#949; because each pair of the three distributions p, q, q -has TV distance &#949; and the marginal distribution P [Y ] is p.</p><p>for a sufficiently small c. Let p + 1:4 := (q, q -, p, p) and p - 1:4 := (q, q -, q, q -). Suppose, for contradiction, there exists an algorithm A that solves CSI(k, &#949;) that uses n 1 observational samples and n 3 interventional samples. A single sample from each of the two conditionals simulates one sample of P [X, Y ]. Hence, we consider quadruplets of the form</p><p>). It is sufficient to show that p + 1:4 cannot be distinguished from p - 1:4 with probability greater than 0.5 by any tester that takes Poi(n i ) samples from p + i , p - i . We show this in the supplementary material by bounding the difference in moments</p><p>for the choices of n 1 , n 2 , n 3 , n 4 , which by Proposition 1 prove the theorem.</p></div>
<div xmlns="http://www.tei-c.org/ns/1.0"><head n="4">Discussion</head><p>We view our work as the first to provide finite sample complexity guarantees for the simplest causal structure identification problem, namely categorical two-variable systems. We parameterize the system using a quantitative notion of correlation between the two variables and quantify the tradeoff between observational and interventional data. Our sample complexity bounds on interventional samples, m 2 , as a function of observation samples, m 1 , exhibit interesting phase transitions and are tight when m 1 , as a function of the domain size k, is either sublinear O(k) or sufficient &#937; k 2 /&#949; 2 . In addition the number of interventional samples is always sublinear in k.</p><p>There are several directions for future work. The most immediate is improving upon the sample complexity of the superlinear regime and proving hardness results for the same. While the simplest property testing algorithms have only recently been studied for synthetic data <ref type="bibr">Gupta and Price (2022)</ref>, validating our proposed algorithms on real-world data is an important next step.</p></div>
<div xmlns="http://www.tei-c.org/ns/1.0"><head n="5">Acknowledgements</head><p>Jayadev Acharya and Saravanan Kandasamy are supported in part by the grant NSF-CCF-1846300 (CAREER), and a fellowship from Google. Sourbh Bhadane is supported by grants NSF-CCF-1846300 (CAREER) and NSF-CCF-1934985. Arnab Bhattacharyya is supported in part by an NRF Fellowship for AI, and part of this work was done while he was visiting the Simons Institute for the Theory of Computing.</p></div>
<div xmlns="http://www.tei-c.org/ns/1.0"><head>A Zero(few) Observational Samples</head><p>Here we look at the case when the number of observational samples m 1 is small. The algorithmic strategy is similar to the one discussed in Section 2 for sublinear observations. For Algorithm 1 because we had access to a large number of observational samples, we were able to estimate the marginals P [X] and P [Y ] and test if the marginals are far from interventions. Since m 1 is small here, we may not have access to the marginals. The idea here is to check if there exists interventions that are far apart, which is aided by the following lemma.</p><p>Lemma 3. Let X and Y be two random variables with joint distribution P [X, Y ] such that &#961; TV (X, Y ) &#949;. Then,</p><p>Proof.</p><p>(By Jensen's inequality)</p><p>Similarly, Let</p><p>We now use this result prove that CSI(k, &#949;) can be solved using O k 2/3 /&#949; 4/3 samples from interventions even with zero samples from observations. Theorem 1.1. There exists an algorithm that uses zero observational samples and m 2 = O max(k 2/3 /&#949; </p><p>Because &#961; TV (X, Y ) &#949;, by Lemma 3, we get,</p><p>We apply Levin's investment strategy (Lemma 1), for the above choice of q. Therefore, there exists j * &#8712; [ ] such that:</p><p>where s(j) := (2 j &#949;)/( + 5 -j) 2 , for all j &#8712; [ ]. For such j * , if we sample (x 1 , x 2 ) from (P [X], P [X]), 20/s(j * ) times, by Chernoff bound, with probability at least 1 -e -10 , there exists (x 1 , x 2 ) in the samples satisfying,</p><p>Part I. The first half of the algorithm considers testing whether X -&#8594; Y or X -&#8594; Y . ) is zero versus greater than 2 -j , with probability 1 -&#948; j , is</p><p>Hence, with probability 1 -e -10 , there exists j * &#8712; [ ] and a sample (x 1 , x 2 ) that satisfies</p><p>For </p></div>
<div xmlns="http://www.tei-c.org/ns/1.0"><head>Sample Analysis</head><p>We now analyze the total number of interventional samples. Let t (j) = ( + 5 -j). The total number of samples taken from interventions to perform the closeness tests in each part is:</p><p>Error Analysis. The total number of tests performed at Part I is at most O j=1 s j where s(j) = 20( + 5 -j) 2 /2 j &#949;. Hence, by union bound, the probability of failure of these tests is at most</p><p>Similarly, for Part II, the error probability of the algorithm is at most 1/300. Hence the algorithm returns the correct graph with error probability at most 1/150.</p></div>
<div xmlns="http://www.tei-c.org/ns/1.0"><head>B Super-quadratic Observations</head><p>Here we analyze the tradeof between m 1 and m 2 when m 1 is &#937; k 2 /&#949; 2 . We show that O(1/&#949; 2 ) interventional samples are sufficient and necessary to solve CSI(k, &#949;). </p><p>). First we prove a claim that assures the existence of x &#8712; &#931; such that we get sufficient samples on X = x and also &#964; x is small. Claim 1. With probability at least 2/3, there exists x &#8712; [k] such that:</p><p>(3) Also,</p><p>Combining Equations ( <ref type="formula">3</ref>) and (4),</p><p>x P (x) (&#964; x ) 2 1 {&#964;x&gt;&#949;/10} &gt; 99&#949; 2 /100.</p><p>Hence, there exists x &#8712; [k] that satisfies (i) &#964; x &#949; 10 and</p><p>Applying Chernoff bound,</p><p>.</p><p>(For any C &gt; 5.)</p><p>This proves the claim. We also have the following: Hardness. Next we show that even with infinite samples from observations, it requires at least &#8486; 1/&#949; 2 samples from interventions to solve CSI(k, &#949;).</p><p>Let P denote the set of all distributions P : [k] &#8594; [0, 1] of the form</p><p>where z i is either -1 or +1. Let q 0 be uniformly chosen at random from P. Let q 1 be the distribution obtained from q 0 by swapping the probabilities of odd and even coordinates (i.e., q 1 (2i) = q 0 (2i -1) and q 1 (2i -1) = q 0 (2i)).</p><p>Let the marginal distribution P [X] be P [X = 0] = P [X = 1] = 1/2 and the conditional distributions be P [Y | X = i] = q i . Here E [d TV (unif(k), q i )] &gt; &#949;. Also the marginal P [Y ] is unif[k] because (1/2) q 0 + (1/2) </p></div><note xmlns="http://www.tei-c.org/ns/1.0" place="foot" xml:id="foot_0"><p>Acharya, Bhadane, Bhattacharyya, Kandasamy, Sun   </p></note>
			<note xmlns="http://www.tei-c.org/ns/1.0" place="foot" n="1" xml:id="foot_1"><p>See(Valiant and Valiant,  </p></note>
			<note xmlns="http://www.tei-c.org/ns/1.0" place="foot" xml:id="foot_2"><p>2011, Theorem 3 and 4) for a constant probability version and the small error probability can be obtained using the boosting trick.</p></note>
		</body>
		</text>
</TEI>
