Akaike information criterion explained

The Akaike information criterion (AIC) is an estimator of prediction error and thereby relative quality of statistical models for a given set of data.[1] [2] Given a collection of models for the data, AIC estimates the quality of each model, relative to each of the other models. Thus, AIC provides a means for model selection.

AIC is founded on information theory. When a statistical model is used to represent the process that generated the data, the representation will almost never be exact; so some information will be lost by using the model to represent the process. AIC estimates the relative amount of information lost by a given model: the less information a model loses, the higher the quality of that model.

In estimating the amount of information lost by a model, AIC deals with the trade-off between the goodness of fit of the model and the simplicity of the model. In other words, AIC deals with both the risk of overfitting and the risk of underfitting.

The Akaike information criterion is named after the Japanese statistician Hirotsugu Akaike, who formulated it. It now forms the basis of a paradigm for the foundations of statistics and is also widely used for statistical inference.

Definition

Suppose that we have a statistical model of some data. Let be the number of estimated parameters in the model. Let

\hatL

be the maximized value of the likelihood function for the model. Then the AIC value of the model is the following.

AIC=2k-2ln(\hatL)

Given a set of candidate models for the data, the preferred model is the one with the minimum AIC value. Thus, AIC rewards goodness of fit (as assessed by the likelihood function), but it also includes a penalty that is an increasing function of the number of estimated parameters. The penalty discourages overfitting, which is desired because increasing the number of parameters in the model almost always improves the goodness of the fit.

AIC is founded in information theory. Suppose that the data is generated by some unknown process f. We consider two candidate models to represent f: g1 and g2. If we knew f, then we could find the information lost from using g1 to represent f by calculating the Kullback–Leibler divergence, ; similarly, the information lost from using g2 to represent f could be found by calculating . We would then, generally, choose the candidate model that minimized the information loss.

We cannot choose with certainty, because we do not know f. showed, however, that we can estimate, via AIC, how much more (or less) information is lost by g1 than by g2. The estimate, though, is only valid asymptotically; if the number of data points is small, then some correction is often necessary (see AICc, below).

Note that AIC tells nothing about the absolute quality of a model, only the quality relative to other models. Thus, if all the candidate models fit poorly, AIC will not give any warning of that. Hence, after selecting a model via AIC, it is usually good practice to validate the absolute quality of the model. Such validation commonly includes checks of the model's residuals (to determine whether the residuals seem like random) and tests of the model's predictions. For more on this topic, see statistical model validation.

How to use AIC in practice

To apply AIC in practice, we start with a set of candidate models, and then find the models' corresponding AIC values. There will almost always be information lost due to using a candidate model to represent the "true model," i.e. the process that generated the data. We wish to select, from among the candidate models, the model that minimizes the information loss. We cannot choose with certainty, but we can minimize the estimated information loss.

Suppose that there are R candidate models. Denote the AIC values of those models by AIC1, AIC2, AIC3, ..., AICR. Let AICmin be the minimum of those values. Then the quantity exp((AICmin − AICi)/2) can be interpreted as being proportional to the probability that the ith model minimizes the (estimated) information loss.

As an example, suppose that there are three candidate models, whose AIC values are 100, 102, and 110. Then the second model is times as probable as the first model to minimize the information loss. Similarly, the third model is times as probable as the first model to minimize the information loss.

In this example, we would omit the third model from further consideration. We then have three options: (1) gather more data, in the hope that this will allow clearly distinguishing between the first two models; (2) simply conclude that the data is insufficient to support selecting one model from among the first two; (3) take a weighted average of the first two models, with weights proportional to 1 and 0.368, respectively, and then do statistical inference based on the weighted multimodel.

The quantity is known as the relative likelihood of model i. It is closely related to the likelihood ratio used in the likelihood-ratio test. Indeed, if all the models in the candidate set have the same number of parameters, then using AIC might at first appear to be very similar to using the likelihood-ratio test. There are, however, important distinctions. In particular, the likelihood-ratio test is valid only for nested models, whereas AIC (and AICc) has no such restriction.

Hypothesis testing

Every statistical hypothesis test can be formulated as a comparison of statistical models. Hence, every statistical hypothesis test can be replicated via AIC. Two examples are briefly described in the subsections below. Details for those examples, and many more examples, are given by and .

Replicating Student's t-test

As an example of a hypothesis test, consider the t-test to compare the means of two normally-distributed populations. The input to the t-test comprises a random sample from each of the two populations.

To formulate the test as a comparison of models, we construct two different models. The first model models the two populations as having potentially different means and standard deviations. The likelihood function for the first model is thus the product of the likelihoods for two distinct normal distributions; so it has four parameters: . To be explicit, the likelihood function is as follows (denoting the sample sizes by and).

l{L}(\mu1,\sigma1,\mu2,\sigma2)=

n1
\prod
i=1
1
\sqrt{2\pi

\sigma1}\exp\left(-

(x
2
1)
i-\mu
2
2\sigma
1

\right)\boldsymbol

n1+n2
\prod
i=n1+1
1
\sqrt{2\pi

\sigma2}\exp\left(-

(x
2
2)
i-\mu
2
2\sigma
2

\right)

The second model models the two populations as having the same means but potentially different standard deviations. The likelihood function for the second model thus sets in the above equation; so it has three parameters.

We then maximize the likelihood functions for the two models (in practice, we maximize the log-likelihood functions); after that, it is easy to calculate the AIC values of the models. We next calculate the relative likelihood. For instance, if the second model was only 0.01 times as likely as the first model, then we would omit the second model from further consideration: so we would conclude that the two populations have different means.

The t-test assumes that the two populations have identical standard deviations; the test tends to be unreliable if the assumption is false and the sizes of the two samples are very different (Welch's t-test would be better). Comparing the means of the populations via AIC, as in the example above, has an advantage by not making such assumptions.

Comparing categorical data sets

For another example of a hypothesis test, suppose that we have two populations, and each member of each population is in one of two categories—category #1 or category #2. Each population is binomially distributed. We want to know whether the distributions of the two populations are the same. We are given a random sample from each of the two populations.

Let be the size of the sample from the first population. Let be the number of observations (in the sample) in category #1; so the number of observations in category #2 is . Similarly, let be the size of the sample from the second population. Let be the number of observations (in the sample) in category #1.

Let be the probability that a randomly-chosen member of the first population is in category #1. Hence, the probability that a randomly-chosen member of the first population is in category #2 is . Note that the distribution of the first population has one parameter. Let be the probability that a randomly-chosen member of the second population is in category #1. Note that the distribution of the second population also has one parameter.

To compare the distributions of the two populations, we construct two different models. The first model models the two populations as having potentially different distributions. The likelihood function for the first model is thus the product of the likelihoods for two distinct binomial distributions; so it has two parameters:, . To be explicit, the likelihood function is as follows.

l{L}(p,q)=

m!
m1!(m-m1)!
m1
p
m-m1
(1-p)

\boldsymbol

n!
n1!(n-n1)!
n1
q
n-n1
(1-q)

The second model models the two populations as having the same distribution. The likelihood function for the second model thus sets in the above equation; so the second model has one parameter.

We then maximize the likelihood functions for the two models (in practice, we maximize the log-likelihood functions); after that, it is easy to calculate the AIC values of the models. We next calculate the relative likelihood. For instance, if the second model was only 0.01 times as likely as the first model, then we would omit the second model from further consideration: so we would conclude that the two populations have different distributions.

Foundations of statistics

Statistical inference is generally regarded as comprising hypothesis testing and estimation. Hypothesis testing can be done via AIC, as discussed above. Regarding estimation, there are two types: point estimation and interval estimation. Point estimation can be done within the AIC paradigm: it is provided by maximum likelihood estimation. Interval estimation can also be done within the AIC paradigm: it is provided by likelihood intervals. Hence, statistical inference generally can be done within the AIC paradigm.

The most commonly used paradigms for statistical inference are frequentist inference and Bayesian inference. AIC, though, can be used to do statistical inference without relying on either the frequentist paradigm or the Bayesian paradigm: because AIC can be interpreted without the aid of significance levels or Bayesian priors. In other words, AIC can be used to form a foundation of statistics that is distinct from both frequentism and Bayesianism.

Modification for small sample size

When the sample size is small, there is a substantial probability that AIC will select models that have too many parameters, i.e. that AIC will overfit. To address such potential overfitting, AICc was developed: AICc is AIC with a correction for small sample sizes.

The formula for AICc depends upon the statistical model. Assuming that the model is univariate, is linear in its parameters, and has normally-distributed residuals (conditional upon regressors), then the formula for AICc is as follows.

AICc=AIC+

2k2+2k
n-k-1

—where denotes the sample size and denotes the number of parameters. Thus, AICc is essentially AIC with an extra penalty term for the number of parameters. Note that as, the extra penalty term converges to 0, and thus AICc converges to AIC.

If the assumption that the model is univariate and linear with normal residuals does not hold, then the formula for AICc will generally be different from the formula above. For some models, the formula can be difficult to determine. For every model that has AICc available, though, the formula for AICc is given by AIC plus terms that includes both and 2. In comparison, the formula for AIC includes but not 2. In other words, AIC is a first-order estimate (of the information loss), whereas AICc is a second-order estimate.

Further discussion of the formula, with examples of other assumptions, is given by and by . In particular, with other assumptions, bootstrap estimation of the formula is often feasible.

To summarize, AICc has the advantage of tending to be more accurate than AIC (especially for small samples), but AICc also has the disadvantage of sometimes being much more difficult to compute than AIC. Note that if all the candidate models have the same and the same formula for AICc, then AICc and AIC will give identical (relative) valuations; hence, there will be no disadvantage in using AIC, instead of AICc. Furthermore, if is many times larger than 2, then the extra penalty term will be negligible; hence, the disadvantage in using AIC, instead of AICc, will be negligible.

History

The Akaike information criterion was formulated by the statistician Hirotsugu Akaike. It was originally named "an information criterion". It was first announced in English by Akaike at a 1971 symposium; the proceedings of the symposium were published in 1973. The 1973 publication, though, was only an informal presentation of the concepts. The first formal publication was a 1974 paper by Akaike.

The initial derivation of AIC relied upon some strong assumptions. showed that the assumptions could be made much weaker. Takeuchi's work, however, was in Japanese and was not widely known outside Japan for many years. (Translated in)

AICc was originally proposed for linear regression (only) by . That instigated the work of, and several further papers by the same authors, which extended the situations in which AICc could be applied.

The first general exposition of the information-theoretic approach was the volume by . It includes an English presentation of the work of Takeuchi. The volume led to far greater use of AIC, and it now has more than 64,000 citations on Google Scholar.

Akaike called his approach an "entropy maximization principle", because the approach is founded on the concept of entropy in information theory. Indeed, minimizing AIC in a statistical model is effectively equivalent to maximizing entropy in a thermodynamic system; in other words, the information-theoretic approach in statistics is essentially applying the Second Law of Thermodynamics. As such, AIC has roots in the work of Ludwig Boltzmann on entropy. For more on these issues, see and .

Usage tips

Counting parameters

A statistical model must account for random errors. A straight line model might be formally described as yi = b0 + b1xi + εi. Here, the εi are the residuals from the straight line fit. If the εi are assumed to be i.i.d. Gaussian (with zero mean), then the model has three parameters:b0, b1, and the variance of the Gaussian distributions.Thus, when calculating the AIC value of this model, we should use k=3. More generally, for any least squares model with i.i.d. Gaussian residuals, the variance of the residuals' distributions should be counted as one of the parameters.

As another example, consider a first-order autoregressive model, defined byxi = c + φxi−1 + εi, with the εi being i.i.d. Gaussian (with zero mean). For this model, there are three parameters: c, φ, and the variance of the εi. More generally, a pth-order autoregressive model has parameters. (If, however, c is not estimated from the data, but instead given in advance, then there are only parameters.)

Transforming data

The AIC values of the candidate models must all be computed with the same data set. Sometimes, though, we might want to compare a model of the response variable,, with a model of the logarithm of the response variable, . More generally, we might want to compare a model of the data with a model of transformed data. Following is an illustration of how to deal with data transforms (adapted from : "Investigators should be sure that all hypotheses are modeled using the same response variable").

Suppose that we want to compare two models: one with a normal distribution of and one with a normal distribution of . We should not directly compare the AIC values of the two models. Instead, we should transform the normal cumulative distribution function to first take the logarithm of . To do that, we need to perform the relevant integration by substitution: thus, we need to multiply by the derivative of the (natural) logarithm function, which is . Hence, the transformed distribution has the following probability density function:

y\mapsto

1
y
1
\sqrt{2\pi\sigma2
}\,\exp \left(-\frac\right)—which is the probability density function for the log-normal distribution. We then compare the AIC value of the normal model against the AIC value of the log-normal model.

For misspecified model, Takeuchi's Information Criterion (TIC) might be more appropriate. However, TIC often suffers from instability caused by estimation errors.[3]

Comparisons with other model selection methods

The critical difference between AIC and BIC (and their variants) is the asymptotic property under well-specified and misspecified model classes.[4] Their fundamental differences have been well-studied in regression variable selection and autoregression order selection[5] problems. In general, if the goal is prediction, AIC and leave-one-out cross-validations are preferred. If the goal is selection, inference, or interpretation, BIC or leave-many-out cross-validations are preferred. A comprehensive overview of AIC and other popular model selection methods is given by Ding et al. (2018)[6]

Comparison with BIC

The formula for the Bayesian information criterion (BIC) is similar to the formula for AIC, but with a different penalty for the number of parameters. With AIC the penalty is, whereas with BIC the penalty is .

A comparison of AIC/AICc and BIC is given by, with follow-up remarks by . The authors show that AIC/AICc can be derived in the same Bayesian framework as BIC, just by using different prior probabilities. In the Bayesian derivation of BIC, though, each candidate model has a prior probability of 1/R (where R is the number of candidate models). Additionally, the authors present a few simulation studies that suggest AICc tends to have practical/performance advantages over BIC.

A point made by several researchers is that AIC and BIC are appropriate for different tasks. In particular, BIC is argued to be appropriate for selecting the "true model" (i.e. the process that generated the data) from the set of candidate models, whereas AIC is not appropriate. To be specific, if the "true model" is in the set of candidates, then BIC will select the "true model" with probability 1, as ; in contrast, when selection is done via AIC, the probability can be less than 1. Proponents of AIC argue that this issue is negligible, because the "true model" is virtually never in the candidate set. Indeed, it is a common aphorism in statistics that "all models are wrong"; hence the "true model" (i.e. reality) cannot be in the candidate set.

Another comparison of AIC and BIC is given by . Vrieze presents a simulation study—which allows the "true model" to be in the candidate set (unlike with virtually all real data). The simulation study demonstrates, in particular, that AIC sometimes selects a much better model than BIC even when the "true model" is in the candidate set. The reason is that, for finite, BIC can have a substantial risk of selecting a very bad model from the candidate set. This reason can arise even when is much larger than 2. With AIC, the risk of selecting a very bad model is minimized.

If the "true model" is not in the candidate set, then the most that we can hope to do is select the model that best approximates the "true model". AIC is appropriate for finding the best approximating model, under certain assumptions. (Those assumptions include, in particular, that the approximating is done with regard to information loss.)

Comparison of AIC and BIC in the context of regression is given by . In regression, AIC is asymptotically optimal for selecting the model with the least mean squared error, under the assumption that the "true model" is not in the candidate set. BIC is not asymptotically optimal under the assumption. Yang additionally shows that the rate at which AIC converges to the optimum is, in a certain sense, the best possible.

Comparison with least squares

Sometimes, each candidate model assumes that the residuals are distributed according to independent identical normal distributions (with zero mean). That gives rise to least squares model fitting.

With least squares fitting, the maximum likelihood estimate for the variance of a model's residuals distributions is

\hat\sigma2=RSS/n

,where the residual sum of squares is

styleRSS=

n
\sum
i=1

(yi-f(x

2
i;\hat{\theta}))
Then, the maximum value of a model's log-likelihood function is (see Normal distribution#Log-likelihood):

ln(\hatL)= -

n
2

ln(2\pi)-

n
2

ln(\hat\sigma2)-

1
2\hat\sigma2

RSS=-

n
2

ln(\hat\sigma2)+C

where is a constant independent of the model, and dependent only on the particular data points, i.e. it does not change if the data does not change.

That gives:

AIC=2k-2ln(\hatL)=2k+nln(\hat\sigma2)-2C

Because only differences in AIC are meaningful, the constant can be ignored, which allows us to conveniently take the following for model comparisons:

\DeltaAIC=2k+nln(\hat\sigma2)

Note that if all the models have the same, then selecting the model with minimum AIC is equivalent to selecting the model with minimum —which is the usual objective of model selection based on least squares.

Comparison with cross-validation

Leave-one-out cross-validation is asymptotically equivalent to AIC, for ordinary linear regression models. Asymptotic equivalence to AIC also holds for mixed-effects models.

Comparison with Mallows's Cp

Mallows's Cp is equivalent to AIC in the case of (Gaussian) linear regression.

See also

References

Further reading

Notes and References

  1. Book: McElreath, Richard . Richard McElreath . Statistical Rethinking: A Bayesian Course with Examples in R and Stan . CRC Press . 2016 . 978-1-4822-5344-3 . 189 . AIC provides a surprisingly simple estimate of the average out-of-sample deviance. .
  2. Book: Taddy, Matt . Business Data Science: Combining Machine Learning and Economics to Optimize, Automate, and Accelerate Business Decisions . 90 . New York . McGraw-Hill . 2019 . 978-1-260-45277-8 . The AIC is an estimate for OOS deviance. .
  3. Matsuda . Takeru . Uehara . Masatoshi . Hyvarinen . Aapo . 2021 . Information criteria for non-normalized models . Journal of Machine Learning Research . 22 . 158 . 1–33 . 1533-7928.
  4. Ding. Jie. Tarokh. Vahid. Yang. Yuhong. November 2018. Model Selection Techniques: An Overview. IEEE Signal Processing Magazine. 35. 6. 16–34. 10.1109/MSP.2018.2867638. 1810.09583. 2018ISPM...35f..16D. 53035396. 1053-5888.
  5. Ding. J.. Tarokh. V.. Yang. Y.. June 2018. Bridging AIC and BIC: A New Criterion for Autoregression. IEEE Transactions on Information Theory. 64. 6. 4024–4043. 10.1109/TIT.2017.2717599. 1557-9654. 1508.02473. 5189440.
  6. Model Selection Techniques: An Overview . IEEE Signal Processing Magazine . 2018-11-14 . 10.1109/MSP.2018.2867638 . 2023-02-18 . Ding . Jie . Tarokh . Vahid . Yang . Yuhong . 35 . 6 . 16–34 . 1810.09583 . 2018ISPM...35f..16D . 53035396 .