Multinomial distribution

From testwiki
Revision as of 12:39, 18 January 2025 by 31.222.81.237 (talk) (Use present, we are not summing.)
(diff) ← Older revision | Latest revision (diff) | Newer revision → (diff)
Jump to navigation Jump to search

In probability theory, the multinomial distribution is a generalization of the binomial distribution. For example, it models the probability of counts for each side of a k-sided die rolled n times. For n independent trials each of which leads to a success for exactly one of k categories, with each category having a given fixed success probability, the multinomial distribution gives the probability of any particular combination of numbers of successes for the various categories.

When k is 2 and n is 1, the multinomial distribution is the Bernoulli distribution. When k is 2 and n is bigger than 1, it is the binomial distribution. When k is bigger than 2 and n is 1, it is the categorical distribution. The term "multinoulli" is sometimes used for the categorical distribution to emphasize this four-way relationship (so n determines the suffix, and k the prefix).

The Bernoulli distribution models the outcome of a single Bernoulli trial. In other words, it models whether flipping a (possibly biased) coin one time will result in either a success (obtaining a head) or failure (obtaining a tail). The binomial distribution generalizes this to the number of heads from performing n independent flips (Bernoulli trials) of the same coin. The multinomial distribution models the outcome of n experiments, where the outcome of each trial has a categorical distribution, such as rolling a k-sided die n times.

Let k be a fixed finite number. Mathematically, we have k possible mutually exclusive outcomes, with corresponding probabilities p1, ..., pk, and n independent trials. Since the k outcomes are mutually exclusive and one must occur we have pi ≥ 0 for i = 1, ..., k and i=1kpi=1. Then if the random variables Xi indicate the number of times outcome number i is observed over the n trials, the vector X = (X1, ..., Xk) follows a multinomial distribution with parameters n and p, where p = (p1, ..., pk). While the trials are independent, their outcomes Xi are dependent because they must sum to n. Template:Short description Template:Probability distribution

Definitions

Probability mass function

Suppose one does an experiment of extracting n balls of k different colors from a bag, replacing the extracted balls after each draw. Balls of the same color are equivalent. Denote the variable which is the number of extracted balls of color i (i = 1, ..., k) as Xi, and denote as pi the probability that a given extraction will be in color i. The probability mass function of this multinomial distribution is:

f(x1,,xk;n,p1,,pk)=Pr(X1=x1 and  and Xk=xk)={n!x1!xk!p1x1××pkxk,when i=1kxi=n0otherwise,

for non-negative integers x1, ..., xk.

The probability mass function can be expressed using the gamma function as:

f(x1,,xk;p1,,pk)=Γ(ixi+1)iΓ(xi+1)i=1kpixi.

This form shows its resemblance to the Dirichlet distribution, which is its conjugate prior.

Example

Suppose that in a three-way election for a large country, candidate A received 20% of the votes, candidate B received 30% of the votes, and candidate C received 50% of the votes. If six voters are selected randomly, what is the probability that there will be exactly one supporter for candidate A, two supporters for candidate B and three supporters for candidate C in the sample?

Note: Since we’re assuming that the voting population is large, it is reasonable and permissible to think of the probabilities as unchanging once a voter is selected for the sample. Technically speaking this is sampling without replacement, so the correct distribution is the multivariate hypergeometric distribution, but the distributions converge as the population grows large in comparison to a fixed sample size[1].

Pr(A=1,B=2,C=3)=6!1!2!3!(0.21)(0.32)(0.53)=0.135

Properties

Normalization

The multinomial distribution is normalized according to:

j=1kxj=nf(x1,...,xk;n,p1,...,pk)=1

where the sum is over all permutations of xjsuch that j=1kxj=n.

Expected value and variance

The expected number of times the outcome i was observed over n trials is

E(Xi)=npi.

The covariance matrix is as follows. Each diagonal entry is the variance of a binomially distributed random variable, and is therefore

Var(Xi)=npi(1pi).

The off-diagonal entries are the covariances:

Cov(Xi,Xj)=npipj

for i, j distinct.

All covariances are negative because for fixed n, an increase in one component of a multinomial vector requires a decrease in another component.

When these expressions are combined into a matrix with i, j element cov(Xi,Xj), the result is a k × k positive-semidefinite covariance matrix of rank k − 1. In the special case where k = n and where the pi are all equal, the covariance matrix is the centering matrix.

The entries of the corresponding correlation matrix are

ρ(Xi,Xi)=1.
ρ(Xi,Xj)=Cov(Xi,Xj)Var(Xi)Var(Xj)=pipjpi(1pi)pj(1pj)=pipj(1pi)(1pj).

Note that the number of trials n drops out of this expression.

Each of the k components separately has a binomial distribution with parameters n and pi, for the appropriate value of the subscript i.

The support of the multinomial distribution is the set

{(n1,,nk)kn1++nk=n}.

Its number of elements is

(n+k1k1).

Matrix notation

In matrix notation,

E(𝐗)=n𝐩,

and

Var(𝐗)=n{diag(𝐩)𝐩𝐩T},

with Template:Math = the row vector transpose of the column vector Template:Math.

Visualization

As slices of generalized Pascal's triangle

Just like one can interpret the binomial distribution as (normalized) one-dimensional (1D) slices of Pascal's triangle, so too can one interpret the multinomial distribution as 2D (triangular) slices of Pascal's pyramid, or 3D/4D/+ (pyramid-shaped) slices of higher-dimensional analogs of Pascal's triangle. This reveals an interpretation of the range of the distribution: discretized equilateral "pyramids" in arbitrary dimension—i.e. a simplex with a grid.Template:Citation needed

As polynomial coefficients

Similarly, just like one can interpret the binomial distribution as the polynomial coefficients of (p+q)n when expanded, one can interpret the multinomial distribution as the coefficients of (p1+p2+p3++pk)n when expanded, noting that just the coefficients must sum up to 1.

Large deviation theory

Template:See also

Asymptotics

By Stirling's formula, at the limit of n,x1,...,xk, we haveln(nx1,,xk)+i=1kxilnpi=nDKL(p^p)k12ln(2πn)12i=1kln(p^i)+o(1)where relative frequencies p^i=xi/n in the data can be interpreted as probabilities from the empirical distribution p^, and DKL is the Kullback–Leibler divergence.

This formula can be interpreted as follows.

Consider Δk, the space of all possible distributions over the categories {1,2,...,k}. It is a simplex. After n independent samples from the categorical distribution p (which is how we construct the multinomial distribution), we obtain an empirical distribution p^.

By the asymptotic formula, the probability that empirical distribution p^ deviates from the actual distribution p decays exponentially, at a rate nDKL(p^p). The more experiments and the more different p^ is from p, the less likely it is to see such an empirical distribution.

If A is a closed subset of Δk, then by dividing up A into pieces, and reasoning about the growth rate of Pr(p^Aϵ) on each piece Aϵ, we obtain Sanov's theorem, which states thatlimn1nlnPr(p^A)=infp^ADKL(p^p)

Concentration at large n

Due to the exponential decay, at large n, almost all the probability mass is concentrated in a small neighborhood of p. In this small neighborhood, we can take the first nonzero term in the Taylor expansion of DKL, to obtainln(nx1,,xk)p1x1pkxkn2i=1k(p^ipi)2pi=12i=1k(xinpi)2npiThis resembles the gaussian distribution, which suggests the following theorem:

Theorem. At the n limit, ni=1k(p^ipi)2pi=i=1k(xinpi)2npi converges in distribution to the chi-squared distribution χ2(k1). File:Convergence of multinomial distribution to the gaussian distribution.webm

Template:Hidden begin

The space of all distributions over categories {1,2,,k} is a simplex: Δk={(y1,,yk):y1,,yk0,iyi=1}, and the set of all possible empirical distributions after n experiments is a subset of the simplex: Δk,n={(x1/n,,xk/n):x1,,xk,ixi=n}. That is, it is the intersection between Δk and the lattice (k)/n.

As n increases, most of the probability mass is concentrated in a subset of Δk,n near p, and the probability distribution near p becomes well-approximated by (nx1,,xk)p1x1pkxken2i(p^ipi)2piFrom this, we see that the subset upon which the mass is concentrated has radius on the order of 1/n, but the points in the subset are separated by distance on the order of 1/n, so at large n, the points merge into a continuum. To convert this from a discrete probability distribution to a continuous probability density, we need to multiply by the volume occupied by each point of Δk,n in Δk. However, by symmetry, every point occupies exactly the same volume (except a negligible set on the boundary), so we obtain a probability density ρ(p^)=Cen2i(p^ipi)2pi, where C is a constant.

Finally, since the simplex Δk is not all of k, but only within a (k1)-dimensional plane, we obtain the desired result.

Template:Hidden end

Conditional concentration at large n

The above concentration phenomenon can be easily generalized to the case where we condition upon linear constraints. This is the theoretical justification for Pearson's chi-squared test.

Theorem. Given frequencies xi observed in a dataset with n points, we impose +1 independent linear constraints {ip^i=1,ia1ip^i=b1,ia2ip^i=b2,,iaip^i=b(notice that the first constraint is simply the requirement that the empirical distributions sum to one), such that empirical p^i=xi/n satisfy all these constraints simultaneously. Let q denote the I-projection of prior distribution p on the sub-region of the simplex allowed by the linear constraints. At the n limit, sampled counts np^i from the multinomial distribution conditional on the linear constraints are governed by 2nDKL(p^||q)ni(p^iqi)2qi which converges in distribution to the chi-squared distribution χ2(k1).

Template:Hidden begin

An analogous proof applies in this Diophantine problem of coupled linear equations in count variables np^i,[2] but this time Δk,n is the intersection of (k)/n with Δk and hyperplanes, all linearly independent, so the probability density ρ(p^) is restricted to a (k1)-dimensional plane. In particular, expanding the KL divergence DKL(p^||p) around its minimum q (the I-projection of p on Δk,n) in the constrained problem ensures by the Pythagorean theorem for I-divergence that any constant and linear term in the counts np^i vanishes from the conditional probability to multinationally sample those counts.

Notice that by definition, every one of p^1,p^2,...,p^k must be a rational number, whereas p1,p2,...,pk may be chosen from any real number in [0,1] and need not satisfy the Diophantine system of equations. Only asymptotically as n, the p^i's can be regarded as probabilities over [0,1].

Template:Hidden end

Away from empirically observed constraints b1,,b (such as moments or prevalences) the theorem can be generalized:

Theorem.

  • Given functions f1,...,f, such that they are continuously differentiable in a neighborhood of p, and the vectors (1,1,...,1),f1(p),...,f(p) are linearly independent;
  • given sequences ϵ1(n),...,ϵ(n), such that asymptotically 1nϵi(n)1n for each i{1,...,};
  • then for the multinomial distribution conditional on constraints f1(p^)[f1(p)ϵ1(n),f1(p)+ϵ1(n)],...,f(p^)[f(p)ϵ(n),f(p)+ϵ(n)], we have the quantity ni(p^ipi)2pi=i(xinpi)2npi converging in distribution to χ2(k1) at the n limit.

In the case that all p^i are equal, the Theorem reduces to the concentration of entropies around the Maximum Entropy.[3][4]

In some fields such as natural language processing, categorical and multinomial distributions are synonymous and it is common to speak of a multinomial distribution when a categorical distribution is actually meant. This stems from the fact that it is sometimes convenient to express the outcome of a categorical distribution as a "1-of-k" vector (a vector with one element containing a 1 and all other elements containing a 0) rather than as an integer in the range 1k; in this form, a categorical distribution is equivalent to a multinomial distribution over a single trial.

Statistical inference

Template:Expand section

Equivalence tests for multinomial distributions

The goal of equivalence testing is to establish the agreement between a theoretical multinomial distribution and observed counting frequencies. The theoretical distribution may be a fully specified multinomial distribution or a parametric family of multinomial distributions.

Let q denote a theoretical multinomial distribution and let p be a true underlying distribution. The distributions p and q are considered equivalent if d(p,q)<ε for a distance d and a tolerance parameter ε>0. The equivalence test problem is H0={d(p,q)ε} versus H1={d(p,q)<ε}. The true underlying distribution p is unknown. Instead, the counting frequencies pn are observed, where n is a sample size. An equivalence test uses pn to reject H0. If H0 can be rejected then the equivalence between p and q is shown at a given significance level. The equivalence test for Euclidean distance can be found in text book of Wellek (2010).[5] The equivalence test for the total variation distance is developed in Ostrovski (2017).[6] The exact equivalence test for the specific cumulative distance is proposed in Frey (2009).[7]

The distance between the true underlying distribution p and a family of the multinomial distributions is defined by d(p,)=minhd(p,h). Then the equivalence test problem is given by H0={d(p,)ε} and H1={d(p,)<ε}. The distance d(p,) is usually computed using numerical optimization. The tests for this case are developed recently in Ostrovski (2018).[8]

Confidence intervals for the difference of two proportions

In the setting of a multinomial distribution, constructing confidence intervals for the difference between the proportions of observations from two events, pipj, requires the incorporation of the negative covariance between the sample estimators p^i=Xin and p^j=Xjn.

Some of the literature on the subject focused on the use-case of matched-pairs binary data, which requires careful attention when translating the formulas to the general case of pipj for any multinomial distribution. Formulas in the current section will be generalized, while formulas in the next section will focus on the matched-pairs binary data use-case.

Wald's standard error (SE) of the difference of proportion can be estimated using:[9]Template:Rp[10]

SE(p^ip^j)^=(p^i+p^j)(p^ip^j)2n

For a 100(1α)% approximate confidence interval, the margin of error may incorporate the appropriate quantile from the standard normal distribution, as follows:

(p^ip^j)±zα/2SE(p^ip^j)^

Template:Hidden begin As the sample size (n) increases, the sample proportions will approximately follow a multivariate normal distribution, thanks to the multidimensional central limit theorem (and it could also be shown using the Cramér–Wold theorem). Therefore, their difference will also be approximately normal. Also, these estimators are weakly consistent and plugging them into the SE estimator makes it also weakly consistent. Hence, thanks to Slutsky's theorem, the pivotal quantity (p^ip^j)(pipj)SE(p^ip^j)^ approximately follows the standard normal distribution. And from that, the above approximate confidence interval is directly derived.

The SE can be constructed using the calculus of the variance of the difference of two random variables: SE(p^ip^j)^=p^i(1p^i)n+p^j(1p^j)n2(p^ip^jn)=1n(p^i+p^jp^i2p^j2+2p^ip^j)=(p^i+p^j)(p^ip^j)2n

Template:Hidden end

A modification which includes a continuity correction adds 1n to the margin of error as follows:[11]Template:Rp

(p^ip^j)±(zα/2SE(p^ip^j)^+1n)

Another alternative is to rely on a Bayesian estimator using Jeffreys prior which leads to using a dirichlet distribution, with all parameters being equal to 0.5, as a prior. The posterior will be the calculations from above, but after adding 1/2 to each of the k elements, leading to an overall increase of the sample size by k2. This was originally developed for a multinomial distribution with four events, and is known as wald+2, for analyzing matched pairs data (see the next section for more details).[12]

This leads to the following SE:

SE(p^ip^j)^wald+k2=(p^i+p^j+1n)nn+k2(p^ip^j)2(nn+k2)2n+k2

Template:Hidden begin SE(p^ip^j)^wald+k2=(xi+1/2n+k2+xj+1/2n+k2)(xi+1/2n+k2xj+1/2n+k2)2n+k2=(xin+xjn+1n)nn+k2(xinxjn)2(nn+k2)2n+k2=(p^i+p^j+1n)nn+k2(p^ip^j)2(nn+k2)2n+k2 Template:Hidden end

Which can just be plugged into the original Wald formula as follows:

(pipj)nn+k2±zα/2SE(p^ip^j)^wald+k2

Occurrence and applications

Confidence intervals for the difference in matched-pairs binary data (using multinomial with k=4)

For the case of matched-pairs binary data, a common task is to build the confidence interval of the difference of the proportion of the matched events. For example, we might have a test for some disease, and we may want to check the results of it for some population at two points in time (1 and 2), to check if there was a change in the proportion of the positives for the disease during that time.

Such scenarios can be represented using a two-by-two contingency table with the number of elements that had each of the combination of events. We can use small f for sampling frequencies: f11,f10,f01,f00, and capital F for population frequencies: F11,F10,F01,F00. These four combinations could be modeled as coming from a multinomial distribution (with four potential outcomes). The sizes of the sample and population can be n and N respectively. And in such a case, there is an interest in building a confidence interval for the difference of proportions from the marginals of the following (sampled) contingency table:

Test 2 positive Test 2 negative Row total
Test 1 positive f11 f10 f1*=f11+f10
Test 1 negative f01 f00 f0*=f01+f00
Column total f*1=f11+f01 f*0=f10+f00 n

In this case, checking the difference in marginal proportions means we are interested in using the following definitions: p1*=F1*N=F11+F10N, p*1=F*1N=F11+F01N. And the difference we want to build confidence intervals for is:

p*1p1*=F11+F01NF11+F10N=F01NF10N=p01p10

Hence, a confidence intervals for the marginal positive proportions (p*1p1*) is the same as building a confidence interval for the difference of the proportions from the secondary diagonal of the two-by-two contingency table (p01p10).

Calculating a p-value for such a difference is known as McNemar's test. Building confidence interval around it can be constructed using methods described above for Confidence intervals for the difference of two proportions.

The Wald confidence intervals from the previous section can be applied to this setting, and appears in the literature using alternative notations. Specifically, the SE often presented is based on the contingency table frequencies instead of the sample proportions. For example, the Wald confidence intervals, provided above, can be written as:[11]Template:Rp

SE(p*1p1*)^=SE(p01p10)^=n(f10+f01)(f10f01)2nn

Further research in the literature has identified several shortcomings in both the Wald and the Wald with continuity correction methods, and other methods have been proposed for practical application.[11]

One such modification includes Agresti and Min’s Wald+2 (similar to some of their other works[13]) in which each cell frequency had an extra 12 added to it.[12] This leads to the Wald+2 confidence intervals. In a Bayesian interpretation, this is like building the estimators taking as prior a dirichlet distribution with all parameters being equal to 0.5 (which is, in fact, the Jeffreys prior). The +2 in the name wald+2 can now be taken to mean that in the context of a two-by-two contingency table, which is a multinomial distribution with four possible events, then since we add 1/2 an observation to each of them, then this translates to an overall addition of 2 observations (due to the prior).

This leads to the following modified SE for the case of matched pairs data:

SE(p*1p1*)^=(n+2)(f10+f01+1)(f10f01)2(n+2)(n+2)

Which can just be plugged into the original Wald formula as follows:

(p*1p1*)nn+2±zα/2SE(p^ip^j)^wald+2

Other modifications include Bonett and Price’s Adjusted Wald, and Newcombe’s Score.

Computational methods

Random variate generation

Template:Further

First, reorder the parameters p1,,pk such that they are sorted in descending order (this is only to speed up computation and not strictly necessary). Now, for each trial, draw an auxiliary variable X from a uniform (0, 1) distribution. The resulting outcome is the component

j=min{j{1,,k}:(i=1jpi)X0}.

{Xj = 1, Xk = 0 for k ≠ j } is one observation from the multinomial distribution with p1,,pk and n = 1. A sum of independent repetitions of this experiment is an observation from a multinomial distribution with n equal to the number of such repetitions.

Sampling using repeated conditional binomial samples

Given the parameters p1,p2,,pk and a total for the sample n such that i=1kXi=n, it is possible to sample sequentially for the number in an arbitrary state Xi, by partitioning the state space into i and not-i, conditioned on any prior samples already taken, repeatedly.

Algorithm: Sequential conditional binomial sampling

S = n
rho = 1
for i in [1,k-1]:
    if rho != 0:
        X[i] ~ Binom(S,p[i]/rho)
    else 
        X[i] = 0
    S = S - X[i]
    rho = rho - p[i]
X[k] = S

Heuristically, each application of the binomial sample reduces the available number to sample from and the conditional probabilities are likewise updated to ensure logical consistency.[14]

Software implementations

  • The MultinomialCI R package allows the computation of simultaneous confidence intervals for the probabilities of a multinomial distribution given a set of observations.[15]

See also

References

Template:Reflist

Further reading

Template:Refbegin

Template:Refend

Template:Clear Template:ProbDistributions Template:Authority control