Year: 2020

Abstract:
We propose a new framework for reasoning about information in complex systems. Our foundation is based on a variational extension of Shannon’s information theory that takes into account the modeling power and computational constraints of the observer. The resulting predictive V-information encompasses mutual information and other notions of informativeness such as the coefficient of determination. Unlike Shannon’s mutual information and in violation of the data processing inequality, V-information can be created through computation. This is consistent with deep neural networks extracting hierarchies of progressively more informative features in representation learning. Additionally, we show that by incorporating computational constraints, V-information can be reliably estimated from data even in high dimensions with PAC-style guarantees. Empirically, we demonstrate predictive V-information is more effective than mutual information for structure learning and fair representation learning.

text: We propose a new framework for reasoning about information in complex systems. Our foundation is based on a variational extension of Shannon’s information theory that takes into account the modeling power and computational constraints of the observer. The resulting predictive V-information encompasses mutual information and other notions of informativeness such as the coefficient of determination. Unlike Shannon’s mutual information and in violation of the data processing inequality, V-information can be created through computation. This is consistent with deep neural networks extracting hierarchies of progressively more informative features in representation learning. Additionally, we show that by incorporating computational constraints, V-information can be reliably estimated from data even in high dimensions with PAC-style guarantees. Empirically, we demonstrate predictive V-information is more effective than mutual information for structure learning and fair representation learning.

heading: 1 INTRODUCTION
text: Extracting actionable information from noisy, possibly redundant, and high-dimensional data sources is a key computational and statistical challenge at the core of AI and machine learning. Information theory, which lies at the foundation of AI and machine learning, provides a conceptual framework to characterize information in a mathematically rigorous sense (Shannon & Weaver, 1948; Cover & Thomas, 1991). However, important computational aspects are not considered in information theory. To illustrate this, consider a dataset of encrypted messages intercepted from an opponent. According to information theory, these encrypted messages have high mutual information with the opponent’s plans. Indeed, with infinite computation, the messages can be decrypted and the plans revealed. Modern cryptography originated from this observation by Shannon that perfect secrecy is (essentially) impossible if the adversary is computationally unbounded (Shannon & Weaver, 1948). This motivated cryptographers to consider restricted classes of adversaries that have access to limited computational resources (Pass & Shelat, 2010). More generally, it is known that information theoretic quantities can be expressed in terms of betting games (Cover & Thomas, 1991). For example, the (conditional) entropy of a random variable X is directly related to how predictable X is in a certain betting game, where an agent is rewarded for correct guesses. Yet, the standard definition unrealistically assumes agents are computationally unbounded, i.e., they can employ arbitrarily complex prediction schemes.
Leveraging modern ideas from variational inference and learning (Ranganath et al., 2013; Kingma & Welling, 2013; LeCun et al., 2015), we propose an alternative formulation based on realistic computational constraints that is in many ways closer to our intuitive notion of information, which we term predictive V-information. Without constraints, predictive V-information specializes to classic mutual information. Under natural restrictions, V-information specializes to other well-known notions of predictiveness, such as the coefficient of determination (R2). A consequence of this new formulation is that computation can “create usable information” (e.g., by decrypting the intercepted messages), invalidating the famous data processing inequality. This generalizes the idea that clever
feature extraction enables prediction with extremely simple (e.g., linear) classifiers, a key notion in modern representation and deep learning (LeCun et al., 2015).
As an additional benefit, we show that predictive V-information can be estimated with statistical guarantees using the Probably Approximately Correct framework (Valiant, 1984). This is in sharp contrast with Shannon information, which is well known to be difficult to estimate for high dimensional or continuous random variables (Battiti, 1994). Theoretically we show that the statistical guarantees of estimating V information translate to statistical guarantees for a variant of the Chow-Liu algorithm for structure learning. In practice, when the observer employs deep neural networks as a prediction scheme, V-information outperforms methods that approximate Shannon information in various applications, including Chow-Liu tree contruction in high dimension and gene regulatory network inference.

heading: 2 DEFINITIONS AND NOTATIONS
text: To formally define the predictive V-information, we begin with a formal model of a computationally bounded agent trying to predict the outcome of a real-valued random variable Y ; the agent is either provided another real-valued random variable X as side information, or provided no side information ∅. We use X and Y to denote the samples spaces of X and Y respectively (while assuming they are separable), and use P(X ) to denote the set of all probability measures over the Borel algebra on X (P(Y) similarly defined for Y). Definition 1 (Predictive Family). 1 Let Ω = {f : X ∪ {∅} → P(Y)}. We say that V ⊆ Ω is a predictive family if it satisfies
∀f ∈ V,∀P ∈ range(f), ∃f ′ ∈ V, s.t. ∀x ∈ X , f ′[x] = P, f ′[∅] = P (1)
A predictive family is a set of predictive models the agent is allowed to use, e.g., due to computational or statistical constraints. We refer to the additional condition in Eq.(1) as optional ignorance. Intuitively, it means that the agent can, in the context of the prediction game we define next, ignore the side information if she chooses to. Definition 2 (Predictive conditional V-entropy). Let X,Y be two random variables taking values in X × Y , and V be a predictive family. Then the predictive conditional V-entropy is defined as
HV(Y |X) = inf f∈V Ex,y∼X,Y [− log f [x](y)]
HV(Y |∅) = inf f∈V Ey∼Y [− log f [∅](y)]
We additionally call HV(Y |∅) the V-entropy, and also denote it as HV(Y )
In our notation f is a function X ∪ {∅} → P(Y), so f [x] ∈ P(Y) is a probability measure on Y chosen based on the received side information x (we use f [·] instead of the more conventional f(·)); and f [x](y) ∈ R is the value of the density evaluated at y ∈ Y . Intuitively, V (conditional) entropy is the smallest expected negative log-likelihood that can be achieved predicting Y given observation (side information) X (or no side information ∅), using models from V . Eq.(1) means that whenever the agent can use P to predict Y’s outcomes, it has the option to ignore the input, and use P no matter whether X is observed or not.
Definition 2 generalizes several known definitions of uncertainty. For example, as shown in proposition 2, if the V is the largest possible predictive family that includes all possible models, i.e. V = Ω, then Definition 2 reduces to Shannon entropy: HΩ(Y |X) = H(Y |X) and HV(Y |∅) = HΩ(Y ) = H(Y ). By choosing more restrictive families V , we recover several other notions of uncertainty such as trace of covariance, as will be shown in Proposition 1.
Shannon mutual information is a measure of changes in entropy when conditioning on new variables:
I(X;Y ) = H(Y )−H(Y |X) = HΩ(Y )−HΩ(Y |X) (2) Here, we will use predictive V-entropy to define an analogous quantity, IV(X → Y ), to represent the change in predictability of an output variable Y when given side information X .
1Regularity Conditions: To minimize technical overhead we restrict out discussion only to distributions with probability density functions (PDF) or probability mass functions (PMF) with respect to the underlying measure. Also ∅ 6∈ X .
Definition 3 (Predictive V-information). Let X,Y be two random variables taking values in X × Y , and V be a predictive family. The predictive V-information from X to Y is defined as
IV(X → Y ) = HV(Y |∅)−HV(Y |X) (3)

heading: 2.1 IMPORTANT SPECIAL CASES
text: Several important notions of uncertainty and predictiveness are special cases of our definition. Note that when we are defining V-entropy of a random variable Y in sample space Y ∈ Rd (without side information), out of convenience we can assume X is empty X = ∅ (this does not violate our requirement that ∅ 6∈ X .) Proposition 1. For V-entropy and V-information, we have
1. Let Ω be as in Def. 1. Then HΩ(Y ) is the Shannon entropy, HΩ(Y | X) is the Shannon conditional entropy, and IΩ(Y → X) is the Shannon mutual information.
2. Let Y = Rd and V = {f : {∅} → Pµ | µ ∈ Rd}, where Pµ is the distribution with density y 7→ 1Z e −‖y−µ‖2 where Z = ∫ e−‖y−µ‖2dy, then the V-entropy of a random variable Y
equals its mean absolute deviation, up to an additive constant.
3. Let Y = Rd and V = {f : {∅} → N (µ,Σ) | µ ∈ Rd,Σ = 1/2Id×d}, then the V-entropy of a random variable Y equals the trace of its covariance tr (Cov(Y )), up to an additive constant.
4. Let V = {f : {∅} → Qt,θ, θ ∈ Θ}, where Qt,θ is a distribution in a minimal exponential family with sufficient statistics t : Y → Rd and set of natural parameters Θ. For a random variable Y with expected sufficient statistics µY = E[t(Y )], the V-entropy of Y is the maximum Shannon entropy over all random variables Ŷ with identical expected sufficient statistics, i.e. E[t(Ŷ )] = µY .
5. Let Y = Rd, X be any vector space, and V = {f : x 7→ N (φ(x),Σ), x ∈ X ;∅ 7→ N (µ,Σ)|µ ∈ Rd; Σ = 1/2Id×d, φ ∈ Φ}, where Φ is the set of linear functions {φ : X → Rd}, then V-information IV(X → Y ) equals the (unnormalized) maximum coefficient of determination R2 · tr (Cov(Y )) for linear regression.
The trace of covariance represents a natural notion of uncertainty – for example, a random variable with zero variance (when d = 1,tr (Cov(Y )) = Var(Y ))) is trivial to predict. Proposition 1.3 shows that the trace of covariance corresponds to a notion of surprise (in the Shannon sense) for an agent restricted to make predictions using certain Gaussian models. More broadly, a similar analogy can be drawn for other exponential families of distributions. In the same spirit, the coefficient of determination, also known as the fraction of variance explained, represents a natural notion of informativeness for computationally bounded agents. Also note that in the case of Proposition 1.4, the V-entropy is invariant if the expected sufficient statistics remain the same.

heading: 3 PROPERTIES OF V -INFORMATION

heading: 3.1 ELEMENTARY PROPERTIES
text: We first show several elementary properties of V-entropy and V-information. In particular, Vinformation preserves many properties of Shannon information that are desirable in a machine learning context. For example, mutual information (and V-information) should be non-negative as conditioning on additional side information X should not reduce an agent’s ability to predict Y . Proposition 2. Let Y and X be any random variables on Y and X , and V and U be any predictive families, then we have
1. Monotonicity: If V ⊆ U , then HV(Y ) ≥ HU (Y ), HV(Y | X) ≥ HU (Y | X).
2. Non-Negativity: IV(X → Y ) ≥ 0.
3. Independence: If X is independent of Y , IV(X → Y ) = IV(Y → X) = 0.
The optional ignorance requirement in Eq.(1) is a technical condition needed for these properties to hold. Intuitively, it guarantees that conditioning on side information does not restrict the class of densities the agent can use to predict Y . This property is satisfied by many existing machine learning models, often by setting some weights to zero so that an input is effectively ignored.

heading: 3.2 ON THE PRODUCTION OF INFORMATION THROUGH PREPROCESSING
text: The Data Processing Inequality guarantees that computing on data cannot increase its mutual information with other random variables. Formally, letting t : X → X be any function, t(X) cannot have higher mutual information with Y than X: I(t(X);Y ) ≤ I(X;Y ). But is this property desirable? In analyzing optimal communication, yes - it demonstrates a fundamental limit to the number of bits that can be transmitted through a communication channel. However, we argue that in machine learning settings this property is less appropriate.
Consider an RSA encryption scheme where the public key is known. Given plain text and its corresponding encrypted text X , if we have infinite computation, we can perfectly compute one from the other. Therefore, the plain text and the encrypted text should have identical Shannon mutual information with respect to any label Y we want to predict. However, to any human (or machine learning algorithm), it is certainly easier to predict the label from the plain text than the encrypted text. In other words, decryption increases a human’s ability to predict the label: processing increases the “usable information”. More formally, denoting t as the decryption algorithm and V as a class of natural language processing functions, we have that: IV(t(X)→ Y ) > IV(X → Y ) ≈ 0. As another example, consider the mutual information between an image’s pixels and its label. Due to data processing inequality, we cannot expect to use a function to map raw pixels to “features” that have higher mutual information with the label. However, the fundamental principle of representation learning is precisely the ability to learn predictive features — functions of the raw inputs that enable predictions with higher accuracy. Because of this key difference between V-information and Shannon information, machine learning practices such as representation learning can be justified in the information theoretic context.

heading: 3.3 ON THE ASYMMETRY OF PREDICTIVE V -INFORMATION
text: V-information also captures the intuition that sometimes, it is easy to predict Y from X but not vice versa. In fact, modern cryptography is founded on the assumption that certain functions h : X → Y are one-way, meaning that there exists an polynomial algorithm to compute h(x) but no polynomial algorithm to compute h−1(y). This means that if V contains all polynomial-time computable functions, then IV(X → h(X)) IV(h(X)→ X). This property is also reasonable in the machine learning context. For example, several important methods for causal discovery (Peters et al., 2017) rely on this asymmetry: if X causes Y , then usually it is easier to predict Y from X than vice versa; another commonly used assumption is that Y |X can be accurately modeled by a Gaussian distribution, while X|Y cannot (Pearl, 2000).

heading: 4 PAC GUARANTEES FOR V -INFORMATION ESTIMATION
text: For many practical applications of mutual information (e.g., structure learning), we do not know the joint distribution of X,Y , so cannot directly compute the mutual information. Instead we only have samples {(xi, yi)}Ni=1 ∼ X,Y and need to estimate mutual information from data. Shannon information is notoriously difficult to estimate for high dimensional random variables. Although non-parametric estimators of mutual information exist (Kraskov et al., 2004; Darbellay & Vajda, 1999; Gao et al., 2017), these estimators do not scale to high dimensions. Several variational estimators for Shannon information have been recently proposed (van den Oord et al., 2018; Nguyen et al., 2010; Belghazi et al., 2018), but have two shortcomings: due to their variational assumptions, their bias/variance tradeoffs are poorly understood and they are still not efficient enough for high dimensional problems. For example, the CPC estimator suffers from large bias, since its estimates saturate at logN where N is the batch size (van den Oord et al., 2018; Poole et al., 2019); the NWJ estimator suffers from large variance that grows at least exponentially in the ground-truth mutual information (Song & Ermon, 2019). Please see Appendix B for more details and proofs.
On the other hand, V-information is explicit about the assumptions (as a feature instead of a bug). V-information is also easy to estimate with guarantees if we can bound the complexity of V (such as its Radamacher or covering number complexity) As we will show, bounds on the complexity of V directly translate to PAC (Valiant, 1984) bounds for V-information estimation. In practice, we can efficiently optimize over V , e.g., via gradient descent. In this paper we will present the Rademacher complexity version; other complexity measures (such as covering number) can be derived similarly. Definition 4 (Empirical V-information). LetX,Y be two random variables taking values inX ,Y and D = {(xi, yi)}Ni=1 ∼ X,Y denotes the set of samples drawn from the joint distribution over X and Y . V is a predictive family. The empirical V-information (under D) is the following V-information under the empirical distribution defined via D:
ÎV(X → Y ;D) = inf f∈V
1 |D| ∑ yi∈D log 1 f [∅](yi) − inf f∈V 1 |D| ∑ xi,yi∈D log
1
f [xi](yi) (4)
Then we have the following PAC bound over the empirical V-information: Theorem 1. Assume ∀f ∈ V, x ∈ X , y ∈ Y, log f [x](y) ∈ [−B,B]. Then for any δ ∈ (0, 0.5), with probability at least 1− 2δ, we have:∣∣∣IV(X → Y )− ÎV(X → Y ;D)∣∣∣ ≤ 4R|D|(GV) + 2B √ 2 log 1δ |D| (5)
where we define the function family GV = {g|g(x, y) = log f [x](y), f ∈ V}, and RN (G) denotes the Rademacher complexity of G with sample number N .
Typically, the Rademacher complexity term satisfies R|D|(GV) = O(|D|− 1 2 ) (Bartlett & Mendelson, 2001; Gao & Zhou, 2016). It’s worth noticing that a complex function family V (i.e., with large Rademacher complexity) could lead to overfitting. On the other hand, an overly-simple V may not be expressive enough to capture the relationship between X and Y . As an example of the theorem, we provide a concrete estimation bound when V is chosen to be linear functions mapping X to the mean of a Gaussian distribution. This was shown in Proposition 1 to lead to the coefficient of determination.
Corollary 1.1. Assume X = {x ∈ Rdx , ‖x‖2 ≤ kx} and Y = {y ∈ Rdy , ‖y‖2 ≤ ky}. If
V = {f : f [x] = N (Wx+ b, I), f [∅] = N (c, I),W ∈ Rdy×dx , b, c ∈ Rdy , ‖(W, b)‖2 ≤ 1}
Denote M = (kx + ky)2 + log 2π, then ∀δ ∈ (0, 0.5), with probability at least 1− 2δ:∣∣∣IV(X → Y )− ÎV(X → Y ;D)∣∣∣ ≤ M√ 4|D| ( 1 + 4 √ 2 log 1 δ )
Similar results can be obtained using other classes of machine learning models with known (Rademacher) complexity.

heading: 5 STRUCTURE LEARNING WITH V -INFORMATION
text: Among many possible applications of V-information, we show how to use it to perform structure learning with provable guarantees. The goal of structure learning is to learn a directed graphical model (Bayesian network) or undirected graphical model (Markov network) that best captures the (conditional) independence structure of an underlying data generating process. Structure learning is difficult in general, but if we restrict ourselves to certain set of graphsG, there are efficient algorithms. In particular, the Chow-Liu algorithm (Chow & Liu, 1968) can efficiently learn tree graphs (i.e. G is the set of trees). Chow & Liu (1968) show that the problem can be reduced to:
g∗ = arg max g∈Gtree ∑ (Xi,Xj)∈edge(g) I(Xi, Xj) (6)
where I(Xi, Xj) is the Shannon mutual information between variables Xi and Xj . In other words, it suffices to construct the maximal weighted spanning tree where the weight between two vertices is
their Shannon mutual information. Chow & Wagner (1973) show that the Chow-Liu algorithm is consistent, i.e, it recovers the true solution as the dataset size goes to infinity. However, the finite sample behavior of the Chow-Liu algorithm for high dimensional problems is much less studied, due to the difficulty of estimating mutual information. In fact, we show in our experiments that the empirical performance is often poor, even with state-of-the-art estimators. Additionally, methods based on mutual information cannot take advantage of intrinsically asymmetric relationships, which are common for example in gene regulatory networks (Meyer et al., 2007).
To address these issues, we propose a new structure learning algorithm based on V-information instead of Shannon information. The idea is that we can associate to each directed edge in G (i.e., each pair of variables) a suitable predictive family Vi,j (cf. Def 1). The main challenge is that we cannot simply replace mutual information with V-information in Eq. 6 because V-information is asymmetric – we now have to optimize over directed trees:
g∗ = arg max g∈Gd−tree m∑ i=2 IVt(g)(i),i(Xt(g)(i) → Xi) (7)
where Gd−tree is the set of directed trees, and t(g) : N→ N is the function mapping each non-root node of directed tree g to its parent, and Vi,j is the predictive family for random variables Xi and Xj . After estimating V-information on each edge, we use the Chu-Liu algorithm (Chu & Liu, 1965) to construct the maximal directed spanning tree. This allows us to solve (7) exactly, even though there is a combinatorially large number of trees to consider. Pseudocode is summarized in Algorithm 1 in Appendix. Denote C(g) = ∑m i=2 IVt(g)(i),i(Xt(g)(i) → Xi), we show in the following theorem that unlike the original Chow-Liu algorithm, our algorithm has guarantees in the finite samples regime, even in continuous settings: Theorem 2. Let {Xi}mi=1 be the set of m random variables, Di,j (resp. Dj) be the set of samples drawn from P (Xi, Xj) (resp. P (Xj)). Denote the optimal directed tree with maximum expected edge weights sum C(g) as g∗ and the optimal directed tree constructed on the dataset D as ĝ. Then with the assumption in theorem 1, for any δ ∈ (0, 12m(m−1) ), with probability at least 1− 2m(m− 1)δ, we have:
C(ĝ) ≥ C(g∗)− 2(m− 1) max i,j
{ 2RDi,j (GVi,j ) + 2RDj (GVj ) +B √ 2 log 1
δ (|Dj |− 1 2 + |Di,j |− 1 2 ) } (8)
Theorem 2 shows that the total edge weights of the maximal directed spanning tree constructed by algorithm 1 would be close to the optimal total edge weights if the Rademacher term is small. Although larger C(g) does not necessarily lead to better Chow-Liu trees, empirically we find that the optimal tree in the sense of equation (7) is consistent with the optimal tree in equation (6) under commonly used V .

heading: 6 EXPERIMENTAL RESULTS

heading: 6.1 STRUCTURE LEARNING WITH CONTINUOUS HIGH-DIMENSIONAL DATA
text: We generate synthetic data using various ground-truth tree structures g∗ with between 7 and 20 variables, where each variable is 10-dimensional. We use Gaussians, Exponentials, and Uniforms as ground truth edge-conditionals. We use V-information(Gaussian) and V-information(Logistic) to denote Algorithm 1 with two different V families. Please refer to Appendix D.1 for more details. We compare with the original Chow-Liu algorithm equipped with state-of-the-art mutual information estimators: CPC (van den Oord et al., 2018), NWJ (Nguyen et al., 2010) and MINE (Belghazi et al., 2018), with the same neural network architecture as the V-families for fair comparison. All the experiments are repeated for 10 times. As a performance metric, we use the wrong-edges-ratio (the ratio of edges that are different from ground truth) as a function of the amount of training data.
We show two illustrative experiments in figure 1a; please refer to Appendix D.1 for all simulations. We can see that although the two V-families used are misspecified with respect to the true underlying (conditional) distributions, the estimated Chow-Liu trees are much more accurate across all data regimes, with CPC (blue) being the best alternative. Surprisingly, V-information(Gaussian) works consistently well in all cases and only requires about 100 samples to recover the ground-truth Chow-Liu tree in simulation-A.

heading: 6.2 GENE REGULATORY NETWORK INFERENCE
text: Mutual information between pairs of gene expressions is often used to construct gene regulatory networks. We evaluate V-information on the in-silico dataset from the DREAM5 challenge (Marbach et al., 2012) and use the setup of Gao et al. (2017), where 20 genes with 660 datapoints are utilized to evaluate all methods. We compare with state-of-the-art non-parametric Shannon mutual information estimators in this low dimensional setting: KDE, the traditional kernel density estimator; the KSG estimator (Kraskov et al., 2004); the Mixed KSG estimator (Gao et al., 2017) and Partitioning, an adaptive partitioning estimator (Darbellay & Vajda, 1999) implemented by Szabó (2014). For fair comparison with these low dimensional estimators, we select V = {f : f [x] = N (g(x), 12 ), x ∈ X ; f [∅] = N (µ, 12 )|µ ∈ range(g)}, where g is a 3-rd order polynomial. The task is to predict whether a directed edge between genes exists in the ground-truth gene network. We use the estimated mutual information and V-information for gene pairs as the test statistic to obtain the AUC for various methods. As shown in Figure 1b, our method outperforms all other methods in network inference under different fractions of data used for estimation. The natural information measure in this task is asymmetry since the goal is to find the pairs of genes (Ai, Bi)s in which Ai regulates Bi, thus V-information is more suitable for such case than mutual information.

heading: 6.3 RECOVERING THE ORDER OF VIDEO FRAMES
text: Let X1, · · · , X20 be random variables each representing a frame in videos from the Moving-MNIST dataset, which contains 10,000 sequences each of length 20 showing two digits moving with stochastic dynamics. Can Algorithm 1 be used to recover the natural (causal) order of the frames? Intuitively, predictability should be inversely related with frame distance, thus enabling structure learning. Using a conditional PixelCNN++ (Salimans et al., 2017) as predictive family V , we shown in Figure 1c that predictive V-information does indeed decrease with frame distance, despite some fluctuations when the frame distances are large. Using Algorithm 1 to construct a Chow-Liu tree, we find that the tree perfectly recovers the relative order of the frames.
We also generate a Deterministic-Moving-MNIST dataset, where digits move according to deterministic dynamics. From the perspective of Shannon mutual information, every pair of frames has the same mutual information. Hence, standard Chow-Liu tree learning algorithm would fail to discover the natural ordering of the frames (causal structure). In contrast, once we constrain the observer to PixelCNN++ models, algorithm 1 with predictive V-information can still recover the order of different frames when the frame distances are relatively small (less than 9). Compared to the stochastic dynamics case, V-information is more irregular with increasing frame distance, since the PixelCNN++ tends to overfit.

heading: 6.4 INFORMATION THEORETIC APPROACHES TO FAIRNESS
text: The goal of fair representation learning is to map inputs X ∈ X to a feature space Z ∈ Z such that the mutual information between Z and some sensitive attribute U ∈ U (such as race or gender) is minimized. The motivation is that using Z (instead of X) as input we can no longer use the sensitive attributes U to make decisions, thus ensuring some notion of fairness. Existing methods obtain fair representations by optimizing against an “adversarial” discriminator so that the discriminator cannot predict U from Z (Edwards & Storkey, 2015; Louizos et al., 2015; Madras et al., 2018; Song et al., 2018). Under some assumptions on U and V , we show in Appendix D.2 that these works actually use V-information minimization as part of their objective, where V depends on the functional form of the discriminator.
However, it is clear from the V-information perspective that features trained with VA-information minimization might not generalize to VB-information and vice versa. To illustrate this, we use a function family Vj as the attacker to extract information from features trained with IVi(Z → U) minimization, where all the Vs are neural nets. On three datasets commonly used in the fairness literature (Adult, German, Heritage), previous methods work well at preventing information “leak” against the class of adversary they’ve been trained on, but fail when we consider different ones. As shown in Figure 3b in Appendix, the diagonal elements in the matrix are usually the smallest in rows, indicating that the attacker function family Vi extracts more information on featured trained with Vj(j 6=i)-information minimization. This challenges the generalizability of fair representations in previous works. Please refer to Appendix D.2 for details.

heading: 7 RELATED WORK
text: Alternative definitions of Information Several alternative definitions of mutual information are available in the literature. Renyi entropy and Renyi mutual information (Lenzi et al., 2000) extend Shannon information by replacing KL divergence with f -divergences. However, they have the same difficulty when applied to high dimensional problems as Shannon information.
The line of work most related to ours is the H entropy and H mutual information (DeGroot et al., 1962; Grünwald et al., 2004), which associate a definition of entropy to every prediction loss. However, there are two key differences. First, literatures in H entropy only consider a few special types of prediction functions that serve unique theoretical purposes; for example, (Duchi et al., 2018) considers the set of all functions on a feature space to prove surrogate risk consistency, and (Grünwald et al., 2004) only considers the H entropy to prove the duality between maximum entropy and worst-case loss minimization. In contrast, our definition takes a completely different perspective — emphasizing bounded computation and intuitive properties of “usable” information. Furthermore H entropy still suffers from difficulty of estimation in high dimension because the definitions do not restrict to functions with small complexity (e.g. Rademacher complexity).
Mutual information estimation The estimation of mutual information in the machine learning field is often on the continuous underlying distribution. For non-parametric mutual information estimators, many methods have exploited the 3H principle to calculate the mutual information, such as the Kernel density estimator (Paninski & Yajima, 2008), k-Nearest-Neighbor estimator and the KSG estimator (Kraskov et al., 2004). However, these non-parametric estimators usually aren’t scalable to high dimension. Recently, several works utilize the variational lower bounds of MI to design MI estimator based on deep neural network in order to estimate MI of high dimension continuous random variables (Nguyen et al., 2010; van den Oord et al., 2018; Belghazi et al., 2018).

heading: 8 CONCLUSION
text: We defined and investigated V-information, a variational extension to classic mutual information that incorporates computational constraints. Unlike Shannon mutual information, V-information attempts to capture usable information, and has very different properties, such as invalidating the data processing inequality. In addition, V-information can be provably estimated, and can thus be more effective for structure learning and fair representation learning.

heading: ACKNOWLEDGEMENTS
text: This research was supported by AFOSR (FA9550-19-1-0024), NSF (#1651565, #1522054, #1733686), ONR, and FLI.

heading: A PROOFS
text: A.1 PROOF OF PROPOSITION 1
Proposition 1. For V-entropy and V-information, we have
1. Let Ω be as in Def. 1. Then HΩ(Y ) is the Shannon entropy, HΩ(Y | X) is the Shannon conditional entropy, and IΩ(Y → X) is the Shannon mutual information.
2. Let Y = Rd and V = {f : {∅} → Pµ | µ ∈ Rd}, where Pµ is the distribution with density y 7→ 1Z e −‖y−µ‖2 where Z = ∫ e−‖y−µ‖2dy, then the V-entropy of a random variable Y
equals its mean absolute deviation, up to an additive constant.
3. Let Y = Rd and V = {f : {∅} → N (µ,Σ) | µ ∈ Rd,Σ = 1/2Id×d}, then the V-entropy of a random variable Y equals the trace of its covariance tr (Cov(Y )), up to an additive constant.
4. Let V = {f : {∅} → Qt,θ, θ ∈ Θ}, where Qt,θ is a distribution in a minimal exponential family with sufficient statistics t : Y → Rd and set of natural parameters Θ. For a random variable Y with expected sufficient statistics µY = E[t(Y )], the V-entropy of Y is the maximum Shannon entropy over all random variables Ŷ with identical expected sufficient statistics, i.e. E[t(Ŷ )] = µY .
5. Let Y = Rd, X be any vector space, and V = {f : x 7→ N (φ(x),Σ), x ∈ X ;∅ 7→ N (µ,Σ)|µ ∈ Rd; Σ = 1/2Id×d, φ ∈ Φ}, where Φ is the set of linear functions {φ : X → Rd}, then V-information IV(X → Y ) equals the (unnormalized) maximum coefficient of determination R2 · tr (Cov(Y )) for linear regression.
Proof. (1)
Let PY |x denote the density function of random variable Y conditioned on X = x (we denote this random variable as Y | x).
HΩ(Y |X) = inf f∈Ω
Ex,y∼X,Y [ log
1
f [x](y) ] = inf f∈Ω Ex∼XEy∼Y |x [ log
PY |x(y)
f [x](y)PY |x(y) ] = inf f∈Ω Ex∼X [ KL(PY |x‖f [x]) +H(Y |x)
] = Ex∼X [H(Y |x)] = H(Y |X) (9)
where infimum is achieved for f where f [x] = PY |x and H is the Shannon (conditional) entropy. The same proof technique can be used to show that HΩ(Y ) = H(Y ), with the infimum achieved by f where f [∅] = PY . Hence we have
IΩ(Y → X) = HΩ(Y )−HΩ(Y |X) = H(Y )−H(Y |X) = I(Y ;X) (10)
(2)
HV(Y ) = inf f∈V Ey∼Y [− log f [∅](y)] = inf µ∈Rd
Ey∼Y [ − log 1
Z e−‖y−µ‖2 ] = inf µ∈Rd Ey∼Y [‖ y − µ ‖2] + logZ
= MAD(Y ) + logZ (11)
where MAD denotes mean absolute deviation Ey∼Y [‖ y − E[Y ] ‖2].
(3) HV(Y ) = inf
f∈V Ey∼Y [− log f [∅](y)]
= inf µ∈Rd
Ey∼Y [ − log 1
(2π) d 2 |Σ| 12
e− 1 2 (y−µ) TΣ−1(y−µ)
]
= inf µ∈Rd
Ey∼Y [(y − µ)T (y − µ)] + d
2 log π
= inf µ∈Rd
Ey∼Y [tr ( (y − µ)(y − µ)T ) ] + d
2 log π (Cyclic property of trace)
= tr (Cov(Y )) + d
2 log π (Linearity of trace)
(4) The density function of an exponential family distribution with sufficient statistics t is y 7→ exp (θ · t(y)−A(θ)) where A(θ) is the partition function.
HV(Y ) = inf f∈V Ey∼Y [− log f [∅](y)] = inf θ∈Θ Ey∼Y [− log exp (θ · Ey∼Y [t(y)]−A(θ))]
=− sup θ∈Θ (θ · Ey∼Y [t(y)]−A(θ))]
=−A∗(Ey∼Y [t(y)]) (12) where A∗ is the Fenchel dual of the log-partition function A(θ). Under mild conditions (Wainwright et al., 2008) −A∗(µ) = H(Pµ) where Pµ is the maximum entropy distribution out of all distributions satisfying Ey∼Pµ [t(y)] = µ (Jaynes, 1982), and H(·) is the Shannon entropy.
(5) Assume random variable Y ∈ Rd, V = {f : x 7→ N (φ(x),Σ), x ∈ X ;∅ 7→ N (µ,Σ)|µ ∈ Rd; Σ = 12Id×d;φ ∈ Φ}. Then the V-information from X to Y is IV(X → Y ) = HV(Y )−HV(Y |X)
= inf µ∈Rd
Ey∼Y [ − log 1
(2π) d 2 |Σ| 12
e−‖y−µ‖ 2 2 ] − inf φ∈Φ Ex,y∼X,Y [ − log 1
(2π) dy 2 |Σ| 12
e−‖y−φ(x)‖ 2 2 ] = inf µ∈Rd Ex,y∼X,Y [ ‖ y − µ ‖22 ] − inf φ∈Φ Ex,y∼X,Y [ ‖ y − φ(x) ‖22
] = tr (Cov(Y )) 1− infφ∈ΦEx,y∼X,Y [ ‖ y − φ(x) ‖22 ] tr (Cov(Y ))
 = tr (Cov(Y ))R2 (13)
A.2 PROOF OF PROPOSITION 2
Proposition 2. Let Y and X be any random variables on Y and X , and V and U be any predictive families, then we have
1. Monotonicity: If V ⊆ U , then HV(Y ) ≥ HU (Y ), HV(Y | X) ≥ HU (Y | X).
2. Non-Negativity: IV(X → Y ) ≥ 0.
3. Independence: If X is independent of Y , IV(X → Y ) = IV(Y → X) = 0.
Proof. (1)
HV(Y ) = inf f∈V
Ey∼Y [ log
1
f [∅](y) ] ≥ inf f∈U Ey∼Y [ log
1
f [∅](y)
] = HU (Y ) (14)
HV(Y |X) = inf f∈V
Ex,y∼X,Y [ log
1
f [x](y) ] ≥ inf f∈U Ex,y∼X,Y [ log
1
f [x](y)
] = HU (Y |X) (15)
The inequalities (14) and (15) are because we are taking the infimum over a larger set.
(2)
Denote V∅ ⊂ V as the subset of f that satisfy f [x] = f [∅], ∀x ∈ X .
HV(Y ) = inf f∈V
Ex,y∼X,Y [− log f [∅](y)]
= inf f∈V∅
Ex,y∼X,Y [− log f [∅](y)] (By Optional Ignorance)
= inf f∈V∅
Ex,y∼X,Y [− log f [x](y)]
≥ inf f∈V Ex,y∼X,Y [− log f [x](y)] = HV(Y | X)
Therefore IV(Y → X) = HV(Y )−HV(Y |X) ≥ 0
(3)
Denote V∅ ⊂ V as the subset of f that satisfy f [x] = f [∅], ∀x ∈ X .
HV(Y | X) = inf f∈V Ex,y∼X,Y [− log f [x](y)]
= inf f∈V
Ex∼XEy∼Y [− log f [x](y)] (Independence) ≥ Ex∼X [
inf f∈V
Ey∼Y [− log f [x](y)] ]
(Jensen)
= Ex∼X [
inf f∈V∅
Ey∼Y [− log f [x](y)] ]
(Optional Ignorance)
= inf f∈V∅
Ey∼Y [− log f [∅](y)] (No dependence on x)
≥ inf f∈V Ey∼Y [− log f [∅](y)] = HV(Y )
Therefore IV(Y → X) = HV(Y ) − HV(Y |X) ≤ 0. Combined with the Proposition 2.2 that IV(X → Y ) must be non-negative, IV(X → Y ) must be 0.
A.3 PROOF OF THEOREM 1
Theorem 1. Assume ∀f ∈ V, x ∈ X , y ∈ Y, log f [x](y) ∈ [−B,B]. Then for any δ ∈ (0, 0.5), with probability at least 1− 2δ, we have:∣∣∣IV(X → Y )− ÎV(X → Y ;D)∣∣∣ ≤ 4R|D|(GV) + 2B √ 2 log 1δ |D| (5)
where we define the function family GV = {g|g(x, y) = log f [x](y), f ∈ V}, and RN (G) denotes the Rademacher complexity of G with sample number N .
Before proving theorem 1, we introduce two lemmas. Proofs for these Lemmas follow the same strategy as theorem 8 in Bartlett & Mendelson (2001): Lemma 3. LetX,Y be two random variables taking values in X ,Y andD denotes the set of samples drawn from the joint distribution overX×Y . Assume ∀f ∈ V, x ∈ X , y ∈ Y, log f [x](y) ∈ [−B,B]. Take f̂ = arg min
f∈V
1 |D| ∑ xi,yi∈D − log f [xi](yi), then ∀δ ∈ (0, 1), with probability at least 1 − δ, we
have: ∣∣∣∣∣∣HV(Y |X)− 1|D| ∑ xi,yi∈D − log f̂ [xi](yi) ∣∣∣∣∣∣ ≤ 2R|D|(GV) + 2B √ 2 log 1δ |D|
(16)
Proof. We apply McDiarmid’s inequality to the function Φ defined for any sample D by
Φ(D) = sup f∈V ∣∣∣∣∣∣Ex,y [− log f [x](y)]− 1|D| ∑ xi,yi∈D − log f [xi](yi) ∣∣∣∣∣∣ (17) Let D and D′ be two samples differing by exactly one point, then since the difference of suprema does not exceed the supremum of the difference and ∀f ∈ V, x ∈ X , y ∈ Y, log f [x](y) ∈ [−B,B], we have:
Φ(D)− Φ(D′)
≤ sup f∈V ∣∣∣∣∣∣ 1|D| ∑ xi,yi∈D log f [xi](yi)− Ex,y [log f [x](y)] ∣∣∣∣∣∣− ∣∣∣∣∣∣ 1|D′| ∑ xi,yi∈D′ log f [xi](yi)− Ex,y [log f [x](y)] ∣∣∣∣∣∣ 
≤ sup f∈V ∣∣∣∣∣∣ 1|D| ∑ xi,yi∈D − log f [xi](yi)| − 1 |D′| ∑ xi,yi∈D′ − log f [xi](yi) ∣∣∣∣∣∣ ≤ 2B |D|
then by McDiarmid’s inequality, for any δ ∈ (0, 1), with probability at least 1 − δ, the following holds:
Φ(D) ≤ ED[Φ(D)] +B √ 2 log 1δ |D|
(18)
Then we bound the ED[Φ(D)] term:
ED[Φ(D)] = ED sup f∈V ∣∣∣∣∣∣Ex,y [− log f [x](y)]− 1|D| ∑ xi,yi∈D − log f [xi](yi) ∣∣∣∣∣∣  (19)
= ED sup f∈V ∣∣∣∣∣∣ED′  1 |D′| ∑ x′i,y ′ i∈D′ log f [x′i](y ′ i) − 1 |D| ∑ xi,yi∈D log f [xi](yi) ∣∣∣∣∣∣  (20)
≤ ED sup f∈V ED′ ∣∣∣∣∣∣ 1|D′| ∑
x′i,y ′ i∈D′
log f [x′i](y ′ i)| −
1 |D| ∑
xi,yi∈D log f [xi](yi) ∣∣∣∣∣∣  (21)
≤ ED,D′ sup f∈V ∣∣∣∣∣∣ 1|D′| ∑
x′i,y ′ i∈D′
log f [x′i](y ′ i)| −
1 |D| ∑
xi,yi∈D log f [xi](yi) ∣∣∣∣∣∣  (22)
= ED,D′ sup f∈V ∣∣∣∣∣∣ 1|D| |D|∑ i=1 (log f [x′i](y ′ i)− log f [xi](yi)) ∣∣∣∣∣∣  (23)
≤ ED,D′,σ sup f∈V ∣∣∣∣∣∣ 1|D| |D|∑ i=1 σi(log f [x ′ i](y ′ i)− log f [xi](yi)) ∣∣∣∣∣∣  (24)
≤ ED,σ sup f∈V ∣∣∣∣∣∣ 1|D| |D|∑ i=1 σi log f [xi](yi) ∣∣∣∣∣∣ + ED′,σ sup f∈V ∣∣∣∣∣∣ 1|D| |D|∑ i=1 σi log f [x ′ i](y ′ i) ∣∣∣∣∣∣  (25)
= 2ED,σ sup f∈V ∣∣∣∣∣∣ 1|D| |D|∑ i=1 σi log f [xi](yi) ∣∣∣∣∣∣  (26)
= 2ED,σ sup g∈G ∣∣∣∣∣∣ 1|D| |D|∑ i=1 σig(xi, yi) ∣∣∣∣∣∣  = 2R|D|(GV) (27)
where σis are Rademacher variables that is uniform in {−1,+1}. Inequality (22) follows from the convexity of sup, inequality (24) follows from the symmetrization argument for `1 norm for Radermacher random variables (Ledoux & Talagrand (2013), Section 6.1), inequality (21) follows from the convexity of |x− c|. (27) follows from the definition of G and Rademacher complexity. Finally, combining inequality (18) and (27) yields for all f ∈ V , with probability at least 1− δ∣∣∣∣∣∣Ex,y[− log f [x](y)]− 1|D| ∑ xi,yi∈D − log f [xi](yi) ∣∣∣∣∣∣ ≤ 2R|D|(GV) +B √ 2 log 1δ |D| (28)
In particular, the inequality holds for f̂ = arg min f∈V 1 |D| ∑ xi,yi∈D − log f [xi](yi) and f̃ = arg min f∈V Ex,y∼X,Y [− log f [x](y)]. Then we have:
Ex,y∼X,Y [ − log f̃ [x](y) ] − 1 |D| ∑ xi,yi∈D − log f̃ [xi](yi) ≤ HV(Y |X)− 1 |D| ∑ xi,yi∈D − log f̂ [xi](yi)
≤ Ex,y∼X,Y [ − log f̂ [x](y) ] − 1 |D| ∑ xi,yi∈D − log f̂ [xi](yi)
Hence the bound (16) holds.
Similar bounds can be derived for HV(Y ) when we choose the domain of x to be X = {∅}: Lemma 4. Let Y be random variable taking values in Y and D denotes the set of samples drawn from the underlying distribution P (Y ). Assume ∀f ∈ V, y ∈ Y, log f [∅](y) ∈ [−B,B]. Take f̂ = arg min
f∈V
1 |D| ∑ xi,yi∈D − log f [∅](yi), then for any δ ∈ (0, 1), with probability at least 1− δ, we
have: ∣∣∣∣∣∣HV(Y )− 1|D| ∑ yi∈D − log f̂ [∅](yi) ∣∣∣∣∣∣ ≤ 2R|D|(GV∅) +B √ 2 log 1δ |D|
(29)
≤ 2R|D|(GV) +B √ 2 log 1δ |D|
(30)
where GV∅ = {g|g(y) = log f [∅](y), f ∈ V}.
Proof. The first inequality (29) can be derived similarly as Lemma 3. Since V is a predictive family, hence there exits a function h : V → V , such that h(f) = f ′ and ∀x ∈ X , f ′[x] = f [∅].
R|D|(GV∅) = ED,σ sup f∈V ∣∣∣∣∣∣ 1|D| |D|∑ i=1 σi log f [∅](yi) ∣∣∣∣∣∣ 
= ED,σ sup f∈V ∣∣∣∣∣∣ 1|D| |D|∑ i=1 σi log h(f)[xi](yi) ∣∣∣∣∣∣ 
≤ ED,σ sup f∈V ∣∣∣∣∣∣ 1|D| |D|∑ i=1 σi log f [xi](yi) ∣∣∣∣∣∣  (31)
= R|D|(GV)
The inequality (31) holds because of h(V) ⊆ V .
Now we prove theorem 1:
Theorem 1. Assume ∀f ∈ V, x ∈ X , y ∈ Y, log f [x](y) ∈ [−B,B], for any δ ∈ (0, 0.5), with probability at least 1− 2δ, we have:∣∣∣IV(X → Y )− ÎV(X → Y ;D)∣∣∣ ≤ 4R|D|(GV) + 2B √ 2 log 1δ |D|
Proof. Define f̂ = arg min f∈V ∑ xi,yi∈D − log f [xi](yi) and f̂∅ = arg min f∈V ∑ yi∈D − log f [∅](yi). Using the triangular inequality we have:∣∣∣IV(X → Y )− ÎV(X → Y ;D)∣∣∣ = ∣∣∣∣∣∣(HV(Y )−HV(Y |X))−  1 |D| ∑ yi∈D − log f̂∅[∅](yi)− 1 |D| ∑ xi,yi∈D − log f̂ [xi](yi)
∣∣∣∣∣∣ ≤ ∣∣∣∣∣∣ HV(Y )− 1|D| ∑ yi∈D − log f̂∅[∅](yi) − HV(Y |X)− 1|D| ∑ xi,yi∈D − log f̂ [xi](yi)
∣∣∣∣∣∣ ≤ ∣∣∣∣∣∣HV(Y |X)− 1|D| ∑ xi,yi∈D − log f̂ [xi](yi) ∣∣∣∣∣∣+ ∣∣∣∣∣∣HV(Y )− 1|D| ∑ yi∈D − log f̂∅[∅](yi)
∣∣∣∣∣∣ (32) For simplicity let
DY |X = ∣∣∣∣∣∣HV(Y |X)− 1|D| ∑ xi,yi∈D − log f̂ [xi](yi) ∣∣∣∣∣∣ and
DY = ∣∣∣∣∣∣HV(Y )− 1|D| ∑ yi∈D − log f̂∅[∅](yi) ∣∣∣∣∣∣ With inequality (32), Lemma 3 and Lemma 4, we have:
Pr ∣∣∣IV(X → Y )− ÎV(X → Y ;D)∣∣∣ > 4R|D|(GV) + 2B √
2 log 1δ |D|  ≤ Pr DY |X +DY > 4R|D|(GV) + 2B √
2 log 1δ |D|  (Inequality (32)) ≤ Pr DY |X > 2R|D|(GV) +B √
2 log 1δ |D|
 ∨ DY > 2R|D|(GV) +B √ 2 log 1δ |D|  ≤ Pr DY |X > 2R|D|(GV) +B √
2 log 1δ |D|
+ Pr DY > 2R|D|(GV) +B √ 2 log 1δ |D|  (Union bound)
≤ 2δ (Lemma 3 and Lemma 4) Hence we have:
Pr ∣∣∣IV(X → Y )− ÎV(X → Y ;D)∣∣∣ ≤ 4R|D|(GV) + 2B √
2 log 1δ |D|  ≥ 1− 2δ which completes the proof.
A.4 PROOF OF COROLLARY 1.1
Corollary 1.1. Assume X = {x ∈ Rdx , ‖x‖2 ≤ kx} and Y = {y ∈ Rdy , ‖y‖2 ≤ ky}. If
V = {f : f [x] = N (Wx+ b, I), f [∅] = N (c, I),W ∈ Rdy×dx , b, c ∈ Rdy , ‖(W, b)‖2 ≤ 1}
Denote M = (kx + ky)2 + log 2π, then ∀δ ∈ (0, 0.5), with probability at least 1− 2δ:∣∣∣IV(X → Y )− ÎV(X → Y ;D)∣∣∣ ≤ M√ 4|D| ( 1 + 4 √ 2 log 1 δ )
The proof is an adaptation of the proof for theorem 3 in Kakade et al. (2008).
Proof. From theorem 1 we have:∣∣∣IV(X → Y )− ÎV(X → Y ;D)∣∣∣ ≤ 4R|D|(GV) + 2B √
2 log 1δ |D|
In the following ‖(W, b)‖2 is the matrix 2-norm of (W, b), then the Rademacher term can be bounded as follows:
R|D|(GV) = 1
|D| Eσ  sup W,b,‖(W,b)‖2≤1 ∣∣∣∣∣∣ |D|∑ i=1 σi ( log 1√ 2π − 1 2 ‖yi −Wxi − b‖22 )∣∣∣∣∣∣ 
≤ 1 |D| Eσ  sup W,b,‖(W, b)‖
2 ≤1
∣∣∣∣∣∣ |D|∑ i=1 σi ( −1 2 ‖yi −Wxi − b‖22 )∣∣∣∣∣∣ + 1 |D| Eσ ∣∣∣∣∣∣ |D|∑ i=1 σi log 1√ 2π ∣∣∣∣∣∣ 
(33)
The second term in RHS can be bounded as follows:
1
|D| Eσ ∣∣∣∣∣∣ |D|∑ i=1 σi log 1√ 2π ∣∣∣∣∣∣  ≤ 1 |D| √√√√√√Eσ   |D|∑ i=1 σi log 1√ 2π 2  (concavity of x 12 )
= 1
|D|
√ |D| ∗ (log 1√
2π )2 (Independence of σis)
=
√ (log 1√
2π )2
|D| (34)
The first term in RHS can be bounded as follows:
1
|D| ED,σ  sup W,b,‖(W, b)‖2≤1 ∣∣∣∣∣∣ |D|∑ i=1 σi ( −1 2 ‖yi −Wxi − b‖2 )∣∣∣∣∣∣ 
= 1
2|D| ED,σ  sup W,b,‖(W, b)‖2≤1 ∣∣∣∣∣∣ |D|∑ i=1 σi ( ‖yi −Wxi − b‖2 )∣∣∣∣∣∣ 
≤ maxi‖yi‖ 2 2
2
√ 1
|D| + max i ‖xi‖2 √ maxi‖yi‖2 |D|
+ 1
2|D| ED,σ  sup W,b,‖(W, b)‖2≤1 ∣∣∣∣∣∣ |D|∑ i=1 σi ( ‖Wxi + b‖2 )∣∣∣∣∣∣  (35)
≤ maxi‖yi‖ 2 2
2
√ 1
|D| + max i ‖xi‖2 √ maxi‖yi‖2 |D|
+ maxi‖xi‖2
2|D| ED,σ  sup W,b,‖(W, b)‖2≤1 ∣∣∣∣∣∣ |D|∑ i=1 σi (‖Wxi + b‖) ∣∣∣∣∣∣  (36)
≤ maxi‖yi‖ 2 2
2
√ 1
|D| + max i ‖xi‖2 √ maxi‖yi‖22 |D| + maxi‖xi‖2 2 √ maxi‖xi‖22 |D|
(37)
≤ M√ 4|D|
The inequalities (36) and (35) follow the same proof in (34).
Hence we have:
R|D|(GV) ≤ M√ 4|D|
(38)
In this example, we can bound the upper bound of functions g ∈ GV by
B = sup x∈X ,y∈Y,‖(W, b)‖2≤1
∣∣∣∣(log 1√2π − 12‖y −Wx− b‖22 )∣∣∣∣
≤ sup x∈X ,y∈Y,‖(W, b)‖2≤1
log 1√ 2π + 1 2
( ‖y‖22 + ‖Wx+ b‖22 + 2‖y‖‖Wx+ b‖ ) ≤ log 1√
2π +
1 2 (kx + ky) 2 < M
Combining inequality (38) we arrive at the theorem.
A.5 PROOF OF THEOREM 2
Theorem 2. Let {Xi}mi=1 be the set of m random variables, Di,j (resp. Dj) be the set of samples drawn from P (Xi, Xj) (resp. P (Xj)). Denote the optimal directed tree with maximum expected edge weights sum C(g) as g∗ and the optimal directed tree constructed on the dataset D as ĝ. Then with the assumption in theorem 1, for any δ ∈ (0, 12m(m−1) ), with probability at least 1− 2m(m− 1)δ, we have:
C(ĝ) ≥ C(g∗)− 2(m− 1) max i,j
{ 2RDi,j (GVi,j ) + 2RDj (GVj ) +B √ 2 log 1
δ (|Dj |− 1 2 + |Di,j |− 1 2 ) } (8)
Proof. Let CD(g∗) be the estimated sum of edge weights on dataset D of the tree g∗, i.e.,
C(g∗) = m∑ i=2 ÎVt(g∗)(i),i(Xt(g)(i) → Xi;D).
where t(g) : N→ N is the function mapping each non-root node of directed tree g to its parent. The same notation for tree ĝ. Let
= max i,j {∣∣∣IV(Xi → Xj)− ÎV(Xi → Xj ;D)∣∣∣} be the maximum absolute estimation error of single edge weight. By the definition of we have ∀g, |C(ĝ)− CD(ĝ)| ≤ (m− 1) , then:
C(ĝ) + (m− 1) ≥ CD(ĝ) ≥ CD(g∗) ≥ C(g∗)− (m− 1) (39)
From lemma 4 and lemma 3 we have:
Pr ( > max
i,j
{ 2RDi,j (Gi,j) + 2RDj (Gj) +B √ 2 log 1
δ (|Dj |− 1 2 + |Di,j |− 1 2 )
})
≤ Pr ( ∃i, j, ∣∣∣IVi,j (Xi → Xj)− ÎVi,j (Xi → Xj ;D)∣∣∣ > 2RDi,j (Gi,j) + 2RDj (Gj) +B√2 log 1δ (|Dj |− 12 + |Di,j |− 12 ) )
≤ Pr ∃i, j, ∣∣∣∣∣∣ HVj (Xj)− 1|Dj | ∑
xj∈Dj
− log f̂∅[∅](xj) − HVi,j (Xj |Xi)− 1|Di,j | ∑
xi,xj∈Di,j
− log f̂ [xi](xj) ∣∣∣∣∣∣ > 2RDi,j (Gi,j) + 2RDj (Gj) +B √ 2 log 1
δ (|Dj |− 1 2 + |Di,j |− 1 2 )
)
≤ Pr ∃i, j, ∣∣∣∣∣∣HVj (Xj)− 1|Dj | ∑ xj∈Dj − log f̂∅[∅](xj) ∣∣∣∣∣∣ > 2RDj (Gj) +B √ 2 log 1 δ |Dj |− 1 2  ∨ ∣∣∣∣∣∣HVi,j (Xj |Xi)− 1|Di,j | ∑
xi,xj∈Di,j
− log f̂ [xi](xj) ∣∣∣∣∣∣ > 2RDi,j (Gi,j) +B √ 2 log 1 δ |Di,j |− 1 2  ≤ m(m− 1)2δ (By lemma 3, 4 and union bound)
Hence
Pr ( ≤ max
i,j
{ 2RDi,j (Gi,j) + 2RDj (Gj) +B √ 2 log 1
δ (|Dj |− 1 2 + |Di,j |− 1 2 )
}) ≥ 1−m(m− 1)2δ
(40)
Then combining inequality (39) and (40) we arrive at the result.

heading: B ANALYSIS OF APPROXIMATE ESTIMATORS FOR SHANNON INFORMATION
text: We consider two approximate estimators for Shannon information. The first is the CPC (or InfoNCE in Poole et al. (2019)) estimator (ICPC) proposed by van den Oord et al. (2018):
ICPC = E
[ 1
N N∑ i=1 log fθ(xi, yi) 1 N ∑N j=1 fθ(xi, yj)
] ≤ I(X;Y ) (41)
where the expectation is over N independent samples form the joint distribution ∏ i p(xi, yi).
The second is the NWJ estimator (INWJ) proposed by Nguyen et al. (2010): INWJ = Ex,y∼p(x,y) [fθ(x, y)]− e−1Ex,y∼p(x)p(y) [ efθ(x,y) ] ≤ I(X;Y ) (42)
In both cases, fθ is a parameterized function, and the objectives are to maximize these lower bounds parameterized by θ to approximate mutual information. Ideally, with sufficiently flexible models and data, we would be able recover the true mutual information. However, these ideal cases does not carry over to practical scenarios.
For ICPC, van den Oord et al. (2018) show that ICPC is no larger than logN , where N is the batch size. This means that the ICPC estimator will incur large bias when I(X;Y ) ≥ logN . We provide a proof for completeness as follows.
Proposition 3. ∀fθ : X × Y → R+,
ICPC ≤ logN. (43)
Proof. We have:
ICPC := E
[ 1
N N∑ i=1 log fθ(xi, yi) 1 N ∑N j=1 fθ(xi, yj)
] (44)
≤ E
[ 1
N N∑ i=1 log fθ(xi, yi) 1 N fθ(xi, yi)
] ≤ E [ 1
N N∑ i=1 logN
] = logN (45)
which completes the proof. For NWJ, we note that the INWJ involves a term denoted as Ex,y∼p(x)p(y) [ efθ(x,y) ] /e, which could be dominated by rare data-points that have high fθ values. Intuitively, this would make it a poor mutual information estimator by optimizing θ. The NWJ estimator may suffer from high variance when the estimator is optimal (Song & Ermon, 2019), this is also empirically observed in Poole et al. (2019). We provide a proof for completeness as follows.
Proposition 4. Assume that fθ achieves the optimum value for INWJ. Then the variance of the empirical NWJ estimator satisfies Var ( ÎNWJ ) ≥ e I(X;Y )−1 N , where
ÎNWJ = 1
N N∑ i=1 [fθ(xi, yi)]− e−1 N N∑ i=1 [ efθ(x̄i,ȳi) ] is the empirical NWJ estimator with N i.i.d. samples {(xi, yi)}Ni=1 from p(x, y) and N i.i.d. samples {(x̄i, ȳi)}Ni=1 from p(x)p(y).
Proof. Let us denote zi = p(xi,yi) p(xi)p(yi) . Clearly Ep(x)p(y) [zi] = 1. Then we have:
Var(zi) = Ep(x)p(y) [ z2i ] − (Ep(x)p(y) [zi])2
= Ep(x)p(y) [ z2i ] − 1
= Ep(x)p(y)
[( p(xi, yi)
p(xi)p(yi)
)2] − 1
= Ep(x,y) [ p(xi, yi)
p(xi)p(yi)
] − 1 (46)
≥ eEp(x,y) [ log p(xi,yi) p(xi)p(yi) ] − 1 = eI(X;Y ) − 1 (47)
where we use Jensen’s inequality for log at the last step.
From Nguyen et al. (2010), we have:
fθ(x, y) = 1 + log p(x, y)
p(x)p(y) . (48)
for all x, y. Since {(xi, yi)}Ni=1 (resp. {(x̄i, ȳi)}Ni=1) are N datapoints independently sampled from the distribution p(x, y) (resp. p(x)p(y)), we have
Var ( ÎNWJ ) = Var
( 1
N N∑ i=1 [fθ(xi, yi)]− e−1 N N∑ i=1 [ efθ(x̄i,ȳi)
])
≥ Var
( e−1
N N∑ i=1 [ efθ(x̄i,ȳi)
])
= Var
( 1
N N∑ i=1 zi
) ≥ e
I(X;Y ) − 1 N
(49)
which completes the proof.
Algorithm 1 Construct Chow-Liu Trees with V-Information
Require: D = {X̂i}mi=1, with each X̂i being a set of datapoints sampled from the underlying distribution of random variable Xi. The set of function families {Vi,j}mi,j=1,i6=j between all the nodes.
1: for i = 1, . . . ,m do 2: for j = 1, . . . ,m do 3: if i 6= j then 4: Calculate the edge weight: ei→j = ÎVi,j (Xi → Xj ; {X̂i, X̂j}). 5: end if 6: end for 7: end for 8: Construct the fully connected graph G = (V,E), with node set V = (X1, . . . , Xm) and edge
set E = {ei→j}mi,j=1,i6=j . 9: Construct the maximal directed spanning tree g on G by Chow-Liu algorithm, where mutual
information is replaced by V-information. 10: return g

heading: C THE NEW ALGORITHM FOR CHU-LIU TREE CONSTRUCTION
text: See Algorithm 1; ÎVi,j (Xi → Xj ; {X̂i, X̂j}) denotes the empirical V-information.

heading: D DETAILED EXPERIMENTS SETUP
text: D.1 CHU-LIU TREE CONSTRUCTION
Figure 2 shows the Chu-Liu tree construction of Simulation-1∼Simulation-6. The Simulation-A and Simulation-B in the main body correspond to Simulation-1 and Simulation-4.
Simulation-1 ∼ Simulation-3 : The ground-truth Chu-Liu tree is a star tree (i.e. all random variables are conditionally independent given X1). We conduct all experiments for 10 times, each time with random simulated orthogonal matrices {Wi}20i=2. Simulation-1: X1 ∼ U(0, 10) and Xi | X1 ∼ N (WiX1, 6I), (2 ≤ i ≤ 20); Simulation-2: X1 ∼ U(0, 10) andXi | X1 ∼WiE(X1 + i), (2 ≤ i ≤ 20), i ∼ E(0.1); Simulation3 is a mixed version:X1 ∼ U(0, 10), Xi | X1 ∼ 12N (WiX1, 6I) + 1 2WiE(X1 + 1), (2 ≤ i ≤ 20).
Simulation-4 ∼ Simulation-6 : The ground-truth Chu-Liu tree is a tree of depth two. We conduct all experiments for 10 times, each time with random simulated orthogonal matrices {Wi}7i=2. Simulation-4: X1 ∼ U(0, 10),Xi | X1 ∼ N (WiX1, 2I)(i = 2, 3), Xi | X2 ∼ N (WiX2, 2I)(i = 4, 5), Xi | X3 ∼ N (WiX3, 2I)(i = 6, 7); Simulation-5: X1 ∼ U(0, 10),Xi | X1 ∼ E(X1+ i)(i = 2, 3),Xi | X2 ∼WiE(X2+ i)(i = 4, 5), Xi | X3 ∼ WiE(X3 + i)(i = 6, 7), i ∼ E(0.1); Simulation-6 is a mixed version: X1 ∼ U(0, 10), Xi | X1 ∼ WiE(X1 + i)(i = 2, 3), Xi | X2 ∼ N (WiX2, 2I)(i = 4, 5), Xi | X3 ∼ N (WiX3, 2I)(i = 6, 7), i ∼ E(0.1).
D.2 FAIRNESS
We can adapt the V-information perspective to fairness. Denote the random variable that represents sensitive data and the representation as U and Z respectively. Assume U is discrete and V belongs to preditive family 1. Then we have HV(U) = H(U) as long as V has softmax on the top and belongs to predictive family. In this case, minimizing IV(Z → U) equals to minimize −HV(Y |X). Let the joint distribution of Z and U be paramterized by φ. Hence the final objective is:
min φ {IV(u; z)} = min φ ( sup f∈V Ez,u∼qφ(z,u)[logPf (z|u)] )
In Edwards & Storkey (2015); Madras et al. (2018); Louizos et al. (2015); Song et al. (2018), functions in V are parameterized by a discriminator.
For the (Fi, Fj) elements described in the main body, please refer to figure 3b. The three datasets are: the UCI Adult dataset2 which has gender as the sensitive attribute; the UCI German credit dataset3 which has age as the sensitive attribute and the Heritage Health dataset4 which has the 18 configurations of ages and gender as the sensitive attribute.
The models in the figure are:
VA = {f : Z → P(U)|f [z](u) = ∑
(zi,ui)∈D
e‖zi−z‖ 2 2/h∑
(zi,ui)∈D e‖zi−z‖
2 2/h ∗ I(ui = u), h ∈ R}, where D is
the training set.
VB = {f : f [z] = softmax(g(z))}, where g is a two-layer MLP with Relu as the activation function. VC = {f : f [z] = softmax(g(z))}, where g is a three-layer MLP with LeakyRelu as the activation function.
We further visualize a special case of the (VA,VB) pair in figure 3a, where the Vi = {f : Z → P(U)|f [z](u) = ∑ (zi,ui)∈D e‖zi−z‖ 2 2/h∑ (zi,ui)∈D e‖zi−z‖ 2 2/h ∗ I(ui = u), h ∈ R} explicitly makes the features of different sensitivity attributes more evenly spread, and functions in VB is a simple two layers MLP with softmax at the top. The leaned features by VA-information minimization appear more evenly spread as expected, however, the attacker functions in VB can still achieve a high AUC of 0.857. The (i, j) elements of tables in Figure 3b stand for using function family Vi to attack features trained with Vj-information minimization. The diagonal elements in the matrix are usually the smallest in rows, indicating that the attacker function family Vi extracts more information on featured trained with Vj(j 6=i)-information minimization.
2https://archive.ics.uci.edu/ml/datasets/adult 3https://archive.ics.uci.edu/ml/datasets 4https://www.kaggle.com/c/hhp

heading: E MINIMALITY OF PREDICTIVE FAMILY
text: Define VX→P(Y) = {g : X → P(Y)|∃f ∈ V,∀x ∈ X , g[x] = f [x]}. Similarly define V∅→P(Y) = {g : ∅→ P(Y)|∃f ∈ V, g[∅] = f [∅]}. Intuitively, VX→P(Y) (resp. V∅→P(Y)) restricts the domain of functions in V to X (resp. ∅).
Non-Negativity As we demonstrated in Proposition 2, optional-ignorance guarantees that information will be non-negative for anyX and Y . Conversely, given any discreteX , Z, V∅→P(Y), VX→P(Y) that does not satisfy optional-ignorance, there exists distribution X , Y such that IV(X → Y ) < 0. Choose Y ∼ f∗[∅] where f∗ is the function that has no corresponding g ∈ VX→P(Y) that can ignore its inputs. Pick X as the uniform distribution, and note that for all g ∈ G, there exists some measurable subset X ′ ⊂ X on which g will produce a distribution unequal to f∗[∅], and therefore having higher cross entropy. The expected cross entropy expressed in HVX→P(Y)(Y |X) is thus higher than in HV∅→P(Y)(Y ), and IV(X → Y ) < 0. Thus, if the function class does not satisfy optional ignorance, then the V-information could be negative.
Independence Given any discrete X , Y , V∅→P(Y), VX→P(Y) that does not satisfy optionalignorance, there exists an independent X , Y such that IV(X → Y ) > 0. Choose Y such that the distribution PY can be expressed as g[x] for some x ∈ X, g ∈ VX→P(Y), but cannot be expressed by any f ∈ V∅→P(Y). Let X be the distribution with all its mass on x; note that the cross entropy of PY with g[x] will be zero, and is less than that of the function f [∅] (because f [∅] and PY differs on a measurable subset, the cross entropy will be positive). Thus, if the function class does not satisfy optional ignorance, then the V-information does not take value 0 when the two distributions are independent.

heading: F LIMITATIONS AND FUTURE WORK
text: V-information is empirically useful, has several intuitive theoretical properties, but exhibits certain limitations. For example, Shannon information can be manipulated with certain additive algebra (e.g. H(X,Y ) = H(X) +H(Y | X)), while the same does not hold true for general V-Information. However, this could be possible if we choose V to be a mathematically simple set, such as the set of polynomial time computable functions. It would be interesting to find special classes of V-Information where additional theoretical development is possible. Another interesting direction is better integration of V-Information with machine learning. The production of usable information (representation learning), acquisition of usable information (active learning) and exploitation of usable information (classification and reinforcement learning) could potentially be framed in a similar V-information-theoretic manner. It is interesting to see whether fruitful theories can arise from these analyses.

Review summaries:
The paper introduces a framework for quantifying information about one random variable, given another random variable (“side information”) and, importantly, a function class of allowed transformations that can be applied to the latter. This matches the typical scenario in machine learning, where observations (playing the role of side information) can be transformed (with a restricted class of transformation-functions) such that they become maximally predictive about another random variable of interest (“labels”). Using this framework, the paper defines the notion of conditional F-entropy and F-entropy (by conditioning on an empty set). Interestingly, both entropic quantities are shown to have many desirable properties known from Shannon entropy - and when allowing the function class of transformations to include all possible models F-entropies are equivalent to Shannon entropies. The paper then further defines “predictive F-information” which quantifies the increase in predictability about one random variable when given side information, under a restricted function-class of allowed transformations of the side information. Importantly, transformations of side information can increase predictive F-information (which is the basis for the notion of “usable” information), which is in contrast to the data processing inequality that applies to Shannon information and states that no transformation of a variable can increase predictability of another variable further than the un-transformed variable (information cannot be generated by transforming random variables). The paper highlights interesting properties of the F-quantities, most notably a PAC bound on F-information estimation from data, which gives reason to expect F-information estimation to be more data-efficient than estimating Shannon-information (particularly in the high-dimensional regime). This finding is confirmed by four types of interesting experiments, some of which make use of a modified version of a tree-structure learning algorithm proposed in the paper (using predictive F-information instead of Shannon mutual information).

The paper presents a generalization of classical definitions of entropy and mutual information that can capture computational constraints. Intuitively, information theoretic results assume infinite computational resources, so they may not correspond to how we treat "information" in practice. One example is public-key encryption. An adversary that has infinite time will eventually break the code so the decrypted message conveys the same amount of information (in a classical sense) as the plaintext message. In practice, this depends on computational time.