Natural AbstractionAbstraction
Frontpage
2023 Top Fifty: 16%

48

Natural latents are a relatively elegant piece of math which we figured out over the past year, in our efforts to boil down and generalize various results and examples involving natural abstraction. In particular, this new framework handles approximation well, which was a major piece missing previously. This post will present the math of natural latents, with a handful of examples but otherwise minimal commentary. If you want the conceptual story, and a conceptual explanation for how this might connect to various problems, that will probably be in a future post.

While this post is not generally written to be a "concepts" post, it is generally written in the hope that people who want to use this math will see how to do so.

2-Variable Theorems

This section will present a simplified but less general version of all the main theorems of this post, in order to emphasize the key ideas and steps.

Simplified Fundamental Theorem

Suppose we have:

  • A distribution  over random variables 
  • A latent variable  which induces independence between  and  (first diagram below)
  • Another latent variable  about which  and  give the same information (second diagram below)

Further, assume that  mediates between  and  (third diagram below). This last assumption can typically be satisfied by construction in the minimality/maximality theorems below.

 induces independence between  and  
 and  give the same information about 
 mediates between  and 

Then, claim:  mediates between  and .

 

Intuition

Picture  as a pipe between  and . The only way any information can get from  to  is via that pipe (that’s the first diagram).  is a piece of information which is present in both  and  - something we can learn from either of them (that’s the second pair of diagrams). Intuitively, the only way that can happen is if the information  went through the pipe - meaning that we can also learn it from .

The third diagram rules out three-variable interactions which could mess up that intuitive picture - for instance, the case where one bit of  is an xor of some independent random bit of  and a bit of .

Qualitative Example

Let  and  be the low-level states of two spatially-separated macroscopic chunks of an ideal gas at equilibrium. By looking at either of the two chunks, I can tell whether the temperature is above 50°C; call that  More generally, the two chunks are independent given the pressure and temperature of the gas; call that .

Notice that  is a function of , i.e. I can compute whether the temperature is above 50°C from the pressure and temperature itself, so  mediates between  and .

Some extensions of this example:

  • We could add more information to . For instance,  could be the pressure and temperature and also the outcome of a die roll unrelated to the gas. Or,  could be the entire low-level state of one of the two chunks.
  • We could remove information from  For instance,  could be a bit indicating whether the temperature is above 100°C

Intuitive mental picture: in general,  can’t have “too little” information; it needs to include all information shared (even partially) across  and  on the other hand, can’t have “too much” information; it can only include information which is fully shared across  and .

Proof

This is a diagrammatic proof; see Some Rules For An Algebra Of Bayes Nets for how to read it. (That post also walks through a version of this same proof as an example, so you should also look there if you want a more detailed walkthrough.)

(Throughout this post,  denotes all components of  except for .) 

Approximation

If the starting diagrams are satisfied approximately (i.e. they have small KL-divergence from the underlying distribution), then the final diagram is also approximately satisfied (i.e. has small KL-divergence from the underlying distribution). We can find quantitative bounds by propagating error through the proof:

Again, see the Algebra of Bayes Nets post for an unpacking of this notation and the proofs for each individual step.

Qualitative Example

Note that our previous example realistically involved approximation: two chunks of an ideal gas won’t be exactly independent given pressure and temperature. But they’ll be independent to a very (very) tight approximation, so our conclusion will also hold to a very tight approximation.

Quantitative Example

Suppose I have a biased coin, with bias . Alice flips the coin 1000 times, then takes the median of her flips. Bob also flips the coin 1000 times, then takes the median of his flips. We’ll need a prior on , so for simplicity let’s say it’s uniform on [0, 1].

Intuitively, so long as bias  is unlikely to be very close to ½, Alice and Bob will find the same median with very high probability. So:

  • Let  be Alice’ 1000 flips, and  be Bob’s 1000 flips.
  • Let  be the bias . Note that the flips are independent given , satisfying our first condition exactly.
  • Let  be the median computed by either Bob or Alice (we’re assuming they are the same with high probability). Since the same median can be computed with high probability from either  or , our second condition is approximately satisfied.
  • Since the median is computed as a deterministic function of , our third condition is satisfied exactly.

The fundamental theorem will then say that the bias approximately mediates between the median (either Alice’ or Bob’s) and the coinflips .

To quantify the approximation on the fundamental theorem, we first need to quantify the approximation on the second condition (the other two conditions hold exactly in this example, so their 's are 0). Let’s take  to be Alice’ median. Alice’ flips mediate between Bob’s flips and the median exactly (i.e. ), but Bob’s flips mediate between Alice’ flips and the median (i.e. ) only approximately. Let’s compute that :

This is a dirichlet-multinomial distribution, so it will be cleaner if we rewrite in terms of , and  is a function of , so the  is

Assuming I simplified the gamma functions correctly, we then get:

 (i.e. uniform over 0, …, n)

… and there’s only  values of , so we can combine those expressions with a python script to evaluate everything:

import numpy as np
from scipy.special import gammaln, logsumexp, xlogy

n = 1000
p_N2 = np.ones(n+1)/(n+1)
N1 = np.outer(np.arange(n + 1), np.ones(n + 1))
N2 = np.outer(np.ones(n + 1), np.arange(n + 1))
# logP[N1|N2]; we're tracking log probs for numerical stability
lp_N1_N2 = (gammaln(n + 2) - gammaln(N2 + 1) - gammaln(n - N2 + 1) +
            gammaln(n + 1) - gammaln(N1 + 1) - gammaln(n - N1 + 1) +
            gammaln(N1 + N2 + 1) + gammaln(2*n - N1 - N2 + 1) - gammaln(2*n + 2))

# logP[\Lambda' = 0|N2] and logP[\Lambda' = 1|N2]
lp_lam0_N2 = logsumexp(lp_N1_N2[:500], axis=0)
lp_lam1_N2 = logsumexp(lp_N1_N2[500:], axis=0)

p_lam0_N2 = np.exp(lp_lam0_N2)
p_lam1_N2 = np.exp(lp_lam1_N2)

print(p_lam0_N2 + p_lam1_N2)  # Check: these should all be 1.0

# ... aaaand then it's just the ol' -p * logp to get the expected entropy E[H(\Lambda')|N2]
H = - np.sum(p_lam0_N2 * lp_lam0_N2 * p_N2) - np.sum(p_lam1_N2 * lp_lam1_N2 * p_N2)
print(H / np.log(2))  # Convert to bits

The script spits out H = 0.058 bits. Sanity check: the main contribution to the entropy should be when  is near 0.5, in which case the median should have roughly 1 bit of entropy. How close to 0.5? Well, with  data points, posterior uncertainty should be of order , so the estimate of  should be precise to roughly  in either direction.  is initially uniform on [0, 1], so distance 0.03 in either direction around 0.5 covers about 0.06 in prior probability, and the entropy should be roughly that times 1 bit, so roughly 0.06 bits. Which is exactly what we found (closer than this sanity check had any business being, really); sanity check passes.

Returning to the fundamental theorem:  is 0,  is 0,  is roughly 0.058 bits. So, the theorem says that the coin’s true bias approximately mediates between the coinflips and Alice’ median, to within  0.12 bits. 

Exercise: if we track the ’s a little more carefully through the proof, we’ll find that we can actually do somewhat better in this case. Show that, for this example, the coin’s true bias approximately mediates between the coinflips and Alice’ median to within , i.e. roughly 0.058 bits.

Extension: More Stuff In The World

Let’s add another random variable  to the setup, representing other stuff in the world.  can have any relationship at all with , but  must still induce independence between  and . Further,  must not give any more information about  than was already available from  or . Diagrams:

 
 
 

Then, as before, we conclude

 

For instance, in our ideal gas example, we could take  to be the rest of the ideal gas (besides the two chunks  and ). Then, our earlier conclusions still hold, even though there’s “more stuff” physically in between  and  and interacting with both of them.

The proof goes through much like before, with an extra Z dangling everywhere:

And, as before, we can propagate error through the proof to obtain an approximate version:

Natural Latents

Suppose that a single latent variable  (approximately) satisfies both the first two conditions of the Fundamental Theorem, i.e.

We call these the “naturality conditions”. If  (approximately) satisfies both conditions, then we call  a “natural latent”.

Example: if  and  are two spatially-separated mesoscopic chunks of an ideal gas, then the tuple (pressure, temperature) is a natural latent between the two. The first property (“mediation”) applies because pressure and temperature together determine all the intensive quantities of the gas (like e.g. density), and the low-level state of the gas is (approximately) independent across spatially-separated chunks given those quantities. The second property (“insensitivity”) applies because the pressure and temperature can be precisely and accurately estimated from either chunk.

The Resample

Recall earlier that we said the condition

 

would be satisfied by construction in our typical use-case. It’s time for that construction.

The trick: if we have a natural latent , then construct a new natural latent by resampling  conditional on  (i.e. sample from ), independently of whatever other stuff  we’re interested in. The resampled  will have the same joint distribution with  as  does, so it will also be a natural latent, but it won’t have any “side-channel” interactions with other variables in the system - all of its interactions with everything else are mediated by , by construction.

From here on out, we’ll usually assume that natural latents have been resampled (and we’ll try to indicate that with the phrase “resampled natural latent”).

Minimality & Maximality

Finally, we’re ready for our main theorems of interest about natural latents.

Assume that  is a resampled natural latent over . Take any other latent  about which  and  give (approximately) the same information, i.e.

 

Then, by the fundamental theorem:

 

So  is, in this sense, the “most informative” latent about which  and  give the same information:

 is the "most informative about " variable which satisfies these conditions.

 and  give (approximately) the same information about , and for any other latent  such that  and  give (approximately) the same information about   tells us (approximately) everything which  does about  (and possibly more). This is the “maximality condition”.

Flipping things around: take any other latent  which (approximately) induces independence between  and :

 

By the fundamental theorem:

 

So  is, in this sense, the “least informative” latent which induces independence between  and :

 is the "most informative about " variable which satisfies this condition.

 (approximately) induces independence between  and , and for any other latent  which (approximately) induces independence between  and  tells us (approximately) everything which  does about  (and possibly more). This is the “minimality condition”.

Further notes:

  • The minimality condition implies the natural latent conditions. Proof: take  to be either  or .
  • Both minimality and maximality imply a unique standard form (see appendix), so any resampled natural latent in standard form is also the unique natural and minimal and maximal latent in standard form. (In the approximate case, this becomes approximate uniqueness.) Furthermore, any natural latent can be transformed into standard form (as the phrase “standard form” implies).
  • In standard form, a natural latent is always approximately a deterministic function of . Specifically: .
  • Natural and minimal latents do not always exist, and maximal latents sometimes exist in cases where natural/minimal latents don’t. (For example, if  and  are two flips of a biased coin of unknown bias, then no natural latent exists to a good approximation. Intuitively, it's because we don't have enough data to precisely identify the bias which mediates between the datapoints.)
  • Basically everything about approximation and other parts of the world carries over to naturality/minimality/maximality.

Example

In the ideal gas example, two chunks of gas are independent given (pressure, temperature), and (pressure, temperature) can be precisely estimated from either chunk. So, (pressure, temperature) is (approximately) a natural latent across the two chunks of gas. The resampled natural latent would be the (pressure, temperature) estimated by looking only at the two chunks of gas.

If there is some other latent which induces independence between the two chunks (like, say, the low-level state trajectory of all the gas in between the two chunks), then the (pressure, temperature) estimated from the two chunks of gas must (approximately) be a function of that latent. And, if there is some other latent about which the two chunks give the same information (like, say, a bit which is 1 if-and-only-if temperature is above 50 °C), then (pressure, temperature) estimated from the two chunks of gas must (approximately) also give that same information.

General Case

Now we move on to more-than-two variables. The proofs generally follow the same structure as the two-variable case, just with more bells and whistles, and are mildly harder to intuit. This time, rather than ease into it, we’ll include approximation and other parts of the world () in the theorems upfront.

The Fundamental Theorem

Suppose we have:

  • A distribution  over random variables , …, 
  • A latent variable  which induces independence between all  (first diagram below)
  • A latent variable  which is insensitive to any  given all the others (second diagram below)

Further, assume that  mediates between  and  (third diagram below). This last assumption will be exactly satisfied by construction in the minimality/maximality theorems below.

 

Then, claim:  mediates between  and .

 

The diagrammatic proof, with approximation:

 

Note that the error bound is somewhat unimpressive, since it scales with the system size. We can strengthen the approximation mainly by using a stronger insensitivity condition for  - for instance, if  is insensitive to either of two halves of the system, then we can reuse the error bounds from the simplified 2-variable case earlier.

Natural Latents

Suppose a single latent  (approximately) satisfies both the main conditions

 

We call these the “naturality conditions”, and we call  a (approximate) “natural latent” over .

The resample step works exactly as before: we’ll generally assume that  has been resampled conditional on , and try to indicate that with the phrase “resampled natural latent”.

Minimality & Maximality

Assume  is a resampled natural latent over . Take any other latent  about which  gives (approximately) the same information as  for any :

 

By the fundamental theorem:

 

So  is, in this sense, the “most informative” latent about which all  give the same information:

 

All  give (approximately) the same information about , and for any other latent  such that all  give (approximately) the same information about   tells us (approximately) everything which  does about  (and possibly more). This is the “maximality condition”.

Flipping things around: take any other latent  which (approximately) induces independence between components of . By the fundamental theorem:

 

So  is, in this sense, the “least informative” latent which induces independence between components of :

 

 (approximately) induces independence between components of , and for any other latent  which (approximately) induces independence between components of  tells us (approximately) everything which  does about  (and possibly more). This is the “minimality condition”.

Much like the simplified 2-variable case:

  • The minimality condition implies the natural latent conditions. Proof: take  to be , for each .
  • Both minimality and maximality imply an approximately unique standard form (see appendix), so any natural latent in standard form is also the approximately unique natural and minimal and maximal latent in standard form. Furthermore, any natural latent can be transformed into standard form (as the phrase “standard form” implies).
  • In standard form, a natural latent is always approximately a deterministic function of . Specifically: .
  • Natural and minimal latents do not always exist, and maximal latents sometimes exist in cases where natural/minimal latents don’t.
  • Everything about approximation and other parts of the world carries over to naturality/minimality/maximality.

Example

Suppose we have a die of unknown bias, which is rolled many times - enough to obtain a precise estimate of the bias many times over. Then, the die-rolls are independent given the bias, and we can get approximately-the-same estimate of the bias while ignoring any one die roll. So, the die’s bias is an approximate natural latent. The resampled natural latent is the bias sampled from a posterior distribution of the bias given the die rolls.

One interesting thing to note in this example: imagine that, instead of resampling the bias given the die rolls, we use the average frequencies from the die rolls. Would that be a natural latent? Answer in spoiler text:

The average frequencies tell us the exact counts of each outcome among the die rolls. So, with the average frequencies and all but one die roll, we can back out the value of that last die roll with certainty: just count up outcomes among all the other die rolls, then see which outcome is “missing”. That means the average frequencies and  together give us much more information about  than either one alone; neither of the two natural latent conditions holds.

This example illustrates that the “small” uncertainty in  is actually load-bearing in typical problems. In this case, the low-order bits of the average frequencies contain lots of information relevant to , while the low-order bits of the natural latent don’t.

The most subtle challenges and mistakes in using natural latents tend to hinge on this point.

Other Threads

We’ve now covered the main theorems of interest. This section offers a couple of conjectures, with minimal commentary.

Universal Natural Latent Conjecture

Suppose there exists an approximate natural latent over . Construct a new random variable  sampled from the distribution . (In other words: simultaneously resample each  given all the others.) Conjecture:  is an approximate natural latent (though the approximation may not be the best possible). And if so, a key question is: how good is the approximation?

Further conjecture: a natural latent exists over  if-and-only-if  is a natural latent over .(Note that we can always construct , regardless of whether a natural latent exists, and  will always exactly satisfy the first natural latent condition over  by construction.) Again, assuming the conjecture holds at all, a key question is to find the relevant approximation bounds.

Maxent Conjecture

Assuming the universal natural latent conjecture holds, whenever there exists a natural latent  and  would approximately satisfy both of these two diagrams:

 

If we forget about all the context and just look at these two diagrams, one natural move is:

  • Convert to undirected graphical models (in this case, just remove the arrowheads)
  • Apply the Hammersley-Clifford method

… and the result would be that  must be of the maxent form

for some .

Unfortunately, Hammersley-Clifford requires  everywhere. Probabilities of zero are extremely load-bearing for natural latents in the exact case, and probabilities near zero are load-bearing in the approximate case; if the distribution is zero nowhere, then it can only have a natural latent if the ’s are all independent (in which case the trivial variable is a natural latent).

On the other hand, maxent sure does seem to be a theme of natural latents in e.g. physics. So the question is: does some version of this argument work? If there are loopholes, are there relatively few qualitative categories of loopholes which could be classified?

Appendix: Machinery For Latent Variables

If you get confused about things like “what exactly is a latent variable?”, this is the appendix for you.

For our purposes, a latent variable  over a random variable  is defined by a distribution function  - i.e. a distribution which tells us how to sample  from . For instance, in the ideal gas example, the latent “average temperature” () can be defined by the distribution of average energy as a function of the gas state () - in this case a deterministic distribution, since average energy is a deterministic function of the gas state. As another example, consider the unknown bias of a die () and a bunch of rolls of the die (). We would take the latent bias to be defined by the function mapping the rolls  to the posterior distribution of the bias given the rolls .

So, if we talk about e.g. “existence of a latent  with the following properties …”, we’re really talking about the existence of a conditional distribution function , such that the random variable  which it introduces satisfies the relevant properties. When we talk about “any latent satisfying the following properties …”, we’re really talking about any conditional distribution function  such that the variable  satisfies the relevant properties.

Toy mental model behind this way of treating latent variables: there are some variables  out in the world, and those variables have some “true frequencies” . Different agents learn to model those true frequencies using different generative models, and those different generative models each contain their own latents. So the underlying distribution  is fixed, but different agents introduce different latents with different  is taken to be definitional because it tells us everything about the joint distribution of  and  which is not already pinned down by . (This is probably not the most general/powerful mental model to use for this math, but it’s a convenient starting point.)

Standard Form of a Latent Variable

Suppose that  consists of a bunch of rolls of a biased die, and a latent  consists of the bias of the die along with the flip of one coin which is completely independent of the rest of the system. A different latent for the same system consists of just the bias of the die. We’d like some way to say that these two latents contain the same information for purposes of .

To do that, we’ll introduce a “standard form”  (relative to ) for any latent variable :

We’ll say that a latent is “in standard form” if, when we use the above function to put it into standard form, we end up with a random variable equal to the original. So, a latent  in standard form satisfies

As the phrase “standard form” suggests, putting a standard form latent into standard form leaves it unchanged, i.e.  will satisfy  for any latent ; that’s a lemma of the minimal map theorem.

Conceptually: the standard form throws out all information which is completely independent of , and keeps everything else. In standard statistics jargon: the likelihood function is always a minimal sufficient statistic. (See the minimal map theorem for more explanation.)

Addendum [Added June 12 2024]: Approximate Isomorphism of Natural Latents Which Are Approximately Determined By X

With just the mediation and redundancy conditions, natural latents aren’t quite unique. For any two natural latents  over the same variables , we have  and . Intuitively, that means the two natural latents “say the same thing about ”; each mediates between  and the other. But either one could still contain some “noise” which is independent of , and  could contain totally different “noise” than . (Example: X consists of 1000 flips of a biased coin. One natural latent might be the bias, another might be the bias and one roll of a die which has nothing to do with the coin, and a third natural latent might be the bias and the amount of rainfall in Honolulu on April 7 2033.)

In practice, it’s common for the natural latents we use to be approximately a deterministic function of . In that case, the natural latent in question can’t include much “noise” independent of . So intuitively, if two natural latents over  are both approximately deterministic functions of , then we’d expect them to be approximately isomorphic.

Let’s prove that.

Other than the tools already used in the post, the main new tool will be diagrams like , i.e.  mediates between  and itself. (If the same variable appearing twice in a Bayes net is confusing, think of it as two separate variables which are equal with probability 1.) The approximation error for this diagram is the entropy of  given :

So if we want to express “ is approximately a deterministic function of ” diagrammatically, we can use the diagram . The approximation error on the diagram is , which makes intuitive sense: low entropy of  given  means that  is approximately determined by .

With that in mind, here’s the full proof of isomorphism, including approximation errors:

Let’s walk through those steps.

The first piece should be familiar: it’s just the fundamental theorem, the main focus of this whole post. As usual, it tells us that either of  mediates between the other and . (Minor note:  is intended to denote a subset in some partition of ; it's a generalization of the notation used in the post, in order to allow for stronger redundancy conditions.)

The next two steps are the new part. They start with two diagrams expressing that  and  are both approximately deterministic functions of 

… and we conclude that  is an approximately deterministic function of 

Since  and  appear symmetrically in the assumptions, the same proof also shows that  is an approximately deterministic function of . Thus, approximate isomorphism, specifically in the sense that either of the two natural latents has approximately zero entropy given the other.

New Comment
5 comments, sorted by Click to highlight new comments since:

Probabilities of zero are extremely load-bearing for natural latents in the exact case, and probabilities near zero are load-bearing in the approximate case; if the distribution is zero nowhere, then it can only have a natural latent if the ’s are all independent (in which case the trivial variable is a natural latent).

I'm a bit confused why this is the case. It seems like in the theorems, the only thing "near zero" is that D_KL (joint, factorized) < epsilon ~= 0 . But you. can satisfy this quite easily even with all probabilities > 0. 

E.g. the trivial case where all variables are completely independents satisfies all the conditions of your theorem, but can clearly have every pair of probabilities > 0. Even in nontrivial cases, this is pretty easy (e.g. by mixing in irreducible noise with every variable).

Roughly speaking, all variables completely independent is the only way to satisfy all the preconditions without zero-ish probabilities.

This is easiest to see if we use a "strong invariance" condition, in which each of the  must mediate between  and . Mental picture: equilibrium gas in a box, in which we can measure roughly the same temperature and pressure () from any little spatially-localized chunk of the gas (). If I estimate a temperature of 10°C from one little chunk of the gas, then the probability of estimating 20°C from another little chunk must be approximately-zero. The only case where that doesn't imply near-zero probabilities is when all values of both chunks of gas always imply the same temperature, i.e.  only ever takes on one value (and is therefore informationally empty). And in that case, the only way the conditions are satisfied is if the chunks of gas are unconditionally independent.

Hm, it sounds like you're claiming that if each pair of x, y, z are pairwise independent conditioned on the third variable, and p(x, y, z) =/= 0 for all x, y, z with nonzero p(x), p(y), p(z), then ?

I tried for a bit to show this but couldn't prove it, let alone the general case without strong invariance. My guess is I'm probably missing something really obvious. 

Yeah, that's right.

The secret handshake is to start with " is independent of  given " and " is independent of  given ", expressed in this particular form:

... then we immediately see that  for all  such that .

So if there are no zero probabilities, then  for all .

That, in turn, implies that  takes on the same value for all Z, which in turn means that it's equal to .  Thus  and  are independent. Likewise for  and . Finally, we leverage independence of  and  given :

(A similar argument is in the middle of this post, along with a helpful-to-me visual.)

Right, the step I missed on was that P(X|Y) = P(X|Z) for all y, z implies P(X|Z) = P(X). Thanks!