What are principled ways for penalising complexity in practice?

Jun 27, 2019

530

I keep meaning to write a post on this...

At least within Bayesian probability, there is a single unique unambiguously-correct answer to "how should we penalize for complexity?": the Bayes factor. The Bayes factor is Hard to compute in general, which is why there's a whole slew of of other numbers which approximate it in various ways.

Here's how Bayes factors work. Want to know whether model 1 or model 2 is more consistent with the data? Then compute $P [m o d e l_{1} | d a t a]$ and $P [m o d e l_{2} | d a t a]$ . Using Bayes' rule:

$P [m o d e l_{i} | d a t a] = \frac{1}{Z} P [d a t a | m o d e l_{i}] P [m o d e l_{i}]$

where Z is the normalizer. If we're just comparing two models, then we can get rid of that annoying Z by computing odds for the two models:

$\frac{P [m o d e l_{1} | d a t a]}{P [m o d e l_{2} | d a t a]} = \frac{P [d a t a | m o d e l_{1}]}{P [d a t a | m o d e l_{2}]} \frac{P [m o d e l_{1}]}{P [m o d e l_{2}]}$

In English: posterior relative odds of the two models is equal to prior odds times the ratio of likelihoods. That likelihood ratio $\frac{P [d a t a | m o d e l_{1}]}{P [d a t a | m o d e l_{2}]}$ is the Bayes' factor: it directly describes the update in the relative odds of the two models, due to the data.

Example

20 coin flips yield 16 heads and 4 tails. Is the coin biased?

Here we have two models:

Model 1: coin unbiased
Model 2: coin has some unknown probability $θ$ of coming up heads (we'll use a uniform prior on $θ$ for simplicity; this is not a good idea in general)

The second model has one free parameter (the bias) which we can use to fit the data, but it’s more complex and prone to over-fitting. When we integrate over that free parameter, it will fit the data poorly over most of the parameter space - thus the "penalty" associated with free parameters in general.

In this example, the integral is exactly tractable (it's a dirichlet-multinomial model), and we get:

$P [d a t a | m o d e l_{1}] = (\frac{20}{4}) (\frac{1}{2})^{16} (\frac{1}{2})^{4} = 0.0046$
$P [d a t a | m o d e l_{2}] = \int_{θ} (\frac{20}{4}) (θ)^{16} (1 - θ)^{4} d θ = 0.048$

So the Bayes factor is (.048)/(.0046) ~ 10, in favor of a biased coin. In practice, I'd say unbiased coins are at least 10x more likely than biased coins in day-to-day life a priori, so we might still think the coin is unbiased. But if we were genuinely unsure to start with, then this would be pretty decent evidence in favor.

Approximation

Why is this hard to compute in general? Well, if the model has an m-dimensional vector of free parameters $θ$ , then $P [d a t a | m o d e l] = \int_{θ} P [d a t a | m o d e l, θ] d P [θ | m o d e l]$ , which is an integral in m dimensions. That's not just NP-complete, it's #P-complete.

There's various ways to approximate that integral. The two most common are:

MCMC (see e.g. here for application to Bayes factors): main limitation is that MCMC gets finicky in high dimensions.
Laplace approximation around the maximum-likelihood point. Aside from the usual limitations of using a single maximum-likelihood point, main limitation is that we need the determinant of the Hessian, which takes somewhere between O(k) and O( $k^{3}$ ) to compute for most k-free-parameter models (depending on how clever we are with the linear algebra).

BIC

The BIC can be derived directly from the Laplace approximation. Laplace says:

$\int_{θ} P [d a t a | θ] p [θ] d θ \approx P [d a t a | θ^{*}] p [θ^{*}] (2 π)^{\frac{k}{2}} d e t (- \frac{\partial^{2} l n P [d a t a | θ]}{\partial θ^{2}})^{- \frac{1}{2}}$

where $θ^{*}$ is the maximum likelihood point and k is the dimension of $θ$ .

Notice that $l n (P [d a t a | θ^{*}] p [θ^{*}])$ is the usual "score" of a model; the rest is what we can think of as a complexity penalty, and that's what the BIC will approximate. The main assumption is that $\frac{\partial^{2} l n P [d a t a | θ]}{\partial θ^{2}}$ scales with the number of data points N; this is trivial if the data points are iiid (given $θ$ ), since in that case $l n P [d a t a | θ]$ is a sum over data points.

Now for the trick: that's the only part of the Laplace approximation's "complexity penalty" which scales with the number of data points N. It scales in proportion to $N^{- \frac{k}{2}}$ , since each of k dimensions of the determinant contributes a factor N. Take the log of the whole thing, and all the rest will be constants which are dominated by the $\frac{k}{2} l n N$ term as N goes to infinity. And that's the BIC: log of the MAP probability, minus $\frac{k}{2} l n N$ .

Finally, the important part: the main benefit of knowing all this is being able to predict when the BIC will and will not work well. If it's an approximation of Laplace, which is itself an approximation of the Bayes factor, then BIC will work well exactly when those approximations are accurate. In particular, BIC fails when the terms we ignored (i.e. $l n ((2 π)^{\frac{k}{2}} d e t (- \frac{1}{N} \frac{\partial^{2} l n P [d a t a | θ]}{\partial θ^{2}})^{- \frac{1}{2}})$ ) are as large as or even larger than $l n (P [d a t a | θ^{*}] p [θ^{*}]) - \frac{k}{2} l n N$ , i.e. the BIC itself.

[-]Sherrinford6y20

Very helpful comment. But why the uniform prior, and not the bias that has the highest likelihood given the data? theta = 4/5 would give you a Bayes factor of 47, right?

[-]johnswentworth6y130

First things first, keep in mind the thing we're actually interested in: P[model | data]. The Bayes factor summarizes the contribution of the data to that number, in a convenient and reusable way, but the Bayes factor itself is not what we really want to know.

In the example, one model is "unbiased coin", the other is "some unknown bias". The prior over the bias is our distribution on $θ$ before seeing the data. Thus the name "prior". It's $P [θ | m o d e l]$ , not $P [θ | m o d e l, d a t a]$ . Before seeing the data, we have no reason at all to think that 4/5 is special. Thus, a uniform prior. This all comes directly from applying the rules of probability: $P [d a t a | m o d e l_{2}] = \int_{θ} P [d a t a | θ] p [θ | m o d e l_{2}] d θ$

If you stick strictly to the rules of probability itself, you will never find any reason to maximize anything. There is no built-in maximization anywhere in probability. There are lots of integrals (or sums, in the discrete case), and we usually approximate those integrals by maximizing things - but those are approximations, always. In particular, if you try to compute P[model | data], you will not find any maximization of anything - unless you introduce it to approximate an i... (read more)

[-]Bucky6y10

Thanks for this. I’m trying to get an intuition on how this works.

My mental picture is to imagine the likelihood function with respect to theta of the more complex model. The simpler model is the equivalent of a square function with height of its likelihood and width 1.

The relative areas under the graphs reflect the likelihood of the models. So if picturing the relative maximum likelihoods and how sharp the peak is on the more complex model gives an impression of the Bayes factor.

Does that work? Or is there a better mental model?

3johnswentworth6y

It's kind of tricky to picture it in terms of areas, because the two models usually don't live in the same space - one is an integral over more dimensions than the other. You can sometimes shoehorn it into the same space, e.g. by imagining that the likelihood for one model is constant with respect to some of the θ's - it sounds like that what's you're trying to do here. That's not wrong, and you can think of it that way if you want, but personally I find it a bit awkward. Here's a similar (but hopefully more natural) intuition: * Pick a model * Generate a θ value at random from that model's prior * Compute the likelihood of the data given that θ value * Take the average likelihood from this process, and you have P[data|model] Compare this to the traditional maximum-likelihood approach: max likelihood says "what's the highest likelihood which I could possibly assign to this data?", whereas the Bayes factor approach says "what's the average likelihood I'd assign to this data, given my prior knowledge?". It's analogous to the distinction between a maximax decision rule ("take the decision with the best available best-case outcome") and a Bayesian decision rule ("take the decision with the best available average outcome"). That said, the above intuition is still somewhat ad-hoc. Personally, I intuit Bayes factors with the same intuitions I use for any other probability calculations. It's just a natural application of the usual probability ideas to the question of "what's the right model, given my background knowledge and all this data?". In particular, the intuitions involved are: * Bayes' Rule * Gears-level understanding, usually manifesting as causal models. In this case, the causal model is θ -> data, with the individual data points independent conditional on θ. I then reflexively think "if I knew the values of all the nodes in this causal model, then I could easily compute the probability of the whole thing" - i.e. I can easily compute P[data,θ|model]. *

SatvikBeri

Jun 28, 2019

130

A different view is to look at the search process for the models, rather than the model itself. If model A is found from a process that evaluates 10 models, and model B is found from a process that evaluates 10,000, and they otherwise have similar results, then A is much more likely to generalize to new data points than B.

The formalization of this concept is called VC dimension and is a big part of Machine Learning Theory (although arguably it hasn't been very helpful in practice): https://en.wikipedia.org/wiki/Vapnik%E2%80%93Chervonenkis_dimension

[-]John_Maxwell6y90

Is there a theoretical justification for using VC dimension as a basis for generalization, or is it treated as an axiomatic desideratum?

5SatvikBeri6y

Yes, roughly speaking, if you multiply the VC dimension by n, then you need n times as much training data to achieve the same performance. (More precise statement here: https://en.wikipedia.org/wiki/Vapnik%E2%80%93Chervonenkis_dimension#Uses) There are also a few other bounds you can get based on VC dimension. In practice these bounds are way too large to be useful, but an algorithm with much higher VC dimension will generally overfit more.

2John_Maxwell6y

Oh yeah, thanks, I think I remember seeing that. Do you happen to know the assumptions that proof makes? I'm having a hard time finding it in Vapnik's textbook. I vaguely remember it assuming that the correct hypothesis is in our hypothesis class, which made it seem kind of uninteresting. To be clear, I agree it's easier to overfit if you try lots of models, but my explanation of this would be more Bayesian than Vapnik's. (Maybe something like: If a researcher restricts themselves to only 10 models, they will choose 10 models they feel relatively optimistic about/assign a high prior probability to. A high prior with a high likelihood gives us better generalization than a low prior with a slightly higher likelihood; the posterior probability of the first model is greater.)

[-]SatvikBeri6y110

A good exposition of the related theorems is in Chapter 6 of Understanding Machine Learning (https://www.amazon.com/Understanding-Machine-Learning-Theory-Algorithms/dp/1107057132/ref=sr_1_1?crid=2MXVW7VOQH6FT&keywords=understanding+machine+learning+from+theory+to+algorithms&qid=1562085244&s=gateway&sprefix=understanding+machine+%2Caps%2C196&sr=8-1)

There are several related theorems. Roughly:

1. The error on real data will be similar to the error on the training set + epsilon, where epsilon is roughly proportional to (datapoints / VC dimension.) This is the one I linked above.

2. The error on real data will be similar to the error of the best hypothesis in the hypothesis class, with similar proportionality

3. Special case of 2 – if the true hypothesis is in the hypothesis class, then the absolute error will be < epsilon (since the absolute error is just the difference from the true, best hypothesis.)

3 is probably the one you're thinking of, but you don't need the hypothesis to be in the class.

SoerenMind

Jun 28, 2019

130

The exact Bayesian solution penalizes complex models as a side effect. Each model should have a prior over its parameters. The more complex model can fit the data better, so P(data | best-fit parameters, model) is higher. But the model gets penalized because P(best-fit parameters | model) is lower on the prior. Why? The prior is thinly spread over a higher dimensional parameter space so it is lower for any particular set of parameters. This is called "Bayesian Occam's razor".

yagudin

Jul 02, 2019

If one has 2 possible models to fit a data set, by how much should one penalize the model which has an additional free parameter?

Penalization might not be necessary if your learning procedure is stochastic and favors simple explanations. I encourage you to take a look on the nice poster/paper «Deep learning generalizes because the parameter-function map is biased towards simple functions» (PAC-Bayesian learning theory + empirical intuitions).

LESSWRONG
LW

LESSWRONG
LW

36

[ Question ]

What are principled ways for penalising complexity in practice?

36

36

4 Answers sorted by
top scoring

Jun 27, 2019

Example

Approximation

BIC

Jun 28, 2019

Jun 28, 2019

Jul 02, 2019

36

[ Question ]

What are principled ways for penalising complexity in practice?

36

36

4 Answers sorted by top scoring

Jun 27, 2019

Example

Approximation

BIC

Jun 28, 2019

Jun 28, 2019

Jul 02, 2019

4 Answers sorted by
top scoring