DSLT 2. Why Neural Networks obey Occam's Razor

post by Liam Carroll (liam-carroll) · 2023-06-18T00:23:14.725Z · LW · GW · 14 comments

Contents

  Information Criteria Help Avoid Underfitting and Overfitting
  Watanabe's Free Energy Formula for Singular Models
    Free Energy, Generalisation and Model Selection
    The Free Energy Formula for Singular Models
    The Widely Applicable Bayesian Information Criterion
      The Accuracy-Complexity Tradeoff
      Why Singular Models (Can) Generalise Well
    From points to local neighbourhoods
  Intuitive Examples to Interpret the WBIC
    Example 1: True parameters are preferred according to their RLCT
      K(w) is like a potential well, where a lower RLCT means a flatter floor
      Animation 1: The posterior concentrates on both true parameters as n→∞, but it prefers the one with lower RLCT
    Example 2: Non-true parameters can be preferred at finite n because of the accuracy-complexity tradeoff
      Animation 2: At low n the simpler model is preferred, but as n→∞, the true parameter is preferred
    Asymptotic Analysis Requires Care
  Where are we going from here?
  References
None
14 comments

TLDR; This is the second main post of Distilling Singular Learning Theory [? · GW] which is introduced in DSLT0 [LW · GW]. I synthesise why Watanabe's free energy formula explains why neural networks have the capacity to generalise well, since different regions of the loss landscape have different accuracy-complexity tradeoffs. I also provide some simple intuitive examples [LW · GW] that visually demonstrate why true parameters (i.e. optimally accurate parameters) are preferred according to the RLCT as $n\to \infty$, and why non-true parameters can still be preferred at finite $n$ if they have lower RLCT's, due to the accuracy-complexity tradeoff. (The RLCT is introduced and explained in DSLT1 [? · GW]).

It is an amazing fact that deep neural networks seem to have an inductive bias towards "simple" models, suggesting [LW · GW] that they obey a kind of Occam's Razor:

Plurality should not be posited without necessity.

or in modern parlance,

If two models of the world are similarly accurate, the simpler explanation should be preferred.

This allows them to achieve exceptionally low generalisation error despite classical statistics predictions that they should overfit data: 

This fact has come to be known as the generalisation problem and has been discussed at length in Zhang et. al 2017 (and a 2021 supplement), and in Bengio et al., amongst countless others. 

Remarkably, Singular Learning Theory can help explain why neural networks, which are singular models, have the capacity to generalise so well.

The degeneracy of the Fisher information matrix is actually a feature of singular models, not a bug. This is because different regions of parameter space can have different complexities as measured by the RLCT $\lambda$, unlike regular models where the complexity is fixed to the total number of parameters in the model $d$. This is the implicit content of Watanabe's profound free energy formula, called the Widely Applicable Bayesian Information Criterion (WBIC), which quantifies a precise asymptotic tradeoff between inaccuracy and complexity, 

$$WBIC=n\underset{\text{inaccuracy}}{\underset{}{L_n\left(w^{\left(0\right)}\right)}}+\underset{\text{complexity}}{\underset{}{\lambda }}\log n,$$

giving a mathematically rigorous realisation of Occam's Razor, since $\lambda \le \frac{d}{2}$ in singular models. 

In this post we will explore Watanabe's free energy formula and provide an intuitive example of why the RLCT matters so much. If you are new to statistical learning theory, I would recommend jumping straight to the examples [LW · GW] and their related animations to gain the intuition first, and then return to the theory afterwards.

The four key points to take away are:

1. As $n\to \infty$, true parameters with the best accuracy will always be preferred.
2. As $n\to \infty$, if two true parameters are equally accurate but have different RLCT's, the parameter with the lower RLCT is preferred.
3. For finite but large $n$, non-true parameters can be preferred by the posterior because of an accuracy-complexity tradeoff as measured by the WBIC.
4. Parameters with low inaccuracy and small RLCT's $\lambda$ have low generalisation error (in a Bayesian sense) since the Bayes generalisation error $G_n$ is the "derivative" of the free energy, so 

$$G_n=L_n\left(w^{\left(0\right)}\right)+\frac{\lambda }{n}+o\left(\frac{1}{n}\right).$$

Information Criteria Help Avoid Underfitting and Overfitting

In the last post, we derived the asymptotic free energy $F_n$ as $n\to \infty$ for regular models, called the Bayesian Information Criterion (BIC): 

$$BIC=n{L}_n\left(w^{\left(0\right)}\right)+\frac{d}{2}\log n,$$

where $n$ is the total number of datapoints in the dataset $D_n$, the optimal loss is $L_n\left(w^{\left(0\right)}\right)$ where $w^{\left(0\right)}\in W$ is a maximum likelihood estimate (i.e. $w^{\left(0\right)}={argmin}_W\left(L\right(w\left)\right)$), and $d$ is the total dimension of parameter space $W\subseteq R^d$.

As a statistical practitioner, given some dataset $D_n$, your goal is to find a model that you hope will represent the truth from some candidate list. You only have access to the truth via your (training) dataset $D_n$, but you also want to ensure that it generalises to data beyond the dataset. You can use the BIC to compare model candidates across a set of model classes that you can think to compare, since it captures a precise asymptotic tradeoff between inaccuracy $L_n\left(w^{\left(0\right)}\right)$ and complexity $\frac{d}{2}$. Under this paradigm, we should choose the model that achieves the lowest BIC as it is the best option for avoiding both underfitting and overfitting the data. Let's consider a simple example in action:

Example 1: Suppose we have $n=61$ datapoints drawn from a quadratic with Gaussian noise, $y=x^2+\epsilon$ where $\epsilon \sim N(0,{0.15}^2)$^[1], where $x$ is drawn according to a uniform prior $q\left(x\right)=\frac{1}{2}1(x\in [-1,1\left]\right)$. After looking at our scatterplot of data $D_n$, we could try models across the following set of model classes:

Name	$d$	Model
Linear	2	$f_1(x,w)=w_{1}x+w_0$
Quadratic	3	$f_2(x,w)=w_2{x}^2+w_{1}x+w_0$
Cubic	4	$f_3(x,w)=w_3{x}^3+w_2{x}^2+w_{1}x+w_0$
Degree 15	16	$f_{15}(x,w)=w_{15}{x}^{15}+w_{14}{x}^{14}+\cdots +w_{1}x+w_0$

(The degree 15 model is an extremity just to illustrate a point.)

Within each model class, we can then perform ordinary least squares regression ^[2] to find the model fit $w^{\left(0\right)}$ with optimal loss $L_n\left(w^{\left(0\right)}\right)$ (which, in the regression case, is simply the mean-squared-error plus a constant^[3]). With the optimal loss of each model class in hand, we can then compare the $BIC$ over our set of candidates.

Model	$L_n\left(w^{\left(0\right)}\right)$	$d$	$BIC$
$f_1(x,w)$	0.9668	1	63.09
$f_2(x,w)$	0.9283	2	62.80
$f_3(x,w)$	0.9280	3	64.83
$f_{15}(x,w)$	0.9243	15	87.22

As one expects, in a similar vein to the bias-variance tradeoff, there is a clear optimal (lowest) $BIC$. As the dimension increases, the accuracy gets better and better, but at the cost of the complexity of the model (and therefore, its generalisability^[4]). The linear model is simple, but has high loss. The cubic has marginally lower loss, but at the expense of a complexity increase that isn't worth it. The degree 15 polynomial has the lowest loss of them all, but is penalised heavily for its complexity (as it should be - it is clearly overfitting). The Goldilocks choice is unsurprisingly the quadratic model, $f_2\left(x\right)$, because the tradeoff between accuracy and complexity is just right.

Other than the fact that the $BIC$ simply does not hold in the singular case, it also points us towards a limitation of regular models. Once you pick your model class to optimise, every point on the loss landscape has a fixed model complexity. If your goal is to minimise the $BIC$, you only have one choice: to find the single point that optimises the loss at the bottom of the well.

In fact, in our particular case, we can calculate the KL divergence ^[5] for the linear model $f_2(x,w)$, 

$$K(w_0,w_1)=\frac{1}{6}{w}_1^2+\frac{1}{2}{(w_0-\frac{1}{3})}^2+\frac{2}{45},$$

 which we can see a plot of below.

In singular models, this all changes: within the same model class, different models $w$ in parameter space $W$ have different effective dimensionalities as measured by the RLCT $\lambda$. The learning procedure does the work of the statistician for us, because the loss landscape contains information about both the accuracy and the complexity. 

Watanabe's Free Energy Formula for Singular Models

Free Energy, Generalisation and Model Selection

Fundamentally, we care about the free energy $F_n=-\log Z_n$ because it is a measure of posterior concentration, and as we showed with the BIC calculation in DSLT1 [LW · GW], it tells us something about the information geometry of the posterior. In particular, given a compact neighbourhood of parameter space $W\subseteq W$ we can define its local free energy 

$$F_n\left(W\right)=-\log \left({\int }_W\phi \right(w\left)e^{-n{L}_n\left(w\right)}dw\right),$$

providing a direct tool for comparing different regions of the posterior, and thus different models. Since there is a correspondence

$$\text{Small}{F}_n\left(W\right)⟺\text{Large posterior concentration}{\int }_{W}p\left(w|D_n\right)dw,$$

we say the posterior prefers a region $W$ when it has low free energy relative to other regions of $W$. 

But this isn't the only reason to care about the free energy. In fact, it is explicitly related to generalisation, at least in the Bayesian sense.

In the frequentist framework that real-world deep learning takes place in (i.e. estimating a single parameter $\hat{w}$ using SGD), we typically split our dataset $D_n$ into a training set and a test set. We then say a model defined by $\hat{w}$ generalises well when it has low loss on the test set - in other words, it performs well on data that it hasn't seen before.

In the Bayesian paradigm, generalisation can be formulated according to a number of different estimation methods and quantities, depending on how you extract information from the posterior (e.g. you could estimate a maximum-likelihood parameter, a maximum a posterior estimate, or an average over samples from the posterior, etc.). Let's focus on one for the moment that involves the Bayes predictive distribution given by 

$$p(y|x,D_n)=E_w\left[p\right(y|x,w\left)\right]={\int }_{W}p(y|x,w)p\left(w|D_n\right)dw,$$

which weights the probability of an output $y$ given an input $x$ according to the posterior measure. The Bayes generalisation loss is then given by 

$$G_n=E_X[-\log p(y|x,D_n\left)\right]=-{\iint }_{R^{N+M}}q(y,x)\log p(y|x,D_n)dxdy.$$

Intuitively, it is the expected loss of the predictive distribution over all possible inputs $x$ and outputs $y$. It can be shown with relative ease ^[6] that the Bayes generalisation loss is equal to the average increase in free energy, 

$$G_n=E_{X_{n+1}}\left[F_{n+1}\right]-F_n.$$

In an informal-yet-conceptually-correct way, we can treat this difference as being the "derivative with respect to $n$" ^[7].

It follows immediately that the generalisation loss of a region $W\subseteq W$ is 

$$G_n\left(W\right)=E_{X_{n+1}}\left[F_{n+1}\right(W\left)\right]-F_n\left(W\right).$$

Ergo, to understand the information contained in the posterior, and which regions contain models with low generalisation error, we want to calculate the free energy. In even modestly simple settings this integral is intractable, thus why we need to calculate its asymptotic form. 

The Free Energy Formula for Singular Models

This subsection is a little bit more technical. If this overwhelms you, I recommend skipping ahead to the next subsection where I interpret the free energy formula. 

As I explained in DSLT1 [LW · GW], finding the asymptotic form of the free energy $F_n$ as $n\to \infty$ when $I\left(w\right)$ is degenerate is hard, and depends of theorems of algebraic geometry and distribution theory. This formula has been refined over the course of many papers ^[8], adjusted and generalised with various hypotheses. Here we will focus on the form given in [Wat13], which also applies to the unrealisable case where the set of true parameters $W_0$ may be empty. Thus we instead care about the set of optimal parameters 

$$W_{\text{opt}}=\{w\in W|L(w)=\underset{w^{\prime }\in W}{min}L(w^{\prime }\left)\right\},$$

and if $W_0$ is non-empty, then $W_0=W_{\text{opt}}$. As it stands, the free energy formula in the unrealisable case depends on a hypothesis of relatively finite variance  ^[9].

Watanabe shows that the free energy of $W$ asymptotically satisfies

$$F_n=n{L}_n\left(w^{\left(0\right)}\right)+\lambda \log n+U_n\sqrt{\frac{\lambda \log n}{2}}+O_p\left(1\right)\text{as}n\to \infty .$$

 where:

• $n$ is the size of the dataset $D_n$.
• $w^{\left(0\right)}\in W_{\text{opt}}$ is a most singular optimal point of $W$ (which we will explain below), which is in the interior of $W$. 
• $\lambda \in Q_{>0}$ is the global RLCT, associated to the most singular point $w^{\left(0\right)}$.
• $\left\{U_n\right\}$ is a sequence of random variables that satisfies $E\left[U_n\right]=0$ and converges to a Gaussian random variable in law as $n\to \infty$.
• $O_p\left(1\right)$ denotes a sequence that is bounded in probability.

We will interpret the formula in a moment, but let me briefly clarify what it means to call $w^{\left(0\right)}$ a "most singular point", a notion is made precise in [Lin11, Proposition 3.9]. (See the below figure, too). 

The gist is that every singularity $w\in W_{\text{opt}}$ has an associated local RLCT ${\lambda }_w$ defined by considering small neighbourhoods around the point. The global RLCT of $W$ is defined to be the minimum over these optimal points, $\lambda ={min}_{w\in W_{\text{opt}}}\left\{{\lambda }_w\right\}$, and an optimal point $w^{\left(0\right)}\in W_{\text{opt}}$ is a most singular point if ${\lambda }_{w^{\left(0\right)}}=\lambda$ ^[10]. Note also that the formula currently depends on an optimal parameter being in the interior of $W$, so you can think of a most singular point as being a local minimum of $K\left(w\right)$. 

The Widely Applicable Bayesian Information Criterion

In the asymptotic limit as $n\to \infty$, we can ignore the last two terms of the free energy formula and arrive at the Widely Applicable Bayesian Information Criterion (WBIC) across the full parameter space $W$,

$$WBIC=n{L}_n\left(w^{\left(0\right)}\right)+\lambda \log n.$$

Notice how the WBIC formula is the same as the BIC except that complexity is measured by $\lambda$ instead of $\frac{d}{2}$. But in DSLT1 [LW · GW], we explained how

• In regular models $\lambda =\frac{d}{2}$.
• In singular models $\lambda \le \frac{d}{2}$ in general. 

Thus the WBIC is a generalisation of the BIC, containing the regular result as a special case. 

Though the WBIC can be used to compare model classes like in the case of the BIC, its real power is what it tells us about the information geometry of the posterior within the same class of models. To this end, we can calculate the local free energy of a compact neighbourhood $W\subseteq W$ 

$$F_n\left(W\right)=n{L}_n\left(w_W^{\left(0\right)}\right)+{\lambda }_W\log n,$$

where $w_W^{\left(0\right)}\in W_{\text{opt}}$ is the most singular optimal point in $W$ with associated RLCT ${\lambda }_W$. 

The Accuracy-Complexity Tradeoff

The WBIC shows that the free energy of a region $W\subseteq W$ is comprised of the accuracy and complexity (or, to the physicists, energy and entropy) of the most singular point in $W$

$$F_n\left(W\right)=n\underset{\text{inaccuracy}}{\underset{}{L_n\left(w_W^{\left(0\right)}\right)}}+\underset{\text{complexity}}{\underset{}{{\lambda }_W}}\log n.$$

What makes this so profound is that in singular models, different regions $W\subset W$ can have different RLCT's ${\lambda }_W$, each with a different tradeoff between accuracy and complexity, unlike the regular model case where every region has a fixed complexity:

$$\begin{array}{cc}\text{Regular models}& :F_n\left(W\right)=n\left(\text{Inaccuracy}\right(W\left)\right)+\log n\left(\text{Constant Complexity}\right)\\ \text{Singular models}& :F_n\left(W\right)=n\left(\text{Inaccuracy}\right(W\left)\right)+\log n\left(\text{Complexity}\right(W\left)\right)\end{array}$$

So, the region in $W$ that minimises the free energy has the best accuracy-complexity tradeoff. This is the sense in which singular models obey Occam's Razor: if two regions are equally accurate, then they are preferred according to which is the simpler model. 

Interpreting the terms in the free energy formula leads us to three main points:

1. As $n\to \infty$, regions $W$ containing optimal parameters $w^{\left(0\right)}\in W_{\text{opt}}$ will always be preferred, since the inaccuracy is the leading order term.
2. As $n\to \infty$, if there are multiple regions $W_1$ and $W_2$ with equally optimal accuracy, then they are preferred according to their respective RLCT's. Lower is better, so if ${\lambda }_1<{\lambda }_2$ then $W_1$ is preferred by the posterior. 
3. For finite but large $n$, regions that do not contain a globally optimal parameter can be preferred by the posterior because they can have a better tradeoff between accuracy and complexity.

Why Singular Models (Can) Generalise Well

Armed with our free energy formula, we can now understand why singular models have the capacity to generalise well.

Recall that the Bayes generalisation error can be expressed as the difference $G_n=E_{X_{n+1}}\left[F_{n+1}\right]-F_n$, which can be interpreted as the "derivative" of the free energy with respect to $n$. Then since $F_n=n{L}_n\left(w^{\left(0\right)}\right)+\lambda \log n$, Watanabe is able to prove in [Wat18, Chapter 8] that, asymptotically, the Bayes generalisation error is

$$E\left[G_n\right]=L\left(w^{\left(0\right)}\right)+\frac{\lambda }{n}+o\left(\frac{1}{n}\right).$$

In fact, he goes one step further by considering two other forms of generalisation, the leave-one-out-cross-validation-loss $CV$, and the WAIC (an empirically measurable form of the WBIC), and shows that asymptotically 

$$E\left[G_n\right]\approx E\left[\text{CV}\right]\approx E\left[\text{WAIC}\right]=L\left(w^{\left(0\right)}\right)+\frac{\lambda }{n}+o\left(\frac{1}{n}\right).$$

Once again, we find that the RLCT $\lambda$ plays a central role in the learning process. Most importantly, we have a correspondence: 

$$\text{Optimal parameters}{w}^{\left(0\right)}\text{with low RLCT}\lambda ⟺\text{Low generalisation error}.$$

On top of this, we can carry out the same analysis on our local $F_n\left(W\right)$ to find that the local generalisation loss of $W$ is

$$E\left[G_n\right(W\left)\right]=L\left(w_W^{\left(0\right)}\right)+\frac{{\lambda }_W}{n}.$$

Since the RLCT can differ region to region, this tells us that:

$$\text{In singular models, different regions}W\text{have different generalisation error.}$$

All of this is to say: under any reasonable conception of Bayesian generalisation, the RLCT plays a central role in minimising generalisation loss, asymptotically. And since $\lambda \le \frac{d}{2}$ in singular models, the generalisation error of a singular model will always be better than that of a regular model with the same number of parameters. In DSLT3 we will show neural networks are singular, so:

 This is why neural networks can generalise well!

...sort of. Don't forget, we are in the Bayesian paradigm here, and it is not a given that Stochastic Gradient Descent (SGD) finds the same regions of parameter space that the Bayesian posterior says are "good". We postulate that in some sense they are equivalent, and the work of Mingard et al. in Is SGD a Bayesian Sampler? Well, Almost. agrees with this postulate. But, formalising this relationship does remain a key open problem in conclusively applying SLT to modern deep learning with SGD and its variants.

From points to local neighbourhoods

For those with less background knowledge, there is an important conceptual shift we have made here that I want to elaborate on briefly.

In frequentist statistics we care about particular point estimates $\hat{w}$ in parameter space $W$. But in Bayesian statistics, we care about measurable regions $W\subset W$ of parameter space, and the probability to which the posterior assigns those regions. This is a powerful shift in perspective, and points towards why SLT is placed in a Bayesian paradigm: the observation the geometry of $K\left(w\right)$ contains a lot more information than simple point estimates do lends itself naturally to Bayesian statistics. 

But in modern deep learning, we only ever have access to a point estimate $\hat{w}$ at the end of training via SGD. Sampling from the Bayesian posterior for large neural networks would not only be silly, but it would be, essentially, computationally impossible. So does this mean SLT has absolutely no applicability to modern deep learning? 

Not at all. By studying the local geometry of the loss landscape $K\left(w\right)$ - which is to say, arbitrarily small neighbourhoods $W$ of the posterior - we are able to analyse the set of points that are arbitrarily close to the singularities of $K\left(w\right)$. 

What Watanabe shows is that the singularities contained in these small neighbourhoods affect the geometry of the other points in the neighbourhood. If $W$ contains one true parameter $w^{\left(0\right)}\in W_0$, the other points in $W$ may not be equal minima of $K\left(w\right)$, but they are extremely close to being so. The same logic applies for the RLCT of a region: perhaps there is only one most singular point in $W$, but any nearby parameter within the small neighbourhood will define a model whose functional output is nearly identical to that of $f(x,w^{\left(0\right)})$ which has a lower complexity. 

In focusing only on points, we lose an extraordinary amount of information. So, we localise to neighbourhoods.

Intuitive Examples to Interpret the WBIC

It's time we looked at a specific example to build intuition about what the WBIC is telling us. 

I have constructed this toy example specifically to illustrate the main points here, and a lot of the details about the sub-leading terms in the free energy expansion are obfuscated, as well as the random fluctuations that make $K_n\left(w\right)$ different to $K\left(w\right)$. But, it is conceptually correct, and helps to illustrate the dominant features of the learning process as $n\to \infty$. Don't take it too literally, but do take it seriously. 

We will start with some calculations, and then visualise what this means for the posterior as $n\to \infty$.

Example 1: True parameters are preferred according to their RLCT

Let's consider a one parameter model $d=1$ with KL divergence defined by 

$$K\left(w\right)=(w+1{)}^2(w-1{)}^4,$$

on the region $W=[-2,2]$ with uniform prior $\phi \left(w\right)=\frac{1}{4}1(w\in W)$. There are two singularities in the set of true parameters, 

$$W_0=\{-1,1\},$$

which we will label as $w_{-1}^{\left(0\right)}$ and $w_1^{\left(0\right)}$ respectively. The Fisher information at true parameters is just the Hessian, which in the one dimensional case is simply the second derivative,

$$\begin{array}{c}I\left(w^{\left(0\right)}\right)=J\left(w^{\left(0\right)}\right)=\frac{d^{2}K}{d{w}^2}{|}_{w=w^{\left(0\right)}}.\end{array}$$

 An easy calculation shows 

$$\frac{d^{2}K}{d{w}^2}=2(w-1{)}^2(15w^2+10w-1),$$

 meaning $w_{-1}^{\left(0\right)}$ is a regular point and $w_1^{\left(0\right)}$ is a singular point since

$$\frac{d^{2}K}{d{w}^2}{|}_{w=-1}=32,\text{whereas}\frac{d^{2}K}{d{w}^2}{|}_{w=1}=0,$$

meaning the Fisher information "matrix" (albeit it is one-dimensional) is degenerate at $w_1^{\left(0\right)}$ but not at $w_{-1}^{\left(0\right)}$. We thus expect the RLCT of $w_1^{\left(0\right)}$ to be less than $\frac{d}{2}=\frac{1}{2}.$

Let $B(w_0,\delta )=[w_0-\delta ,w_0+\delta ]\subset W$ denote a small neighbourhood of radius $\delta >0$ centred at $w_0$. To analyse the geometry of $K\left(w\right)$ near the two singularities, we can define compact local regions 

$$W_{-1}=B(w_{-1}^{\left(0\right)},\delta )\text{and}{W}_1=B(w_1^{\left(0\right)},\delta ).$$

By taking a Taylor expansion about each singularity, the leading order terms of $K\left(w\right)$ in each region are 

$$\begin{array}{cc}{W}_{-1}& :K\left(w\right)\approx 16(w+1{)}^2,\\ \text{and in}{W}_1& :K\left(w\right)\approx 4(w-1{)}^4,\end{array}$$

since $\delta$ is very small. Recalling the definition of the RLCT from DSLT1 [LW · GW], since $K\left(w\right)$ is in normal crossing form we can read off the local RLCTs

$${\lambda }_{-1}=\frac{1}{2}\text{and}{\lambda }_1=\frac{1}{4},$$

meaning the effective dimensionalities associated to each singularity are $2{\lambda }_{-1}=1$ and $2{\lambda }_1=\frac{1}{2}$ respectively.

Here's the crux: since both singularities $w_{-1}^{\left(0\right)}$ and $w_1^{\left(0\right)}$ are true parameters, so $K\left(w_{-1}^{\left(0\right)}\right)=K\left(w_1^{\left(0\right)}\right)=0$, they both have the same accuracy $S$ (a constant), 

$$L\left(w_{-1}^{\left(0\right)}\right)=L\left(w_1^{\left(0\right)}\right)=S.$$

But, since the RLCT associated to $W_1$ is smaller, ${\lambda }_1<{\lambda }_{-1}$, the free energy formula tells us that $W_1$ is preferred by the posterior since it has lower free energy, 

$$F_n\left(W_1\right)<F_n\left(W_{-1}\right),$$

and our generalisation formula tells us that $W_1$ will have lower expected Bayesian generalisation loss, 

$$G_n\left(W_1\right)<G_n\left(W_{-1}\right).$$

The simpler model is preferred, and has a lower generalisation loss, because it has a lower RLCT.

Notice here how our $\delta$ was arbitrary. It doesn't matter exactly what it is, because as long as it is small, $K\left(w\right)$ will look like $16(w+1{)}^2$ and $4(w-1{)}^4$ in each respective region. Its geometry is dominated by these terms when very close to each singularity.

$K\left(w\right)$ is like a potential well, where a lower RLCT means a flatter floor

We argued in the first post that the RLCT is the correct measure of "flatness" in the loss landscape. We can now see that with our own eyes by plotting $K\left(w\right)$ in our above example:

Looking at this plot, a good intuition to have is that the loss landscape $K\left(w\right)$ is a potential well for some kinetic ball bouncing around, with its Hamiltonian given by $H\left(w\right)\approx nK\left(w\right)$. The ball, with enough random kinetic energy, will explore both regions $W_{-1}$ and $W_1$, but it will spend more time in the latter since it is "flatter" ^[11].

Animation 1: The posterior concentrates on both true parameters as $n\to \infty$, but it prefers the one with lower RLCT

For large $n$, the empirical KL divergence is approximately equal to the KL divergence $K_n\left(w\right)\approx K\left(w\right)$, and so normalising out by the constant $S$, the posterior effectively only depends on $K\left(w\right)$:

$$p\left(w|D_n\right)\approx \frac{\phi \left(w\right)e^{-nK\left(w\right)}}{{\int }_{-2}^2\phi \left(w\right)e^{-nK\left(w\right)}dw},$$

which is shown with some easy algebraic manipulations ^[12].

The free energy formula predicts that $W_1$ is preferred because ${\lambda }_1$ is smaller than ${\lambda }_{-1}$. So let's look at how the posterior changes with $n$:

As predicted, we visually see in the top posterior plot that the posterior concentration in $W_1$ is greater than $W_{-1}$ for all $n$. The bottom plot backs this up precisely by showing that the free energy $F_n\left(W_1\right)$ is always less than $F_n\left(W_{-1}\right)$. The RLCT clearly matters! 

Note here that we have been able to numerically calculate the integrals $Z_n\left(W_{-1}\right)$ and $Z_n\left(W_1\right)$, indicated by the shaded regions precisely, and not based on the free energy formula. This is because numerical integration is easy in low dimensions. In even modest dimensionality, these partition functions are notoriously difficult to compute, thus giving rise to the field of computational Bayesian statistics and algorithms like Markov Chain Monte Carlo. Our neural network experiments in DSLT4 require the MCMC approach. 

Example 2: Non-true parameters can be preferred at finite $n$ because of the accuracy-complexity tradeoff

Let's alter our example ever so slightly to the case where $W_1$ still has a lower RLCT, but it has marginally worse accuracy than $W_{-1}$ and thus $w_1^{\left(0\right)}$ is not a true parameter, but is an optimal parameter within $W_1$. We will set 

$$\begin{array}{cc}& L\left(w_{-1}^{\left(0\right)}\right)=S,\\ \text{and}& L\left(w_1^{\left(0\right)}\right)=S+C\end{array}$$

for some constant $C>0$. Our KL divergence is then given by 

$$K\left(w\right)=(w+1{)}^2((w-(1+h_C){)}^4-k_C)$$

where $h_C$ and $k_C$ are constants chosen such that $w_1^{\left(0\right)}=1$ is a local but not global minima of $K\left(w\right)$, so $K\left(1\right)=C$ and $K^{\prime }\left(1\right)=0$.

Then by the free energy formula ^[13], 

$$\begin{array}{cc}{F}_n\left(W_{-1}\right)& =nS+\frac{1}{2}\log n,\\ F_n\left(W_1\right)& =n(S+C)+\frac{1}{4}\log n.\end{array}$$

We can visualise how this changes the loss landscape:

Animation 2: At low $n$ the simpler model is preferred, but as $n\to \infty$, the true parameter is preferred

Let's visualise how the posterior concentration changes with $n$ for  $C=0.1$ and $\delta =0.5$ (yes, $\delta$ is reasonably large here, but you want to be able to see this process, right?).

For the first few values of $n$, the free energy is minimised by $W_1$ with its lower RLCT. But at about $n\approx 17$ the inaccuracy of $W_1$ has become too great to bare and the more accurate $W_{-1}$ is preferred. After this point the free energy curves are monotonic - the more accurate model will be preferred for all $n>17$.

Asymptotic Analysis Requires Care

I know what you're thinking: how come we could just approximate $K\left(w\right)$ in each region? Don't other curvature effects matter to the posterior density too? Like the prior, and those Taylor coefficients in $K\left(w\right)$? Wait, and don't we care about $K_n\left(w\right)$ instead of $K\left(w\right)$? Wait wait, $K_n\left(w\right)=L_n\left(w\right)-S_n$, but we never have access to $S_n$, so why would we care about $K_n\left(w\right)$, let alone $K\left(w\right)$? This all seems far too contrived...

Yes, and no. Remember, the free energy formula gives us an asymptotic approximation of the posterior density has $n\to \infty$. Thinking carefully about asymptotic approximations does require some care, particularly when one tries to convert these formulas into statements about finite $n$ - after all, we can only ever train a model on a finite number of datapoints. But Watanabe's free energy formula tells us what aspects of the geometry of $K\left(w\right)$ affect the learning process as more data is collected. He is able to rigorously prove that the structure of singularities, as measured by the RLCT, are central to the long-term behaviour of model training. Yes, many other effects get thrown away in the $O\left(1\right)$ term of the free energy formula, and the sequence $U_n$ is also loaded with other random fluctuations, but Watanabe nonetheless shows what makes singular models so powerful. 

Where are we going from here?

In this post, we have argued for the shift in perspective from points to local neighbourhoods (frequentist to Bayesian statistics), and then from local neighbourhoods to singularities of $K\left(w\right)$ (classical learning theory to singular learning theory). We have gone from caring about an enormous parameter space $W$, to only caring about certain singularities. Since singularities have different local geometry, and local neighbourhoods around them minimise the free energy, we have found our phases of statistical learning: 

$$\text{Phases in learning}⟺\text{Local regions}W\subset W\text{containing a singularity of interest}{w}^{\left(0\right)}.$$

We will explain this correspondence between phases and singularities in detail in DSLT4. Before that, though, in DSLT3 we are going to study the set of true parameters $W_0$ of a particular toy model: two layer feedforward ReLU neural networks with two inputs and one output. In doing so, we will see how the singularities of $K\left(w\right)$ in these models are identifiable with the different functionally equivalent symmetries of the neural network. With a full classification of $W_0$ in hand, we will then look at some experiments on these toy models that provide tractable and precise illustrations of phase transitions in the posterior in these ReLU neural networks, which Watanabe's free energy formula predicts.

References

[Wat09] S. Watanabe Algebraic Geometry and Statistical Learning Theory 2009

[Wat13] S. Watanabe A Widely Applicable Bayesian Information Criterion 2013

[Wat18] S. Watanabe Mathematical Theory of Bayesian Statistics [? · GW] 2018

[Lin11] S. Lin Algebraic Methods for Evaluating Integrals in Bayesian Statistics 2011

---

1. ^{^}

    In other words, the true distribution is $q\left(y|x\right)=N(x^2,1)$. ↩︎

2. ^{^}

   In other circumstances without a closed-form solution we can use an optimisation method like SGD. 

3. ^{^}

   Precisely, 

   $$L_n\left(w\right)=\frac{1}{n}\sum _{i=1}^n\frac{1}{2}\parallel y_i-f(x_i,w){\parallel }_M^2+\frac{M}{2}\log 2\pi ,$$

    where $M$ is the dimension of the output $y\in R^M$. ↩︎

4. ^{^}

   In the sense that if we kept drawing more data from the truth $q\left(y|x\right)$, the error of $f_{15}\left(x\right)$ would get worse and worse - it is overfitting data. ↩︎

5. ^{^}

   Okay, so this is a bit of a white lie. If the model and truth are both defined by a regression model with equal variance ${\sigma }^2=1$, then 

   $$K\left(w\right)=\frac{1}{2}{\int }_{R^N}\parallel f(x,w)-f(x,w_0){\parallel }_M^{2}q\left(x\right)dx.$$

   I haven't checked the case where the variances are different, but I believe it only requires a slight adjustment to this formula with a scaling factor somewhere. But this example nonetheless illustrates the point - $K\left(w\right)$ is a paraboloid in regular models.

6. ^{^}

   Lemma 3.5 of my thesis. ↩︎

7. ^{^}

   The reason this works is because of the asymptotics of $\log (n+1)-\log n$.

8. ^{^}

   Originally, in [Wat09], Watanabe proves the formula for the realisable case where $w^{\left(0\right)}\in W_0$ is a true parameter, meaning $K\left(w^{\left(0\right)}\right)=0$ and $L_n\left(w^{\left(0\right)}\right)=S_n$. 

9. ^{^}

   [Wat18, Def 7, Chap 3] For a given pair $w_0\in W_{\text{opt}}$ and $w\in W$, the log density ratio function is defined by 

   $$f(x,w_0,w)=\log \frac{p(y|x,w_0)}{p\left(x|w\right)}.$$

   If there exists $c_0>0$ such that for any arbitrary pair $w_0\in W_{\text{opt}}$ and $w\in W$ 

   $$E_X\left[f\right(X,w_0,w\left)\right]\ge c_0{E}_X\left[f\right(X,w_0,w{)}^2],$$

   then it is said that the log density ratio function $f(x,w_0,w)$ has a relatively finite variance. ↩︎

10. ^{^}

    And if it achieves the maximum multiplicity, but we will leave this vague to avoid more technical clutter. 

11. ^{^}

    In fact, this analogy to statistical physics runs quite deep. Hamiltonian Monte Carlo, one of the main methods for sampling from a Bayesian posterior, is essentially just simulating a particle subject to a Gibbs distribution $e^{-H\left(w\right)}$. There's a reason we've been calling it the free "energy", after all! ↩︎

12. ^{^}

    Specifically, as $n\to \infty$, the KL divergence converges, 

    $$L_n\left(w\right)-S_n=K_n\left(w\right)\to K\left(w\right)=L\left(w\right)-S.$$

     Since $S$ is a constant, we are able to write 

    $$\begin{array}{cc}p\left(w|D_n\right)& =\frac{\phi \left(w\right)e^{-n{L}_n\left(w\right)}}{{\int }_W\phi \left(w\right)e^{-n{L}_n\left(w\right)}dw}\\ & \approx \frac{\phi \left(w\right)e^{-nL\left(w\right)}}{{\int }_W\phi \left(w\right)e^{-nL\left(w\right)}dw}\frac{e^{nS}}{e^{nS}}\\ & =\frac{\phi \left(w\right)e^{-nK\left(w\right)}}{{\int }_W\phi \left(w\right)e^{-nK\left(w\right)}dw}.\end{array}$$

    If one defines the posterior in terms of $K_n\left(w\right)$, we call this the normalised posterior. 

13. ^{^}

    For this pathological example, these formulas are certainly not an adequate approximation for the exact numerical free energy at low $n$ - for one thing, they don't even depend on the radius of the ball $\delta$. However, writing the formulas like this does point us towards the conceptual idea, which is verified in the below animations. 

14 comments

Comments sorted by top scores.