Watanabe–Akaike information criterion explained

In statistics, the Widely Applicable Information Criterion (WAIC), also known as Watanabe–Akaike information criterion, is the generalized version of the Akaike information criterion (AIC) onto singular statistical models.[1] It is used as measure how well will model predict data it wasn't trained on. It is asymptotically equivalent to cross-validation loss.

If we take log pointwise predictive density:

lppd(y,\Theta)=\sumilog

1
S

\sumsp(yi\mid\Thetas)

Then:

WAIC(y,\Theta)=-2\left(lppd-\underbrace{\sumi\operatorname{Var}\thetalogp(yi\mid\theta)}penaltyterm\right)

Where

y

is predicted output in training data. Θ is models posterior distribution,

s

are samples from posterior, and i iterates over training data.

In other words, in Bayesian statistics the posterior is represented by list of samples from it. WAIC penalty is then the variance of predictions among these samples, calculated and added for each datapoint from dataset.[2]

The penalty term is often referred to as the "effective number of parameters". This terminology stems from historical conventions, as a similar term is used in the Akaike Information Criterion.

Watanabe recommends in practice calculating both WAIC and PSIS – Pareto Smoothed Importance Sampling. Both are approximations of leave-one-out cross-validation. If they disagree then at least one of them is not reliable. Similarly PSIS can sometimes detect if its estimate is not reliable (if

\hat{k}

> 0.7).[3]

Some textbooks of Bayesian statistics recommend WAIC over other information criteria, especially for multilevel and mixture models.[4]

Widely applicable Bayesian information criterion (WBIC) is the generalized version of Bayesian information criterion (BIC) onto singular statistical models.[5]

WBIC is the average log likelihood function over the posterior distribution with the inverse temperature > 1/log n where n is the sample size.

Both WAIC and WBIC can be numerically calculated without any information about a true distribution.

See also

Notes and References

  1. Sumio Watanabe . Sumio . Watanabe . 2010 . Asymptotic Equivalence of Bayes Cross Validation and Widely Applicable Information Criterion in Singular Learning Theory . Journal of Machine Learning Research . 11 . 3571–3594 .
  2. Book: McElreath . Richard . Richard McElreath . Statistical Rethinking : A Bayesian Course with Examples in R and Stan . Chapman and Hall/CRC . 2020 . 978-0-367-13991-9 . 2nd.
  3. Book: Watanabe, Sumio . Mathematical theory of Bayesian statistics . 2020 . CRC Press, Taylor & Francis Group . 978-1-4822-3806-8 . First issued in paperback . Boca Raton London New York.
  4. Book: Gelman . Andrew . Andrew Gelman . Bayesian Data Analysis . Carlin . John B. . John Carlin (professor) . Stern . Hal S. . Dunson . David B. . Vehtari . Aki . Rubin . Donald B. . Donald Rubin . Chapman and Hall/CRC . 2013 . 978-1-4398-4095-5 . Third.
  5. Sumio . Watanabe . 2013 . A Widely Applicable Bayesian Information Criterion . Journal of Machine Learning Research . 14 . 867–897 .