Abstract:We define a family of probability distributions for random count matrices with a potentially unbounded number of rows and columns. The three distributions we consider are derived from the gamma-Poisson, gamma-negative binomial, and beta-negative binomial processes, which we refer to generically as a family of negative-binomial processes. Because the models lead to closed-form update equations within the context of a Gibbs sampler, they are natural candidates for nonparametric Bayesian priors over count matrices. A key aspect of our analysis is the recognition that although the random count matrices within the family are defined by a row-wise construction, their columns can be shown to be independent and identically distributed (iid). This fact is used to derive explicit formulas for drawing all the columns at once. Moreover, by analyzing these matrices’ combinatorial structure, we describe how to sequentially construct a column-iid random count matrix one row at a time, and derive the predictive distribution of a new row count vector with previously unseen features. We describe the similarities and differences between the three priors, and argue that the greater flexibility of the gamma- and beta-negative binomial processes—especially their ability to model over-dispersed, heavy-tailed count data—makes these well suited to a wide variety of real-world applications. As an example of our framework, we construct a naive-Bayes text classifier to categorize a count vector to one of several existing random count matrices of different categories. The classifier supports an unbounded number of features and, unlike most existing methods, it does not require a predefined finite vocabulary to be shared by all the categories, and needs neither feature selection nor parameter tuning. Both the gamma- and beta-negative binomial processes are shown to significantly outperform the gamma-Poisson process when applied to document categorization, with comparable performance to other state-of-the-art supervised text classification algorithms. Supplementary materials for this article are available online.

A Generic Multivariate Distribution for Counting Data

On mean and/or variance mixtures of normal distributions

Bayesian estimation and prediction for certain mixtures

A formulation for continuous mixtures of multivariate normal distributions

A parsimonious family of multivariate Poisson-lognormal distributions for clustering multivariate count data

A new infinitely divisible discrete distribution with applications to count data modeling

Multivariate Bernoulli distribution

A Bayesian Zero-Inflated Dirichlet-Multinomial Regression Model for Multivariate Compositional Count Data

Review of Probability Distributions for Modeling Count Data

On the Generalized Poisson Regression Mixture Model for Mapping Quantitative Trait Loci with Count Data

Multivariate binary probability distribution in the Grassmann formalism

Parametric Modelling of Multivariate Count Data Using Probabilistic Graphical Models

Some Theoretical and Computational Aspects of the Truncated Multivariate Skew-Normal/Independent Distributions

Priors for Random Count Matrices Derived from a Family of Negative Binomial Processes

Splitting models for multivariate count data

Generating multivariate extreme value distributions

Developing multivariate distributions using Dirichlet generator

On count data models based on Bernstein functions or their inverses

Copula-type Estimators for Flexible Multivariate Density Modeling using Mixtures

Nonparametric Bayesian Negative Binomial Factor Analysis

Multimatrix variate distributions