Models for count data

May 15, 2015 — August 3, 2021

branching
count data
regression
statistics

I have data and/or predictions made up of non-negative integers \({\mathbb N }\cup\{0\}\). What probability distributions can I use to model it?

All the distributions I discuss here have support unbounded above. Bounded distributions (e.g. vanilla Binomial) are for some other time. The exception is the Bernoulli RV, a.k.a. the biased coin, which is simple enough to sneak in.

For some related ideas in a time series context, see

Should I also raid the document topic model literature for this. Surely they are implicitly count data? See string bags compare with Steyvers and Tenenbaum’s semantic network model (Steyvers and Tenenbaum 2005).

Much of this material is in (Johnson, Kemp, and Kotz 2005), if you’d like a convenient 1-stop reference.

Figure 1

1 Poisson

The Poisson is reminiscent of the Gaussian for count data, in terms of the number of places that it pops up, and the vast number of things that have a limiting Poisson distribution.

Conveniently, it has only one parameter, which means you don’t need to justify how you chose any other parameters, saving valuable time and thought. It’s useful as a “null model”, in that the number of particles in realisation of a point process without interaction will be Poisson-distributed, conditional upon the mean measure. Conversely, non-Poisson residuals are evidence that your model has failed to take out some kind of interaction or hidden variable.

Spelled
\(\operatorname{Poisson}(\lambda)\)
Pmf
\(\mathbb{P}(k;\lambda)={\frac {\lambda ^{k}}{k!}}e^{{-\lambda }}\)
Mean
\(\lambda\)
Variance
\(\lambda\)
Pgf
\(G(s;\lambda)=\exp(\lambda (s-1))\)

2 Negative Binomial

Nearly a century old (Greenwood and Yule 1920). A generic count data model which, unlike the Poisson, has both location-like and scale-like parameters, instead of only one parameter. This makes one feel better about using it than a Poisson, since 2 parameters seems like a good number. Has a traditional rationale in terms of drawing balls from urns and such, which is of little interest here. The key virtue here is that it is both flexible and uncontroversial.

It includes Geometric as special cases when \(r=1\), and the Gamma and Poisson as limiting cases. More precisely, the Poisson is a limiting case of the Pólya distribution. for example, in the large-k limit, it approximates the Gamma distribution, and, when the mean is held constant, in the large \(r\) limit, it approaches Poisson. For fixed r it is an exponential family

For all that, it’s still contrived, this model, and tedious to fit.

Spelled
\(\operatorname{NB}(p,r)\)
Pmf
\(\mathbb{P}(k;p,r) = {k+r-1 \choose k}\cdot (1-p)^{r}p^{k} = \frac{\Gamma(k+r)}{k!\,\Gamma(r)} (1-p)^rp^k\)
Mean
\({\frac {pr}{1-p}}\)
Variance
\({\frac {pr}{(1-p)^{2}}}\)
Pgf
\(G(s;p,r)=\left({\frac {1-p}{1-ps}}\right)^{{\!r}}{\text{ for }}|s|<{\frac 1p}\)

2.1 Mean/dispersion parameterisation (Pólya)

Commonly, where the \(r\) parameter is not required to be a non-negative integer, we call it a Pólya model, and use it for over-dispersed data i.e. data that looks like a Poisson process if we are drunk, but whose variance is too big in comparison to its mean for soberer minds.

To see how that works we reparameterise the model in terms of a “location” parameter \(\lambda\) and “dispersion”/scale-ish parameter \(\alpha\), such that we can rewrite it.

Spelled
\(\text{Pólya}(\lambda,\alpha)\)
Pmf
\[ \mathbb{P}(k;\lambda,\alpha) = \frac{ \Gamma\left( k+\frac{1}{\alpha} \right) }{ \Gamma(k+1)\Gamma\left( \frac{1}{\alpha} \right) } \left( \frac{ \lambda }{ \lambda+\frac{1}{\alpha} }\right)^k\left( 1+\lambda\alpha \right)^{-\frac{1}{\alpha} } \]
Mean
\(\lambda\)
Variance
\(\lambda + \lambda^2\alpha\)
Pgf
\(G(s;\lambda,\alpha)=(\alpha (\lambda -\lambda s)+1)^{-1/\alpha }\)

The log Pmf is thus

\[ \mathbb{P}(k;\lambda,\alpha) =\begin{array}{l} \log\Gamma\left(k+\frac{1}{\alpha}\right)\\- \log\Gamma(k+1)\\- \log\Gamma\left( \frac{1}{\alpha} \right)\\+k\left( \log\lambda-\log\left( \lambda+\frac{1}{\alpha} \right) \right)\\- \frac{\log(1+\lambda\alpha)}{\alpha} \end{array}\]

All these log-gamma differences are numerically unstable if calculated naïvely, so we need to use different numerical methods depending upon the combination of \(k,\lambda\) and \(\alpha\) values in play. Aieee! Tedious!

To bother with this nonsense, one would need a strong reason to believe the negative binomial was the true model; otherwise, for example, the Lagrangian Poisson-Paisson is easier to fit, in that the log Pmf is numerically tractable for all parameter combinations even with large count values, and it is typically no less natural for common data sets.

3 Geometric

A discrete analogue of the exponential, i.e. the probability distribution of the number X of Bernoulli trials before the first success, supported on the set \(\{0, 1, 2, 3, …\}\)

Spelled
\(\text{Geom}(p)\)
Pmf
\(\mathbb{P}(k;p) = (1-p)^k\,p\!\)
Pgf
\(G(s;p)=\frac{p}{1-s+sp}.\)
Mean
\(\frac {1-p}{p}\)
Variance
\(\frac {1-p}{p^{2}}\)

Note that \({\operatorname{Geom}}(p)={\operatorname{NB}}(1,\,1-p).\,\).

3.1 Mean parameterisation

We can parameterise this in terms of the mean \(\lambda=\frac {1-p}{p}\Rightarrow p=\frac{1}{\lambda+1}\)

Spelled
\(\operatorname{MGeom}(\lambda)\)
Pmf
\(\mathbb{P}(k;\lambda) = \left(\frac{\lambda}{\lambda+1}\right)^k\frac{1}{\lambda+1}\)
Pgf
\(G(s;\lambda)=\frac{\lambda+1}{s\lambda+1}.\)
Mean
\(\lambda\)
Variance
\(\lambda^2+\lambda\)

4 Lagrangian distributions

I like this family. Handled under cascade models. This family includes various others on this page; I will check which some day.

5 Discrete Stable

Another generalisation of Poisson, with features such as a power-law tail.

By analogy with the continuous-stable distribution, a “stable” family for count data.

In particular, this is stable in the sense that it is a limiting distribution for sums of count random variables, analogous to the continuous stable family for real-valued RVs.

No (convenient) closed form for the Pmf in the general case, but the Pgf is simple, so that’s something.

Spelled
\(\operatorname{DS}(\nu,a)\)
Pmf
\(\mathbb{P}(k;\nu,a)= \left.\frac{1}{k!} \frac{d^kG(s;\nu,a)}{ds^k}\right|_{s=0}.\) (which is simply the formula for extracting the Pmf from any Pgf.)
Pgf
\(G(s;\nu,a)=\exp(-a (1-s)^\nu).\)
Mean
\(\operatorname{E}(X) = G'(1^-) = a \nu e^{-a}.\)
Variance
\(\operatorname{Var}(X)=G''(1^-) + \operatorname{E}(X) - \operatorname{E}^2(X).\)

Here, \(a>0\) is a scale parameter and \(0<\nu\leq 1\) a dispersion parameter describing in particular a power-law tail such that when \(\nu< 1\),

\[ \lim_{k \to \infty}\mathbb{P}(k;\nu,a) \simeq \frac{1}{k^{\nu+1}}. \]

Question: the Pgf formulation implies this is a non-negative distribution. Does that mean that symmetric discrete RVs cannot be stable? Possibly-negative ones?

(John P. Nolan 2001; J. P. Nolan 1997) give approximate ML-estimators of the parameter. (Wai Ha Lee 2010) does some legwork:

This thesis considers the interplay between the continuous and discrete properties of random stochastic processes. It is shown that the special cases of the one-sided Lévy-stable distributions can be connected to the class of discrete-stable distributions through a doubly-stochastic Poisson transform. This facilitates the creation of a one-sided stable process for which the N-fold statistics can be factorised explicitly. […] Using the same Poisson transform interrelationship, an exact method for generating discrete-stable variates is found. It has already been shown that discrete-stable distributions occur in the crossing statistics of continuous processes whose autocorrelation exhibits fractal properties. The statistical properties of a nonlinear filter analogue of a phase-screen model are calculated, and the level crossings of the intensity analysed. […] The asymptotic properties of the inter-event density of the process are found to be accurately approximated by a function of the Fano factor and the mean of the crossings alone.

Brrr. That sounds gruelling. You can probably just read the shorter article version though (W. H. Lee, Hopcraft, and Jakeman 2008).

6 Zipf/Zeta models

The discrete version of the basic power-law models.

While we are here, the plainest explanation of the relation of Zips to Pareto distribution that I know is Lada Adamic’s Zipf, Power-laws, and Pareto - a ranking tutorial.

Spelled
\(\operatorname{Zipf}(s)\)
Pmf
\(\mathbb{P}(k;s)={\frac {1/k^{s}}{\zeta (s)}}\)
Mean
\({\frac {\zeta (s-1)}{\zeta (s)}}~{\text{ for }}~s>2\)
Variance
\({\frac {\zeta (s)\zeta (s-2)-\zeta (s-1)^{2}}{\zeta (s)^{2}}}~{\text{ for }}~s>3\)

This has unbounded support. In the bounded case, it becomes the Zipf—Mandelbrot law, which is too fiddly for me to discuss here unless I turn out to really need it, which would likely be for ranking statistics.

This is a heavy tailed distribution, in that it does not necessarily have finite higher moments.

John D. Cook wrote about these recently.

7 Yule-Simon

Spelled
\(\operatorname{YS}(\rho)\)
Pmf
\(\mathbb{P}(k;\rho)=\rho \,{\mathrm {B}}(k,\rho +1)\)
Mean
\({\frac {\rho }{\rho -1}},\, \rho > 1\)
Variance
\({\frac {\rho^2}{(\rho-1)^2\;(\rho -2)}}, \rho > 2\)

where B is the beta function.

Zipf law in the tail. See also the two-parameter version, which replaces the beta function with an incomplete beta function, giving Pmf \(\mathbb{P}(k;\rho,\alpha )={\frac {\rho }{1-\alpha ^{{\rho }}}}\;{\mathrm {B}}_{{1-\alpha }}(k,\rho +1),\,\)

This is also a heavy tailed distribution, in that it does not necessarily have finite higher moments.

I’m bored with this one too.

8 Conway-Maxwell-Poisson

Exponential family count model with free variance parameter. See (Chatla and Shmueli 2016).

9 Decomposability properties

For background, see decomposability.

9.1 Stability

By analogy with the continuous case we may construct a stability equation for count RVs:

\[ X(a) \triangleq W^{1/\alpha}\odot X(b), \]

\(\odot\) here is Steutel and van Harn’s discrete multiplication operator, which I won’t define here exhaustively because there are variously complex formulations of it, and I don’t care enough to wrangle them. In the simplest case it gives us a binomial thinning of the left operand by the right

\[ A\odot B \sim \mathrm{Binom}(A,B) \]

9.2 Self-divisibility

Poisson RVs are self-divisible, in the sense that

\[\begin{aligned} X_1&\sim \operatorname{Poisson}(\lambda_1),\\ X_2&\sim \operatorname{Poisson}(\lambda_2),\\ X_1&\perp X_2\\ &\Rightarrow \\ X_1+X_2&\sim \operatorname{Poisson}(\lambda_1+\lambda_2). \end{aligned}\]

Pólya RVs likewise are self-divisible:

\[\begin{aligned} X_1&\sim \operatorname{Pólya}(\lambda_1, \alpha_1),\,\\ X_2&\sim \operatorname{Pólya}(\lambda_1, \alpha_1),\,\\ X_1&\perp X_2 &\Rightarrow\\ X_1+X_2&\sim \operatorname{Pólya}\left(\lambda_1+\lambda_2,\, \frac{\alpha_1\lambda_1^2+\alpha_2\lambda_2^2}{(\lambda_1+\lambda_2)^2}\right)\\ \operatorname{E}(X_1+X_2)&=\operatorname{E}(X_1)+\operatorname{E}(X_2)\\ \operatorname{Var}(X_1+X_2)&=\operatorname{Var}(X_1)+\operatorname{Var}(X_2) \end{aligned}\]

So, AFAICS, are GPDs, in \(\lambda\).

10 References

Blaker. 2000. Confidence Curves and Improved Exact Confidence Intervals for Discrete Distributions.” The Canadian Journal of Statistics / La Revue Canadienne de Statistique.
Burridge. 2013a. Cascade Sizes in a Branching Process with Gamma Distributed Generations.” arXiv:1304.3741 [Math].
———. 2013b. Crossover Behavior in Driven Cascades.” Physical Review E.
Cao, and Xie. 2015. Poisson Matrix Recovery and Completion.” arXiv:1504.05229 [Cs, Math, Stat].
Chatla, and Shmueli. 2016. Modeling Big Count Data: An IRLS Framework for CMP Regression and GAM.” arXiv:1610.08244 [Stat].
Consul, P. C. 1988. Generalized Poisson Distributions.
Consul, P. C., and Famoye. 1992. Generalized Poisson Regression Model.” Communications in Statistics - Theory and Methods.
———. 2006. Lagrangian Probability Distributions.
Consul, P. C., and Felix. 1989. Minimum Variance Unbiased Estimation for the Lagrange Power Series Distributions.” Statistics.
Consul, P. C., and Jain. 1973. A Generalization of the Poisson Distribution.” Technometrics.
Consul, P., and Shenton. 1972. Use of Lagrange Expansion for Generating Discrete Generalized Probability Distributions.” SIAM Journal on Applied Mathematics.
Consul, P. C., and Shenton. 1973. Some Interesting Properties of Lagrangian Distributions.” Communications in Statistics.
———. 1975. On the Probabilistic Structure and Properties of Discrete Lagrangian Distributions.” In A Modern Course on Statistical Distributions in Scientific Work. NATO Advanced Study Institutes Series 17.
Consul, P.C., and Shoukri. 1984. Maximum Likelihood Estimation for the Generalized Poisson Distribution.” Communications in Statistics - Theory and Methods.
Consul, P.C., and Shoukri. 1988. Some Chance Mechanisms Related to a Generalized Poisson Probability Model.” American Journal of Mathematical and Management Sciences.
Cox. 1983. Some Remarks on Overdispersion.” Biometrika.
Cui, and Lund. 2009. A New Look at Time Series of Counts.” Biometrika.
Doray, Jiang, and Luong. 2009. Some Simple Method of Estimation for the Parameters of the Discrete Stable Distribution with the Probability Generating Function.” Communications in Statistics - Simulation and Computation.
Dwass. 1969. The Total Progeny in a Branching Process and a Related Random Walk.” Journal of Applied Probability.
Felix, and Lee. 1992. Estimation of Generalized Poisson Distribution.” Communications in Statistics - Simulation and Computation.
Goldstein, Morris, and Yen. 2004. Problems with Fitting to the Power-Law Distribution.” The European Physical Journal B - Condensed Matter and Complex Systems.
Greenwood, and Yule. 1920. An Inquiry into the Nature of Frequency Distributions Representative of Multiple Happenings with Particular Reference to the Occurrence of Multiple Attacks of Disease or of Repeated Accidents.” Journal of the Royal Statistical Society.
Haight, and Breuer. 1960. The Borel-Tanner Distribution.” Biometrika.
Imoto. 2016. Properties of Lagrangian Distributions.” Communications in Statistics - Theory and Methods.
Janardan. 1984. Moments of Certain Series Distributions and Their Applications.” SIAM Journal on Applied Mathematics.
Johnson, Kemp, and Kotz. 2005. Univariate Discrete Distributions.
Karlis, and Mamode Khan. 2023. Models for Integer Data.” Annual Review of Statistics and Its Application.
Lee, Wai Ha. 2010. Continuous and Discrete Properties of Stochastic Processes.”
Lee, W. H., Hopcraft, and Jakeman. 2008. Continuous and Discrete Stable Processes.” Physical Review E.
Li, Famoye, and Lee. 2010. “On the Generalized Lagrangian Probability Distributions.” Journal of Probability and Statistical Science.
Lloyd-Smith. 2007. Maximum Likelihood Estimation of the Negative Binomial Dispersion Parameter for Highly Overdispersed Data, with Applications to Infectious Diseases.” PLoS ONE.
Mutafchiev. 1995. Local Limit Approximations for Lagrangian Distributions.” Aequationes Mathematicae.
Neyman. 1965. Certain Chance Mechanisms Involving Discrete Distributions.” Sankhyā: The Indian Journal of Statistics, Series A (1961-2002).
Nolan, J.P. 1997. Parameter Estimation and Data Analysis for Stable Distributions.” In.
Nolan, John P. 1997. Numerical Calculation of Stable Densities and Distribution Functions.” Communications in Statistics. Stochastic Models.
———. 2001. Maximum Likelihood Estimation and Diagnostics for Stable Distributions.” In Lévy Processes.
Pitman. 1998. Enumerations of Trees and Forests Related to Branching Processes and Random Walks.” In Microsurveys in Discrete Probability. DIMACS Series in Discrete Mathematics and Theoretical Computer Science.
Saha, and Paul. 2005. Bias-corrected maximum likelihood estimator of the negative binomial dispersion parameter.” Biometrics.
Schein, Wallach, and Zhou. 2016. Poisson-Gamma Dynamical Systems.” In Advances In Neural Information Processing Systems.
Shmueli, Minka, Kadane, et al. 2005. A Useful Distribution for Fitting Discrete Data: Revival of the Conway-Maxwell-Poisson Distribution.” Journal of the Royal Statistical Society. Series C (Applied Statistics).
Shoukri, and Consul. 1987. Some Chance Mechanisms Generating the Generalized Poisson Probability Models.” In Biostatistics.
Sibuya, Miyawaki, and Sumita. 1994. Aspects of Lagrangian Probability Distributions.” Journal of Applied Probability.
Soltani, Shirvani, and Alqallaf. 2009. A Class of Discrete Distributions Induced by Stable Laws.” Statistics & Probability Letters.
Steutel, and van Harn. 1979. Discrete Analogues of Self-Decomposability and Stability.” The Annals of Probability.
Steyvers, and Tenenbaum. 2005. The Large-Scale Structure of Semantic Networks: Statistical Analyses and a Model of Semantic Growth.” Cognitive Science.
Tanner. 1961. A Derivation of the Borel Distribution.” Biometrika.
Tsou. 2006. Robust Poisson Regression.” Journal of Statistical Planning and Inference.
Tuenter. 2000. On the Generalized Poisson Distribution.” Statistica Neerlandica.
Turkman, Scotto, and Bermudez. 2014. “Models for Integer-Valued Time Series.” In Non-Linear Time Series.
van Harn, and Steutel. 1993. Stability Equations for Processes with Stationary Independent Increments Using Branching Processes and Poisson Mixtures.” Stochastic Processes and Their Applications.
van Harn, Steutel, and Vervaat. 1982. Self-Decomposable Discrete Distributions and Branching Processes.” Zeitschrift Für Wahrscheinlichkeitstheorie Und Verwandte Gebiete.
Wedel, Böckenholt, and Kamakura. 2003. Factor Models for Multivariate Count Data.” Journal of Multivariate Analysis.
Wolpert, and Brown. 2021. Markov Infinitely-Divisible Stationary Time-Reversible Integer-Valued Processes.” arXiv:2105.14591 [Math].