Determinantal point processes for machine learning

Alex Kulesza, Ben Taskar

Introduction

Probabilistic modeling and learning techniques have become indispensable tools for analyzing data, discovering patterns, and making predictions in a variety of real-world settings. In recent years, the widespread availability of both data and processing capacity has led to new applications and methods involving more complex, structured output spaces, where the goal is to simultaneously make a large number of interrelated decisions. Unfortunately, the introduction of structure typically involves a combinatorial explosion of output possibilities, making inference computationally impractical without further assumptions.

A popular compromise is to employ graphical models, which are tractable when the graph encoding local interactions between variables is a tree. For loopy graphs, inference can often be approximated in the special case when the interactions between variables are positive and neighboring nodes tend to have the same labels. However, dealing with global, negative interactions in graphical models remains intractable, and heuristic methods often fail in practice.

Determinantal point processes (DPPs) offer a promising and complementary approach. Arising in quantum physics and random matrix theory, DPPs are elegant probabilistic models of global, negative correlations, and offer efficient algorithms for sampling, marginalization, conditioning, and other inference tasks. While they have been studied extensively by mathematicians, giving rise to a deep and beautiful theory, DPPs are relatively new in machine learning. We aim to provide a comprehensible introduction to DPPs, focusing on the intuitions, algorithms, and extensions that are most relevant to our community.

A DPP is a distribution over subsets of a fixed ground set, for instance, sets of search results selected from a large database. Equivalently, a DPP over a ground set of NN items can be seen as modeling a binary characteristic vector of length NN. The essential characteristic of a DPP is that these binary variables are negatively correlated; that is, the inclusion of one item makes the inclusion of other items less likely. The strengths of these negative correlations are derived from a kernel matrix that defines a global measure of similarity between pairs of items, so that more similar items are less likely to co-occur. As a result, DPPs assign higher probability to sets of items that are diverse; for example, a DPP will prefer search results that cover multiple distinct aspects of a user’s query, rather than focusing on the most popular or salient one.

This focus on diversity places DPPs alongside a number of recently developed techniques for working with diverse sets, particularly in the information retrieval community (Carbonell and Goldstein, 1998; Zhai et al., 2003; Chen and Karger, 2006; Yue and Joachims, 2008; Radlinski et al., 2008; Swaminathan et al., 2009; Raman et al., 2012). However, unlike these methods, DPPs are fully probabilistic, opening the door to a wider variety of potential applications, without compromising algorithmic tractability.

The general concept of diversity can take on a number of forms depending on context and application. Including multiple kinds of search results might be seen as covering or summarizing relevant interpretations of the query or its associated topics; see Figure 1. Alternatively, items inhabiting a continuous space may exhibit diversity as a result of repulsion, as in Figure 2. In fact, certain repulsive quantum particles are known to be precisely described by a DPP; however, a DPP can also serve as a model for general repulsive phenomena, such as the locations of trees in a forest, which appear diverse due to physical and resource constraints. Finally, diversity can be used as a filtering prior when multiple selections must be based on a single detector or scoring metric. For instance, in Figure 3 a weak pose detector favors large clusters of poses that are nearly identical, but filtering through a DPP ensures that the final predictions are well-separated.

Throughout this survey we demonstrate applications for DPPs in a variety of settings, including:

The DUC 2003/2004 text summarization task, where we form extractive summaries of news articles by choosing diverse subsets of sentences (Section 4.2.1);

An image search task, where we model human judgments of diversity for image sets returned by Google Image Search (Section 5.3);

A multiple pose estimation task, where we improve the detection of human poses in images from television shows by incorporating a bias toward non-overlapping predictions (Section 6.4);

A news threading task, where we automatically extract timelines of important news stories from a large corpus by balancing intra-timeline coherence with inter-timeline diversity (Section 6.6.4).

2 Outline

In this paper we present general mathematical background on DPPs along with a range of modeling extensions, efficient algorithms, and theoretical results that aim to enable practical modeling and learning. The material is organized as follows.

We begin with an introduction to determinantal point processes tailored to the interests of the machine learning community. We focus on discrete DPPs, emphasizing intuitions and including new, simplified proofs for some theoretical results. We provide descriptions of known efficient inference algorithms, and characterize their computational properties.

We describe a decomposition of the DPP that makes explicit its fundamental tradeoff between quality and diversity. We compare the expressive power of DPPs and MRFs, characterizing the tradeoffs in terms of modeling power and computational efficiency. We also introduce a dual representation for DPPs, showing how it can be used to perform efficient inference over large ground sets. When the data are high-dimensional and dual inference is still too slow, we show that random projections can be used to maintain a provably close approximation to the original model while greatly reducing computational requirements.

We derive an efficient algorithm for learning the parameters of a quality model when the diversity model is held fixed. We employ this learning algorithm to perform extractive summarization of news text.

We present an extension of DPPs that allows for explicit control over the number of items selected by the model. We show not only that this extension solves an important practical problem, but also that it increases expressive power: a kk-DPP can capture distributions that a standard DPP cannot. The extension to kk-DPPs necessitates new algorithms for efficient inference based on recursions for the elementary symmetric polynomials. We validate the new model experimentally on an image search task.

We extend DPPs to model diverse sets of structured items, such as sequences or trees, where there are combinatorially many possible configurations. In this setting the number of possible subsets is doubly-exponential, presenting a daunting computational challenge. However, we show that a factorization of the quality and diversity models together with the dual representation for DPPs makes efficient inference possible using second-order message passing. We demonstrate structured DPPs on a toy geographical paths problem, a still-image multiple pose estimation task, and two high-dimensional text threading tasks.

Determinantal point processes

Determinantal point processes (DPPs) were first identified as a class by Macchi (1975), who called them “fermion processes” because they give the distributions of fermion systems at thermal equilibrium. The Pauli exclusion principle states that no two fermions can occupy the same quantum state; as a consequence fermions exhibit what is known as the “anti-bunching” effect. This repulsion is described precisely by a DPP.

In fact, years before Macchi gave them a general treatment, specific DPPs appeared in major results in random matrix theory (Mehta and Gaudin, 1960; Dyson, 1962a, b, c; Ginibre, 1965), where they continue to play an important role (Diaconis, 2003; Johansson, 2005b). Recently, DPPs have attracted a flurry of attention in the mathematics community (Borodin and Olshanski, 2000; Borodin and Soshnikov, 2003; Borodin and Rains, 2005; Borodin et al., 2010; Burton and Pemantle, 1993; Johansson, 2002, 2004, 2005a; Okounkov, 2001; Okounkov and Reshetikhin, 2003; Shirai and Takahashi, 2000), and much progress has been made in understanding their formal combinatorial and probabilistic properties. The term “determinantal” was first used by Borodin and Olshanski (2000), and has since become accepted as standard. Many good mathematical surveys are now available (Borodin, 2009; Hough et al., 2006; Shirai and Takahashi, 2003a, b; Lyons, 2003; Soshnikov, 2000; Tao, 2009).

We begin with an overview of the aspects of DPPs most relevant to the machine learning community, emphasizing intuitions, algorithms, and computational properties.

A point process P\mathcal{P} on a ground set Y\mathcal{Y} is a probability measure over “point patterns” or “point configurations” of Y\mathcal{Y}, which are finite subsets of Y\mathcal{Y}. For instance, Y\mathcal{Y} could be a continuous time interval during which a scientist records the output of a brain electrode, with P({y1,y2,y3})\mathcal{P}(\{y_{1},y_{2},y_{3}\}) characterizing the likelihood of seeing neural spikes at times y1y_{1}, y2y_{2}, and y3y_{3}. Depending on the experiment, the spikes might tend to cluster together, or they might occur independently, or they might tend to spread out in time. P\mathcal{P} captures these correlations.

For the remainder of this paper, we will focus on discrete, finite point processes, where we assume without loss of generality that Y={1,2,,N}\mathcal{Y}=\{1,2,\dots,N\}; in this setting we sometimes refer to elements of Y\mathcal{Y} as items. Much of our discussion extends to the continuous case, but the discrete setting is computationally simpler and often more appropriate for real-world data—e.g., in practice, the electrode voltage will only be sampled at discrete intervals. The distinction will become even more apparent when we apply our methods to Y\mathcal{Y} with no natural continuous interpretation, such as the set of documents in a corpus.

In the discrete case, a point process is simply a probability measure on 2Y2^{\mathcal{Y}}, the set of all subsets of Y\mathcal{Y}. A sample from P\mathcal{P} might be the empty set, the entirety of Y\mathcal{Y}, or anything in between. P\mathcal{P} is called a determinantal point process if, when Y\boldsymbol{Y} is a random subset drawn according to P\mathcal{P}, we have, for every AYA\subseteq\mathcal{Y},

for some real, symmetric N×NN\times N matrix KK indexed by the elements of Y\mathcal{Y}.In general, KK need not be symmetric. However, in the interest of simplicity, we proceed with this assumption; it is not a significant limitation for our purposes. Here, KA[Kij]i,jAK_{A}\equiv\left[K_{ij}\right]_{i,j\in A} denotes the restriction of KK to the entries indexed by elements of AA, and we adopt det(K)=1\det(K_{\emptyset})=1. Note that normalization is unnecessary here, since we are defining marginal probabilities that need not sum to 1.

Since P\mathcal{P} is a probability measure, all principal minors det(KA)\det(K_{A}) of KK must be nonnegative, and thus KK itself must be positive semidefinite. It is possible to show in the same way that the eigenvalues of KK are bounded above by one using Equation (35), which we introduce later. These requirements turn out to be sufficient: any KK, 0KI0\preceq K\preceq I, defines a DPP. This will be a consequence of Theorem 2.3.

We refer to KK as the marginal kernel since it contains all the information needed to compute the probability of any subset AA being included in Y\boldsymbol{Y}. A few simple observations follow from Equation (1). If A={i}A=\{i\} is a singleton, then we have

That is, the diagonal of KK gives the marginal probabilities of inclusion for individual elements of Y\mathcal{Y}. Diagonal entries close to 1 correspond to elements of Y\mathcal{Y} that are almost always selected by the DPP. Furthermore, if A={i,j}A=\{i,j\} is a two-element set, then

Thus, the off-diagonal elements determine the negative correlations between pairs of elements: large values of KijK_{ij} imply that ii and jj tend not to co-occur.

Equation (7) demonstrates why DPPs are “diversifying”. If we think of the entries of the marginal kernel as measurements of similarity between pairs of elements in Y\mathcal{Y}, then highly similar elements are unlikely to appear together. If Kij=KiiKjjK_{ij}=\sqrt{K_{ii}K_{jj}}, then ii and jj are “perfectly similar” and do not appear together almost surely. Conversely, when KK is diagonal there are no correlations and the elements appear independently. Note that DPPs cannot represent distributions where elements are more likely to co-occur than if they were independent: correlations are always nonpositive.

Figure 4 shows the difference between sampling a set of points in the plane using a DPP (with KijK_{ij} inversely related to the distance between points ii and jj), which leads to a relatively uniformly spread set with good coverage, and sampling points independently, which results in random clumping.

In this paper, our focus is on using DPPs to model real-world data. However, many theoretical point processes turn out to be exactly determinantal, which is one of the main reasons they have received so much recent attention. In this section we briefly describe a few examples; some of them are quite remarkable on their own, and as a whole they offer some intuition about the types of distributions that are realizable by DPPs. Technical details for each example can be found in the accompanying reference.

Given a sequence of NN random numbers drawn uniformly and independently from a finite set (say, the digits 0–9), the locations in the sequence where the current number is less than the previous number form a subset of {2,3,,N}\{2,3,\dots,N\}. This subset is distributed as a determinantal point process. Intuitively, if the current number is less than the previous number, it is probably not too large, thus it becomes less likely that the next number will be smaller yet. In this sense, the positions of decreases repel one another.

Let GG be an arbitrary finite graph with NN edges, and let TT be a random spanning tree chosen uniformly from the set of all the spanning trees of GG. The edges in TT form a subset of the edges of GG that is distributed as a DPP. The marginal kernel in this case is the transfer-impedance matrix, whose entry Ke1e2K_{e_{1}e_{2}} is the expected signed number of traversals of edge e2e_{2} when a random walk begins at one endpoint of e1e_{1} and ends at the other (the graph edges are first oriented arbitrarily). Thus, edges that are in some sense “nearby” in GG are similar according to KK, and as a result less likely to participate in a single uniformly chosen spanning tree. As this example demonstrates, some DPPs assign zero probability to sets whose cardinality is not equal to a particular kk; in this case, kk is the number of nodes in the graph minus one—the number of edges in any spanning tree. We will return to this issue in Section 5.

Let MM be a random matrix obtained by drawing every entry independently from the complex normal distribution. This is the complex Ginibre ensemble. The eigenvalues of MM, which form a finite subset of the complex plane, are distributed according to a DPP. If a Hermitian matrix is generated in the corresponding way, drawing each diagonal entry from the normal distribution and each pair of off-diagonal entries from the complex normal distribution, then we obtain the Gaussian unitary ensemble, and the eigenvalues are now a DPP-distributed subset of the real line.

The Aztec diamond is a diamond-shaped union of lattice squares, as depicted in Figure 5a. (Half of the squares have been colored gray in a checkerboard pattern.) A domino tiling is a perfect cover of the Aztec diamond using 2×12\times 1 rectangles, as in Figure 5b. Suppose that we draw a tiling uniformly at random from among all possible tilings. (The number of tilings is known to be exponential in the width of the diamond.) We can identify this tiling with the subset of the squares that are (a) painted gray in the checkerboard pattern and (b) covered by the left half of a horizontal tile or the bottom half of a vertical tile (see Figure 5c). This subset is distributed as a DPP.

2 L-ensembles

For the purposes of modeling real data, it is useful to slightly restrict the class of DPPs by focusing on L-ensembles. First introduced by Borodin and Rains (2005), an L-ensemble defines a DPP not through the marginal kernel KK, but through a real, symmetric matrix LL indexed by the elements of Y\mathcal{Y}:

Whereas Equation (1) gave the marginal probabilities of inclusion for subsets AA, Equation (8) directly specifies the atomic probabilities for every possible instantiation of Y\boldsymbol{Y}. As for KK, it is easy to see that LL must be positive semidefinite. However, since Equation (8) is only a statement of proportionality, the eigenvalues of LL need not be less than one; any positive semidefinite LL defines an L-ensemble. The required normalization constant can be given in closed form due to the fact that YYdet(LY)=det(L+I)\sum_{Y\subseteq\mathcal{Y}}\det(L_{Y})=\det(L+I), where II is the N×NN\times N identity matrix. This is a special case of the following theorem.

where IAˉI_{\bar{A}} is the diagonal matrix with ones in the diagonal positions corresponding to elements of Aˉ=YA\bar{A}=\mathcal{Y}-A, and zeros everywhere else.

Suppose that A=YA=\mathcal{Y}; then Equation (9) holds trivially. Now suppose inductively that the theorem holds whenever Aˉ\bar{A} has cardinality less than kk. Given AA such that Aˉ=k>0|\bar{A}|=k>0, let ii be an element of Y\mathcal{Y} where iAˉi\in\bar{A}. Splitting blockwise according to the partition Y={i}Y{i}\mathcal{Y}=\{i\}\cup\mathcal{Y}-\{i\}, we can write

where LiˉiL_{\bar{i}i} is the subcolumn of the iith column of LL whose rows correspond to iˉ\bar{i}, and similarly for LiiˉL_{i\bar{i}}. By multilinearity of the determinant, then,

We can now apply the inductive hypothesis separately to each term, giving

where we observe that every YY either contains ii and is included only in the first sum, or does not contain ii and is included only in the second sum. ∎

As a shorthand, we will write PL(Y)\mathcal{P}_{L}(Y) instead of PL(Y=Y)\mathcal{P}_{L}(\boldsymbol{Y}=Y) when the meaning is clear.

We can write a version of Equation (7) for L-ensembles, showing that if LL is a measure of similarity then diversity is preferred:

In this case we are reasoning about the full contents of Y\boldsymbol{Y} rather than its marginals, but the intuition is essentially the same. Furthermore, we have the following result of Macchi (1975).

An L-ensemble is a DPP, and its marginal kernel is

Using Theorem 2.1, the marginal probability of a set AA is

We can use the fact that L(L+I)1=I(L+I)1L(L+I)^{-1}=I-(L+I)^{-1} to simplify and obtain

where we let K=I(L+I)1K=I-(L+I)^{-1}. Now, we observe that left multiplication by IAI_{A} zeros out all the rows of a matrix except those corresponding to AA. Therefore we can split blockwise using the partition Y=AˉA\mathcal{Y}=\bar{A}\cup A to get

Note that KK can be computed from an eigendecomposition of L=n=1NλnvnvnL=\sum_{n=1}^{N}\lambda_{n}\boldsymbol{v}_{n}\boldsymbol{v}_{n}^{\top} by a simple rescaling of eigenvalues:

Conversely, we can ask when a DPP with marginal kernel KK is also an L-ensemble. By inverting Equation (21), we have

and again the computation can be performed by eigendecomposition. However, while the inverse in Equation (21) always exists due to the positive coefficient on the identity matrix, the inverse in Equation (33) may not. In particular, when any eigenvalue of KK achieves the upper bound of 1, the DPP is not an L-ensemble. We will see later that the existence of the inverse in Equation (33) is equivalent to P\mathcal{P} giving nonzero probability to the empty set. (This is somewhat analogous to the positive probability assumption in the Hammersley-Clifford theorem for Markov random fields.) This is not a major restriction, for two reasons. First, when modeling real data we must typically allocate some nonzero probability for rare or noisy events, so when cardinality is one of the aspects we wish to model, the condition is not unreasonable. Second, we will show in Section 5 how to control the cardinality of samples drawn from the DPP, thus sidestepping the representational limits of L-ensembles.

Modulo the restriction described above, KK and LL offer alternative representations of DPPs. Under both representations, subsets that have higher diversity, as measured by the corresponding kernel, have higher likelihood. However, while KK gives marginal probabilities, L-ensembles directly model the atomic probabilities of observing each subset of Y\mathcal{Y}, which offers an appealing target for optimization. Furthermore, LL need only be positive semidefinite, while the eigenvalues of KK are bounded above. For these reasons we will focus our modeling efforts on DPPs represented as L-ensembles.

Determinants have an intuitive geometric interpretation. Let BB be a DD x NN matrix such that L=BBL=B^{\top}B. (Such a BB can always be found for DND\leq N when LL is positive semidefinite.) Denote the columns of BB by BiB_{i} for i=1,2,,Ni=1,2,\dots,N. Then:

where the right hand side is the squared Y|Y|-dimensional volume of the parallelepiped spanned by the columns of BB corresponding to elements in YY.

Intuitively, we can think of the columns of BB as feature vectors describing the elements of Y\mathcal{Y}. Then the kernel LL measures similarity using dot products between feature vectors, and Equation (34) says that the probability assigned by a DPP to a set YY is related to the volume spanned by its associated feature vectors. This is illustrated in Figure 6.

From this intuition we can verify several important DPP properties. Diverse sets are more probable because their feature vectors are more orthogonal, and hence span larger volumes. Items with parallel feature vectors are selected together with probability zero, since their feature vectors define a degenerate parallelepiped. All else being equal, items with large-magnitude feature vectors are more likely to appear, because they multiply the spanned volumes for sets containing them.

We will revisit these intuitions in Section 3.1, where we decompose the kernel LL so as to separately model the direction and magnitude of the vectors BiB_{i}.

3 Properties

In this section we review several useful properties of DPPs.

If Y\boldsymbol{Y} is distributed as a DPP with marginal kernel KK, then YA\boldsymbol{Y}\cap A, where AYA\subseteq\mathcal{Y}, is also distributed as a DPP, with marginal kernel KAK_{A}.

If Y\boldsymbol{Y} is distributed as a DPP with marginal kernel KK, then YY\mathcal{Y}-\boldsymbol{Y} is also distributed as a DPP, with marginal kernel Kˉ=IK\bar{K}=I-K. In particular, we have

where II indicates the identity matrix of appropriate size. It may seem counterintuitive that the complement of a diversifying process should also encourage diversity. However, it is easy to see that

where the inequality follows from Equation (7).

If KKK\preceq K^{\prime}, that is, KKK^{\prime}-K is positive semidefinite, then for all AYA\subseteq\mathcal{Y} we have

In other words, the DPP defined by KK^{\prime} is larger than the one defined by KK in the sense that it assigns higher marginal probabilities to every set AA. An analogous result fails to hold for LL due to the normalization constant.

If K=γKK=\gamma K^{\prime} for some 0γ<10\leq\gamma<1, then for all AYA\subseteq\mathcal{Y} we have

It is easy to see that KK defines the distribution obtained by taking a random set distributed according to the DPP with marginal kernel KK^{\prime}, and then independently deleting each of its elements with probability 1γ1-\gamma.

Note that, by Equation (21), λ1λ1+1,λ2λ2+1,,λNλN+1\frac{\lambda_{1}}{\lambda_{1}+1},\frac{\lambda_{2}}{\lambda_{2}+1},\dots,\frac{\lambda_{N}}{\lambda_{N}+1} are the eigenvalues of KK. Figure 7 shows a plot of the function f(λ)=λλ+1f(\lambda)=\frac{\lambda}{\lambda+1}. It is easy to see from this why the class of L-ensembles does not include DPPs where the empty set has probability zero—at least one of the Bernoulli trials would need to always succeed, and in turn one or more of the eigenvalues of LL would be infinite.

In some instances, the sum of Bernoullis may be an appropriate model for uncertain cardinality in real-world data, for instance when identifying objects in images where the number of objects is unknown in advance. In other situations, it may be more practical to fix the cardinality of Y\boldsymbol{Y} up front, for instance when a set of exactly ten search results is desired, or to replace the sum of Bernoullis with an alternative cardinality model. We show how these goals can be can be achieved in Section 5.

4 Inference

One of the primary advantages of DPPs is that, although the number of possible realizations of Y\boldsymbol{Y} is exponential in NN, many types of inference can be performed in polynomial time. In this section we review the inference questions that can (and cannot) be answered efficiently. We also discuss the empirical practicality of the associated computations and algorithms, estimating the largest values of NN that can be handled at interactive speeds (within 2–3 seconds) as well as under more generous time and memory constraints. The reference machine used for estimating real-world performance has eight Intel Xeon E5450 3Ghz cores and 32GB of memory.

As we have already seen, the partition function, despite being a sum over 2N2^{N} terms, can be written in closed form as det(L+I)\det(L+I). Determinants of N×NN\times N matrices can be computed through matrix decomposition in O(N3)O(N^{3}) time, or reduced to matrix multiplication for better asymptotic performance. The Coppersmith-Winograd algorithm, for example, can be used to compute determinants in about O(N2.376)O(N^{2.376}) time. Going forward, we will use ω\omega to denote the exponent of whatever matrix multiplication algorithm is used.

Practically speaking, modern computers can calculate determinants up to N5,000N\approx 5{,}000 at interactive speeds, or up to N40,000N\approx 40{,}000 in about five minutes. When NN grows much larger, the memory required simply to store the matrix becomes limiting. (Sparse storage of larger matrices is possible, but computing determinants remains prohibitively expensive unless the level of sparsity is extreme.)

4.2 Marginalization

The marginal probability of any set of items AA can be computed using the marginal kernel as in Equation (1). From Equation (35) we can also compute the marginal probability that none of the elements in AA appear. (We will see below how marginal probabilities of arbitrary configurations can be computed using conditional DPPs.)

If the DPP is specified as an L-ensemble, then the computational bottleneck for marginalization is the computation of KK. The dominant operation is the matrix inversion, which requires at least O(Nω)O(N^{\omega}) time by reduction to multiplication, or O(N3)O(N^{3}) using Gauss-Jordan elimination or various matrix decompositions, such as the eigendecomposition method proposed in Section 2.2. Since an eigendecomposition of the kernel will be central to sampling, the latter approach is often the most practical when working with DPPs.

Matrices up to N2,000N\approx 2{,}000 can be inverted at interactive speeds, and problems up to N20,000N\approx 20{,}000 can be completed in about ten minutes.

4.3 Conditioning

The distribution obtained by conditioning a DPP on the event that none of the elements in a set AA appear is easy to compute. For BYB\subseteq{\mathcal{Y}} not intersecting with AA we have

where LAˉL_{\bar{A}} is the restriction of LL to the rows and columns indexed by elements in YA\mathcal{Y}-A. In other words, the conditional distribution (over subsets of YA\mathcal{Y}-A) is itself a DPP, and its kernel LAˉL_{\bar{A}} is obtained by simply dropping the rows and columns of LL that correspond to elements in AA.

We can also condition a DPP on the event that all of the elements in a set AA are observed. For BB not intersecting with AA we have

where IAˉI_{\bar{A}} is the matrix with ones in the diagonal entries indexed by elements of YA\mathcal{Y}-A and zeros everywhere else. Though it is not immediately obvious, Borodin and Rains (2005) showed that this conditional distribution (over subsets of YA\mathcal{Y}-A) is again a DPP, with a kernel given by

(Following the N×NN\times N inversion, the matrix is restricted to rows and columns indexed by elements in YA\mathcal{Y}-A, then inverted again.) It is easy to show that the inverses exist if and only if the probability of AA appearing is nonzero.

Combining Equation (46) and Equation (49), we can write the conditional DPP given an arbitrary combination of appearing and non-appearing elements:

Thus, the class of DPPs is closed under most natural conditioning operations.

These formulas also allow us to compute arbitrary marginals. For example, applying Equation (21) to Equation (50) yields the marginal kernel for the conditional DPP given the appearance of AA:

(Note that KAK^{A} is indexed by elements of YA\mathcal{Y}-A, so this is only defined when AA and BB are disjoint.) Using Equation (35) for the complement of a DPP, we can now compute the marginal probability of any partial assignment, i.e.,

Computing conditional DPP kernels in general is asymptotically as expensive as the dominant matrix inversion, although in some cases (conditioning only on non-appearance), the inversion is not necessary. In any case, conditioning is at most a small constant factor more expensive than marginalization.

4.4 Sampling

Algorithm 1, due to Hough et al. (2006), gives an efficient algorithm for sampling a configuration YY from a DPP. The input to the algorithm is an eigendecomposition of the DPP kernel LL. Note that ei\boldsymbol{e}_{i} is the iith standard basis NN-vector, which is all zeros except for a one in the iith position. We will prove the following theorem.

Let L=n=1NλnvnvnL=\sum_{n=1}^{N}\lambda_{n}\boldsymbol{v}_{n}\boldsymbol{v}_{n}^{\top} be an orthonormal eigendecomposition of a positive semidefinite matrix LL. Then Algorithm 1 samples YPL\boldsymbol{Y}\sim\mathcal{P}_{L}.

Algorithm 1 has two main loops, corresponding to two phases of sampling. In the first phase, a subset of the eigenvectors is selected at random, where the probability of selecting each eigenvector depends on its associated eigenvalue. In the second phase, a sample YY is produced based on the selected vectors. Note that on each iteration of the second loop, the cardinality of YY increases by one and the dimension of VV is reduced by one. Since the initial dimension of VV is simply the number of selected eigenvectors (J|J|), Theorem 2.3 has the previously stated corollary that the cardinality of a random sample is distributed as a sum of Bernoulli variables.

To prove Theorem 2.3 we will first show that a DPP can be expressed as a mixture of simpler, elementary DPPs. We will then show that the first phase chooses an elementary DPP according to its mixing coefficient, while the second phase samples from the elementary DPP chosen in phase one.

A DPP is called elementary if every eigenvalue of its marginal kernel is in {0,1}\{0,1\}. We write PV\mathcal{P}^{V}, where VV is a set of orthonormal vectors, to denote an elementary DPP with marginal kernel KV=vVvvK^{V}=\sum_{\boldsymbol{v}\in V}\boldsymbol{v}\boldsymbol{v}^{\top}.

We introduce the term “elementary” here; Hough et al. (2006) refer to elementary DPPs as determinantal projection processes, since KVK^{V} is an orthonormal projection matrix to the subspace spanned by VV. Note that, due to Equation (33), elementary DPPs are not generally L-ensembles. We start with a technical lemma.

Let WnW_{n} for n=1,2,,Nn=1,2,\dots,N be an arbitrary sequence of k×kk\times k rank-one matrices, and let (Wn)i(W_{n})_{i} denote the ithith column of WnW_{n}. Let WJ=nJWnW_{J}=\sum_{n\in J}W_{n}. Then

Expanding on the first column of WJW_{J} using the multilinearity of the determinant,

and, applying the same operation inductively to all columns,

Since WnW_{n} has rank one, the determinant of any matrix containing two or more columns of WnW_{n} is zero; thus the terms in the sum vanish unless n1,n2,,nkn_{1},n_{2},\dots,n_{k} are distinct. ∎

A DPP with kernel L=n=1NλnvnvnL=\sum_{n=1}^{N}\lambda_{n}\boldsymbol{v}_{n}\boldsymbol{v}_{n}^{\top} is a mixture of elementary DPPs:

where VJV_{J} denotes the set {vn}nJ\{\boldsymbol{v}_{n}\}_{n\in J}.

Consider an arbitrary set AA, with k=Ak=|A|. Let Wn=[vnvn]AW_{n}=[\boldsymbol{v}_{n}\boldsymbol{v}_{n}^{\top}]_{A} for n=1,2,,Nn=1,2,\dots,N; note that all of the WnW_{n} have rank one. From the definition of KVJK^{V_{J}}, the mixture distribution on the right hand side of Equation (60) gives the following expression for the marginal probability of AA:

using the fact that det(L+I)=n=1N(λn+1)\det(L+I)=\prod_{n=1}^{N}(\lambda_{n}+1). Applying Lemma 57 in reverse and then the definition of KK in terms of the eigendecomposition of LL, we have that the marginal probability of AA given by the mixture is

Since the two distributions agree on all marginals, they are equal. ∎

Next, we show that elementary DPPs have fixed cardinality.

If Y\boldsymbol{Y} is drawn according to an elementary DPP PV\mathcal{P}^{V}, then Y=V|\boldsymbol{Y}|=|V| with probability one.

Since KVK^{V} has rank V|V|, PV(YY)=0\mathcal{P}^{V}(Y\subseteq\boldsymbol{Y})=0 whenever Y>V|Y|>|V|, so YV|\boldsymbol{Y}|\leq|V|. But we also have

Thus Y=V|\boldsymbol{Y}|=|V| almost surely. ∎

Lemma 2.6 says that the mixture weight of PVJ\mathcal{P}^{V_{J}} is given by the product of the eigenvalues λn\lambda_{n} corresponding to the eigenvectors vnVJ\boldsymbol{v}_{n}\in V_{J}, normalized by det(L+I)=n=1N(λn+1)\det(L+I)=\prod_{n=1}^{N}(\lambda_{n}+1). This shows that the first loop of Algorithm 1 selects an elementary DPP PV\mathcal{P}^{V} with probability equal to its mixture component. All that remains is to show that the second loop samples YPV\boldsymbol{Y}\sim\mathcal{P}^{V}.

Let BB represent the matrix whose rows are the eigenvectors in VV, so that KV=BBK^{V}=B^{\top}B. Using the geometric interpretation of determinants introduced in Section 2.2.1, det(KYV)\det(K^{V}_{Y}) is equal to the squared volume of the parallelepiped spanned by {Bi}iY\{B_{i}\}_{i\in Y}. Note that since VV is an orthonormal set, BiB_{i} is just the projection of ei\boldsymbol{e}_{i} onto the subspace spanned by VV.

Let k=Vk=|V|. By Lemma 2.7 and symmetry, we can consider without loss of generality a single Y={1,2,,k}Y=\{1,2,\dots,k\}. Using the fact that any vector both in the span of VV and perpendicular to ei\boldsymbol{e}_{i} is also perpendicular to the projection of ei\boldsymbol{e}_{i} onto the span of VV, by the base ×\times height formula for the volume of a parallelepiped we have

Assume that, as iteration jj of the second loop in Algorithm 1 begins, we have already selected y1=1,y2=2,,yj1=j1y_{1}=1,y_{2}=2,\dots,y_{j-1}=j-1. Then VV in the algorithm has been updated to an orthonormal basis for the subspace of the original VV perpendicular to e1,,ej1\boldsymbol{e}_{1},\dots,\boldsymbol{e}_{j-1}, and the probability of choosing yj=jy_{j}=j is exactly

Therefore, the probability of selecting the sequence 1,2,,k1,2,\dots,k is

Since volume is symmetric, the argument holds identically for all of the k!k! orderings of YY. Thus the total probability that Algorithm 1 selects YY is det(KYV)\det(K^{V}_{Y}). ∎

Algorithm 1 generates YY in uniformly random order.

To get a feel for the sampling algorithm, it is useful to visualize the distributions used to select ii at each iteration, and to see how they are influenced by previously chosen items. Figure 8a shows this progression for a simple DPP where Y\mathcal{Y} is a finely sampled grid of points in $,andthekernelissuchthatpointsaremoresimilartheclosertogethertheyare.Initially,theeigenvectors, and the kernel is such that points are more similar the closer together they are. Initially, the eigenvectorsVgiverisetoafairlyuniformdistributionoverpointsingive rise to a fairly uniform distribution over points in\mathcal{Y},butaseachsuccessivepointisselectedand, but as each successive point is selected andV$ is updated, the distribution shifts to avoid points near those already chosen. Figure 8b shows a similar progression for a DPP over points in the unit square.

The sampling algorithm also offers an interesting analogy to clustering. If we think of the eigenvectors of LL as representing soft clusters, and the eigenvalues as representing their strengths—the way we do for the eigenvectors and eigenvalues of the Laplacian matrix in spectral clustering—then a DPP can be seen as performing a clustering of the elements in Y\mathcal{Y}, selecting a random subset of clusters based on their strength, and then choosing one element per selected cluster. Of course, the elements are not chosen independently and cannot be identified with specific clusters; instead, the second loop of Algorithm 1 coordinates the choices in a particular way, accounting for overlap between the eigenvectors.

Algorithm 1 runs in time O(Nk3)O(Nk^{3}), where k=Vk=|V| is the number of eigenvectors selected in phase one. The most expensive operation is the O(Nk2)O(Nk^{2}) Gram-Schmidt orthonormalization required to compute VV_{\perp}. If kk is large, this can be reasonably expensive, but for most applications we do not want high-cardinality DPPs. (And if we want very high-cardinality DPPs, we can potentially save time by using Equation (35) to sample the complement instead.) In practice, the initial eigendecomposition of LL is often the computational bottleneck, requiring O(N3)O(N^{3}) time. Modern multi-core machines can compute eigendecompositions up to N1,000N\approx 1{,}000 at interactive speeds of a few seconds, or larger problems up to N10,000N\approx 10{,}000 in around ten minutes. In some instances, it may be cheaper to compute only the top kk eigenvectors; since phase one tends to choose eigenvectors with large eigenvalues anyway, this can be a reasonable approximation when the kernel is expected to be low rank. Note that when multiple samples are desired, the eigendecomposition needs to be performed only once.

Deshpande and Rademacher (2010) recently proposed a (1ϵ)(1-\epsilon)-approximate algorithm for sampling that runs in time O(N2logNk2ϵ2+NlogωNk2ω+1ϵ2ωlog(kϵlogN))O(N^{2}\log N\frac{k^{2}}{\epsilon^{2}}+N\log^{\omega}N\frac{k^{2\omega+1}}{\epsilon^{2\omega}}\log(\frac{k}{\epsilon}\log N)) when LL is already decomposed as a Gram matrix, L=BBL=B^{\top}B. When BB is known but an eigendecomposition is not (and NN is sufficiently large), this may be significantly faster than the exact algorithm.

4.5 Finding the mode

Finding the mode of a DPP—that is, finding the set YYY\subseteq\mathcal{Y} that maximizes PL(Y)\mathcal{P}_{L}(Y)—is NP-hard. In conditional models, this problem is sometimes referred to as maximum a posteriori (or MAP) inference, and it is also NP-hard for most general structured models such as Markov random fields. Hardness was first shown for DPPs by Ko et al. (1995), who studied the closely-related maximum entropy sampling problem: the entropy of a set of jointly Gaussian random variables is given (up to constants) by the log-determinant of their covariance matrix; thus finding the maximum entropy subset of those variables requires finding the principal covariance submatrix with maximum determinant. Here, we adapt the argument of Çivril and Magdon-Ismail (2009), who studied the problem of finding maximum-volume submatrices.

Let dpp-mode be the optimization problem of finding, for a positive semidefinite N×NN\times N input matrix LL indexed by elements of Y\mathcal{Y}, the maximum value of det(LY)\det(L_{Y}) over all YYY\subseteq\mathcal{Y}. dpp-mode is NP-hard, and furthermore it is NP-hard even to approximate dpp-mode to a factor of 89+ϵ\frac{8}{9}+\epsilon.

We reduce from exact 3-cover (X3C). An instance of X3C is a set SS and a collection CC of three-element subsets of SS; the problem is to decide whether there is a sub-collection CCC^{\prime}\subseteq C such that every element of SS appears exactly once in CC^{\prime} (that is, CC^{\prime} is an exact 3-cover). X3C is known to be NP-complete.

The reduction is as follows. Let Y={1,2,,C}\mathcal{Y}=\{1,2,\dots,|C|\}, and let BB be an S×C|S|\times|C| matrix where Bsi=13B_{si}=\frac{1}{\sqrt{3}} if CiC_{i} contains sSs\in S and zero otherwise. Define L=γBBL=\gamma B^{\top}B, where 1<γ981<\gamma\leq\frac{9}{8}. Note that the diagonal of LL is constant and equal to γ\gamma, and an off-diagonal entry LijL_{ij} is zero if and only if CiC_{i} and CjC_{j} do not intersect. LL is positive semidefinite by construction, and the reduction requires only polynomial time. Let k=S3k=\frac{|S|}{3}. We will show that the maximum value of det(LY)\det(L_{Y}) is greater than γk1\gamma^{k-1} if and only if CC contains an exact 3-cover of SS.

()(\leftarrow) If CCC^{\prime}\subseteq C is an exact 3-cover of SS, then it must contain exactly kk 3-sets. Letting YY be the set of indices in CC^{\prime}, we have LY=γIL_{Y}=\gamma I, and thus its determinant is γk>γk1\gamma^{k}>\gamma^{k-1}.

()(\rightarrow) Suppose there is no 3-cover of SS in CC. Let YY be an arbitrary subset of Y\mathcal{Y}. If Y<k|Y|<k, then

Now suppose Yk|Y|\geq k, and assume without loss of generality that Y={1,2,,Y}Y=\{1,2,\dots,|Y|\}. We have LY=γBYBYL_{Y}=\gamma B_{Y}^{\top}B_{Y}, and

Note that, since the columns of BB are normalized, each term in the product is at most one. Furthermore, at least Yk+1|Y|-k+1 of the terms must be strictly less than one, because otherwise there would be kk orthogonal columns, which would correspond to a 3-cover. By the construction of BB, if two columns BiB_{i} and BjB_{j} are not orthogonal then CiC_{i} and CjC_{j} overlap in at least one of three elements, so we have

We have shown that the existence of a 3-cover implies that the optimal value of dpp-mode is at least γk\gamma^{k}, while the optimal value cannot be more than γk1\gamma^{k-1} if there is no 3-cover. Thus any algorithm that can approximate dpp-mode to better than a factor of 1γ\frac{1}{\gamma} can be used to solve X3C in polynomial time. We can choose γ=98\gamma=\frac{9}{8} to show that an approximation ratio of 89+ϵ\frac{8}{9}+\epsilon is NP-hard. ∎

Since there are only C|C| possible cardinalities for YY, Theorem 2.9 shows that dpp-mode is NP-hard even under cardinality constraints.

(Ko et al., 1995) propose an exact, exponential branch-and-bound algorithm for finding the mode using greedy heuristics to build candidate sets; they tested their algorithm on problems up to N=75N=75, successfully finding optimal solutions in up to about an hour. Modern computers are likely a few orders of magnitude faster; however, this algorithm is still probably impractical for applications with large NN. Çivril and Magdon-Ismail (2009) propose an efficient greedy algorithm for finding a set of size kk, and prove that it achieves an approximation ratio of O(1k!)O(\frac{1}{k!}). While this guarantee is relatively poor for all but very small kk, in practice the results may be useful nonetheless.

PL\mathcal{P}_{L} is log-submodular; that is,

whenever YYY{i}Y\subseteq Y^{\prime}\subseteq\mathcal{Y}-\{i\}. Intuitively, adding elements to YY yields diminishing returns as YY gets larger. (This is easy to show by a volume argument.) Submodular functions can be minimized in polynomial time (Schrijver, 2000), and many results exist for approximately maximizing monotone submodular functions, which have the special property that supersets always have higher function values than their subsets (Nemhauser et al., 1978; Fisher et al., 1978; Feige, 1998). In Section 4.2.1 we will discuss how these kinds of greedy algorithms can be adapted for DPPs. However, in general PL\mathcal{P}_{L} is highly non-monotone, since the addition of even a single element can decrease the probability to zero.

Recently, Feige et al. (2007) showed that even non-monotone submodular functions can be approximately maximized in polynomial time using a local search algorithm, and a growing body of research has focused on extending this result in a variety of ways (Lee et al., 2009; Gharan and Vondrák, 2011; Vondrák et al., 2011; Feldman et al., 2011a, b; Chekuri et al., 2011). In our recent work we showed how the computational structure of DPPs gives rise to a particularly efficient variant of these methods (Kulesza et al., 2012).

5 Related processes

Historically, a wide variety of point process models have been proposed and applied to applications involving diverse subsets, particularly in settings where the items can be seen as points in a physical space and diversity is taken to mean some sort of “spreading” behavior. However, DPPs are essentially unique among this class in having efficient and exact algorithms for probabilistic inference, which is why they are particularly appealing models for machine learning applications. In this section we briefly survey the wider world of point processes and discuss the computational properties of alternative models; we will focus on point processes that lead to what is variously described as diversity, repulsion, (over)dispersion, regularity, order, and inhibition.

Perhaps the most fundamental point process is the Poisson point process, which is depicted on the right side of Figure 4 (Daley and Vere-Jones, 2003). While defined for continuous Y\mathcal{Y}, in the discrete setting the Poisson point process can be simulated by flipping a coin independently for each item, and including those items for which the coin comes up heads. Formally,

where pip_{i}\in is the bias of the iith coin. The process is called stationary when pip_{i} does not depend on ii; in a spatial setting this means that no region has higher density than any other region.

A random set Y\boldsymbol{Y} distributed as a Poisson point process has the property that whenever AA and BB are disjoint subsets of Y\mathcal{Y}, the random variables YA\boldsymbol{Y}\cap A and YB\boldsymbol{Y}\cap B are independent; that is, the points in Y\boldsymbol{Y} are not correlated. It is easy to see that DPPs generalize Poisson point processes by choosing the marginal kernel KK with Kii=piK_{ii}=p_{i} and Kij=0,ijK_{ij}=0,i\neq j. This implies that inference for Poisson point processes is at least as efficient as for DPPs; in fact, it is more efficient, since for instance it is easy to compute the most likely configuration. However, since Poisson point processes do not model correlations between variables, they are rather uninteresting for most real-world applications.

Addressing this weakness, various procedural modifications of the Poisson process have been proposed in order to introduce correlations between items. While such constructions can be simple and intuitive, leading to straightforward sampling algorithms, they tend to make general statistical inference difficult.

Matérn (1960, 1986) proposed a set of techniques for thinning Poisson point processes in order to induce a type of repulsion when the items are embedded in a Euclidean space. The Type I process is obtained from a Poisson set Y\boldsymbol{Y} by removing all items in Y\boldsymbol{Y} that lie within some radius of another item in Y\boldsymbol{Y}. That is, if two items are close to each other, they are both removed; as a result all items in the final process are spaced at least a fixed distance apart. The Type II Matérn repulsive process, designed to achieve the same minimum distance property while keeping more items, begins by independently assigning each item in Y\boldsymbol{Y} a uniformly random “time” in $$. Then, any item within a given radius of a point having a smaller time value is removed. Under this construction, when two items are close to each other only the later one is removed. Still, an item may be removed due to its proximity with an earlier item that was itself removed. This leads to the Type III process, which proceeds dynamically, eliminating items in time order whenever an earlier point which has not been removed lies within the radius.

Inference for the Matérn processes is computationally daunting. First and second order moments can be computed for Types I and II, but in those cases computing the likelihood of a set YY is seemingly intractable (Møller et al., 2010). Recent work by Huber and Wolpert (2009) shows that it is possible to compute likelihood for certain restricted Type III processes, but computing moments cannot be done in closed form. In the general case, likelihood for Type III processes must be estimated using an expensive Markov chain Monte Carlo algorithm.

The Matérn processes are called “hard-core” because they strictly enforce a minimum radius between selected items. While this property leads to one kind of diversity, it is somewhat limited, and due to the procedural definition it is difficult to characterize the side effects of the thinning process in a general way. Stoyan and Stoyan (1985) considered an extension where the radius is itself chosen randomly, which may be more natural for certain settings, but it does not alleviate the computational issues.

The Matérn repulsive processes are related in spirit to the random sequential adsorption (RSA) model, which has been used in physics and chemistry to model particles that bind to two-dimensional surfaces, e.g., proteins on a cell membrane (Tanemura, 1979; Finegold and Donnell, 1979; Feder, 1980; Swendsen, 1981; Hinrichsen et al., 1986; Ramsden, 1993). RSA is described generatively as follows. Initially, the surface is empty; iteratively, particles arrive and bind uniformly at random to a location from among all locations that are not within a given radius of any previously bound particle. When no such locations remain (the “jamming limit”), the process is complete.

Like the Matérn processes, RSA is a hard-core model, designed primarily to capture packing distributions, with much of the theoretical analysis focused on the achievable density. If the set of locations is further restricted at each step to those found in an initially selected Poisson set Y\boldsymbol{Y}, then it is equivalent to a Matérn Type III process (Huber and Wolpert, 2009); it therefore shares the same computational burdens.

5.2 Gibbs and Markov point processes

While manipulating the Poisson process procedurally has some intuitive appeal, it seems plausible that a more holistically-defined process might be easier to work with, both analytically and algorithmically. The Gibbs point process provides such an approach, offering a general framework for incorporating correlations among selected items (Preston, 1976; Ripley and Kelly, 1977; Ripley, 1991; Van Lieshout, 2000; Møller and Waagepetersen, 2004, 2007; Daley and Vere-Jones, 2008). The Gibbs probability of a set YY is given by

where UU is an energy function. Of course, this definition is fully general without further constraints on UU. A typical assumption is that UU decomposes over subsets of items in YY; for instance

for some small constant order kk and potential functions ψ\psi. In practice, the most common case is k=2k=2, which is sometimes called a pairwise interaction point process (Diggle et al., 1987):

In spatial settings, a Gibbs point process whose potential functions are identically 1 whenever their input arguments do not lie within a ball of fixed radius—that is, whose energy function can be decomposed into only local terms—is called a Markov point process. A number of specific Markov point processes have become well-known.

Strauss (1975) introduced a simple pairwise Markov point process for spatial data in which the potential function ψ2(i,j)\psi_{2}(i,j) is piecewise constant, taking the value 1 whenever ii and jj are at least a fixed radius apart, and the constant value γ\gamma otherwise. When γ>1\gamma>1, the resulting process prefers clustered items. (Note that γ>1\gamma>1 is only possible in the discrete case; in the continuous setting the distribution becomes non-integrable.) We are more interested in the case 0<γ<10<\gamma<1, where configurations in which selected items are near one another are discounted. When γ=0\gamma=0, the resulting process becomes hard-core, but in general the Strauss process is “soft-core”, preferring but not requiring diversity.

The Strauss process is typical of pairwise Markov processes in that its potential function ψ2(i,j)=ψ(ij)\psi_{2}(i,j)=\psi(|i-j|) depends only on the distance between its arguments. A variety of alternative definitions for ψ()\psi(\cdot) have been proposed (Ripley and Kelly, 1977; Ogata and Tanemura, 1984). For instance,

where σ\sigma controls the degree of repulsion in each case. Each definition leads to a point process with a slightly different concept of diversity.

Baddeley and Van Lieshout (1995) proposed a non-pairwise spatial Markov point process called the area-interaction model, where U(Y)U(Y) is given by logγ\log\gamma times the total area contained in the union of discs of fixed radius centered at all of the items in YY. When γ>1\gamma>1, we have logγ>0\log\gamma>0 and the process prefers sets whose discs cover as little area as possible, i.e., whose items are clustered. When 0<γ<10<\gamma<1, logγ\log\gamma becomes negative, so the process prefers “diverse” sets covering as much area as possible.

If none of the selected items fall within twice the disc radius of each other, then exp(U(Y))\exp(-U(Y)) can be decomposed into potential functions over single items, since the total area is simply the sum of the individual discs. Similarly, if each disc intersects with at most one other disc, the area-interaction process can be written as a pairwise interaction model. However, in general, an unbounded number of items might appear in a given disc; as a result the area-interaction process is an infinite-order Gibbs process. Since items only interact when they are near one another, however, local potential functions are sufficient and the process is Markov.

Markov point processes have many intuitive properties. In fact, it is not difficult to see that, for discrete ground sets Y\mathcal{Y}, the Markov point process is equivalent to a Markov random field (MRF) on binary variables corresponding to the elements of Y\mathcal{Y}. In Section 3.2.2 we will return to this equivalence in order to discuss the relative expressive possibilities of DPPs and MRFs. For now, however, we simply observe that, as for MRFs with negative correlations, repulsive Markov point processes are computationally intractable. Even computing the normalizing constant for Equation (84) is NP-hard in the cases outlined above (Daley and Vere-Jones, 2003; Møller and Waagepetersen, 2004).

On the other hand, quite a bit of attention has been paid to approximate inference algorithms for Markov point processes, employing pseudolikelihood (Besag, 1977; Besag et al., 1982; Jensen and Moller, 1991; Ripley, 1991), Markov chain Monte Carlo methods (Ripley and Kelly, 1977; Besag and Green, 1993; Häggström et al., 1999; Berthelsen and Møller, 2006), and other approximations (Ogata and Tanemura, 1985; Diggle et al., 1994). Nonetheless, in general these methods are slow and/or inexact, and closed-form expressions for moments and densities rarely exist (Møller and Waagepetersen, 2007). In this sense the DPP is unique.

5.3 Generalizations of determinants

The determinant of a k×kk\times k matrix KK can be written as a polynomial of degree kk in the entries of KK; in particular,

Computationally, obtaining the permanent of a matrix is #P-complete (Valiant, 1979), making the permanental process difficult to work with in practice. Complexity results for immanants are less definitive, with certain classes of immanants apparently hard to compute (Bürgisser, 2000; Brylinski and Brylinski, 2003), while some upper bounds on complexity are known (Hartmann, 1985; Barvinok, 1990), and at least one non-trivial case is efficiently computable (Grone and Merris, 1984). It is not clear whether the latter result provides enough leverage to perform inference beyond computing marginals.

A third possible generalization of Equation (90) is the hyperdeterminant originally proposed by Cayley (1843) and discussed in the context of point processes by Evans and Gottlieb (2009). Whereas the standard determinant operates on a two-dimensional matrix with entries indexed by pairs of items, the hyperdeterminant operates on higher-dimensional kernel matrices indexed by sets of items. The hyperdeterminant potentially offers additional modeling power, and Evans and Gottlieb (2009) show that some useful properties of DPPs are preserved in this setting. However, so far relatively little is known about these processes.

5.4 Quasirandom processes

Monte Carlo methods rely on draws of random points in order to approximate quantities of interest; randomness guarantees that, regardless of the function being studied, the estimates will be accurate in expectation and converge in the limit. However, in practice we get to observe only a finite set of values drawn from the random source. If, by chance, this set is “bad”, the resulting estimate may be poor. This concern has led to the development of so-called quasirandom sets, which are in fact deterministically generated, but can be substituted for random sets in some instances to obtain improved convergence guarantees (Niederreiter, 1992; Sobol, 1998).

In contrast with pseudorandom generators, which attempt to mimic randomness by satisfying statistical tests that ensure unpredictability, quasirandom sets are not designed to appear random, and their elements are not (even approximately) independent. Instead, they are designed to have low discrepancy; roughly speaking, low-discrepancy sets are “diverse” in that they cover the sample space evenly. Consider a finite subset YY of D^{D}, with elements x(i)=(x1(i),x2(i),,xD(i))\boldsymbol{x}^{(i)}=(x^{(i)}_{1},x^{(i)}_{2},\dots,x^{(i)}_{D}) for i=1,2,,ki=1,2,\dots,k. Let Sx=[0,x1)×[0,x2)××[0,xD)S_{\boldsymbol{x}}=[0,x_{1})\times[0,x_{2})\times\cdots\times[0,x_{D}) denote the box defined by the origin and the point x\boldsymbol{x}. The discrepancy of YY is defined as follows.

This deterministic uniformity property makes quasirandom sets useful for Monte Carlo estimation via (among other results) the Koksma-Hlawka inequality (Hlawka, 1961; Niederreiter, 1992). For a function ff with bounded variation V(f)V(f) on the unit cube, the inequality states that

Thus, low-discrepancy sets lead to accurate quasi-Monte Carlo estimates. In contrast to typical Monte Carlo guarantees, the Koksma-Hlawka inequality is deterministic. Moreover, since the rate of convergence for standard stochastic Monte Carlo methods is k1/2k^{-1/2}, this result is an (asymptotic) improvement when the discrepancy diminishes faster than k1/2k^{-1/2}.

In fact, it is possible to construct quasirandom sequences where the discrepancy of the first kk elements is O((logk)D/k)O((\log k)^{D}/k); the first such sequence was proposed by Halton (1960). The Sobol sequence (Sobol, 1967), introduced later, offers improved uniformity properties and can be generated efficiently (Bratley and Fox, 1988).

It seems plausible that, due to their uniformity characteristics, low-discrepancy sets could be used as computationally efficient but non-probabilistic tools for working with data exhibiting diversity. An algorithm generating quasirandom sets could be seen as an efficient prediction procedure if made to depend somehow on input data and a set of learned parameters. However, to our knowledge no work has yet addressed this possibility.

Representation and algorithms

Determinantal point processes come with a deep and beautiful theory, and, as we have seen, exactly characterize many theoretical processes. However, they are also promising models for real-world data that exhibit diversity, and we are interested in making such applications as intuitive, practical, and computationally efficient as possible. In this section, we present a variety of fundamental techniques and algorithms that serve these goals and form the basis of the extensions we discuss later.

We begin by describing a decomposition of the DPP kernel that offers an intuitive tradeoff between a unary model of quality over the items in the ground set and a global model of diversity. The geometric intuitions from Section 2 extend naturally to this decomposition. Splitting the model into quality and diversity components then allows us to make a comparative study of expressiveness—that is, the range of distributions that the model can describe. We compare the expressive powers of DPPs and negative-interaction Markov random fields, showing that the two models are incomparable in general but exhibit qualitatively similar characteristics, despite the computational advantages offered by DPPs.

Next, we turn to the challenges imposed by large data sets, which are common in practice. We first address the case where NN, the number of items in the ground set, is very large. In this setting, the super-linear number of operations required for most DPP inference algorithms can be prohibitively expensive. However, by introducing a dual representation of a DPP we show that efficient DPP inference remains possible when the kernel is low-rank. When the kernel is not low-rank, we prove that a simple approximation based on random projections dramatically speeds inference while guaranteeing that the deviation from the original distribution is bounded. These techniques will be especially useful in Section 6, when we consider exponentially large NN.

Finally, we discuss some alternative formulas for the likelihood of a set YY in terms of the marginal kernel KK. Compared to the L-ensemble formula in Equation (19), these may be analytically more convenient, since they do not involve ratios or arbitrary principal minors.

An important practical concern for modeling is interpretability; that is, practitioners should be able to understand the parameters of the model in an intuitive way. While the entries of the DPP kernel are not totally opaque in that they can be seen as measures of similarity—reflecting our primary qualitative characterization of DPPs as diversifying processes—in most practical situations we want diversity to be balanced against some underlying preferences for different items in Y\mathcal{Y}. In this section, we propose a decomposition of the DPP that more directly illustrates the tension between diversity and a per-item measure of quality.

This decomposition of LL has two main advantages. First, it implicitly enforces the constraint that LL must be positive semidefinite, which can potentially simplify learning (see Section 4). Second, it allows us to independently model quality and diversity, and then combine them into a unified model. In particular, we have:

where the first term increases with the quality of the selected items, and the second term increases with the diversity of the selected items. We will refer to qq as the quality model and SS or ϕ\phi as the diversity model. Without the diversity model, we would choose high-quality items, but we would tend to choose similar high-quality items over and over. Without the quality model, we would get a very diverse set, but we might fail to include the most important items in Y\mathcal{Y}, focusing instead on low-quality outliers. By combining the two models we can achieve a more balanced result.

Returning to the geometric intuitions from Section 2.2.1, the determinant of LYL_{Y} is equal to the squared volume of the parallelepiped spanned by the vectors qiϕiq_{i}\phi_{i} for iYi\in Y. The magnitude of the vector representing item ii is qiq_{i}, and its direction is ϕi\phi_{i}. Figure 9 (reproduced from the previous section) now makes clear how DPPs decomposed in this way naturally balance the two objectives of high quality and high diversity. Going forward, we will nearly always assume that our models are decomposed into quality and diversity components. This provides us not only with a natural and intuitive setup for real-world applications, but also a useful perspective for comparing DPPs with existing models, which we turn to next.

2 Expressive power

Many probabilistic models are known and widely used within the machine learning community. A natural question, therefore, is what advantages DPPs offer that standard models do not. We have seen already how a large variety of inference tasks, like sampling and conditioning, can be performed efficiently for DPPs; however efficiency is essentially a prerequisite for any practical model. What makes DPPs particularly unique is the marriage of these computational advantages with the ability to express global, negative interactions between modeling variables; this repulsive domain is notoriously intractable using traditional approaches like graphical models (Murphy et al., 1999; Boros and Hammer, 2002; Ishikawa, 2003; Taskar et al., 2004; Yanover and Weiss, 2002; Yanover et al., 2006; Kulesza and Pereira, 2008). In this section we elaborate on the expressive powers of DPPs and compare them with those of Markov random fields, which we take as representative graphical models.

A Markov random field (MRF) is an undirected graphical model defined by a graph GG whose nodes 1,2,,N1,2,\dots,N represent random variables. For our purposes, we will consider binary MRFs, where each output variable takes a value from {0,1}\{0,1\}. We use yiy_{i} to denote a value of the iith output variable, bold yc\boldsymbol{y}_{c} to denote an assignment to a set of variables cc, and y\boldsymbol{y} for an assignment to all of the output variables. The graph edges EE encode direct dependence relationships between variables; for example, there might be edges between similar elements ii and jj to represent the fact that they tend not to co-occur. MRFs are often referred to as conditional random fields when they are parameterized to depend on input data, and especially when GG is a chain (Lafferty et al., 2001).

An MRF defines a joint probability distribution over the output variables that factorizes across the cliques C\mathcal{C} of GG:

Here each ψc\psi_{c} is a potential function that assigns a nonnegative value to every possible assignment yc\boldsymbol{y}_{c} of the clique cc, and ZZ is the normalization constant ycCψc(yc)\sum_{\boldsymbol{y}^{\prime}}\prod_{c\in\mathcal{C}}\psi_{c}(\boldsymbol{y}^{\prime}_{c}). Note that, for a binary MRF, we can think of y\boldsymbol{y} as the characteristic vector for a subset YY of Y={1,2,,N}\mathcal{Y}=\{1,2,\dots,N\}. Then the MRF is equivalently the distribution of a random subset Y\boldsymbol{Y}, where P(Y=Y)\mathcal{P}(\boldsymbol{Y}=Y) is equivalent to P(y)\mathcal{P}(\boldsymbol{y}).

The Hammersley-Clifford theorem states that P(y)\mathcal{P}(\boldsymbol{y}) defined in Equation (96) is always Markov with respect to GG; that is, each variable is conditionally independent of all other variables given its neighbors in GG. The converse also holds: any distribution that is Markov with respect to GG, as long as it is strictly positive, can be decomposed over the cliques of GG as in Equation (96) (Grimmett, 1973). MRFs therefore offer an intuitive way to model problem structure. Given domain knowledge about the nature of the ways in which outputs interact, a practitioner can construct a graph that encodes a precise set of conditional independence relations. (Because the number of unique assignments to a clique cc is exponential in c|c|, computational constraints generally limit us to small cliques.)

For comparison with DPPs, we will focus on pairwise MRFs, where the largest cliques with interesting potential functions are the edges; that is, ψc(yc)=1\psi_{c}(\boldsymbol{y}_{c})=1 for all cliques cc where c>2|c|>2. The pairwise distribution is

We refer to ψi(yi)\psi_{i}(y_{i}) as node potentials, and ψij(yi,yj)\psi_{ij}(y_{i},y_{j}) as edge potentials.

MRFs are very general models—in fact, if the cliques are unbounded in size, they are fully general—but inference is only tractable in certain special cases. Cooper (1990) showed that general probabilistic inference (conditioning and marginalization) in MRFs is NP-hard, and this was later extended by Dagum and Luby (1993), who showed that inference is NP-hard even to approximate. Shimony (1994) proved that the MAP inference problem (finding the mode of an MRF) is also NP-hard, and Abdelbar and Hedetniemi (1998) showed that the MAP problem is likewise hard to approximate. In contrast, we showed in Section 2 that DPPs offer efficient exact probabilistic inference; furthermore, although the MAP problem for DPPs is NP-hard, it can be approximated to a constant factor under cardinality constraints in polynomial time.

The first tractable subclass of MRFs was identified by Pearl (1982), who showed that belief propagation can be used to perform inference in polynomial time when GG is a tree. More recently, certain types of inference in binary MRFs with associative (or submodular) potentials ψ\psi have been shown to be tractable (Boros and Hammer, 2002; Taskar et al., 2004; Kolmogorov and Zabih, 2004). Inference in non-binary associative MRFs is NP-hard, but can be efficiently approximated to a constant factor depending on the size of the largest clique (Taskar et al., 2004). Intuitively, an edge potential is called associative if it encourages the endpoint nodes take the same value (e.g., to be both in or both out of the set YY). More formally, associative potentials are at least one whenever the variables they depend on are all equal, and exactly one otherwise.

We can rewrite the pairwise, binary MRF of Equation (97) in a canonical log-linear form:

Here we have eliminated redundancies by forcing ψi(0)=1\psi_{i}(0)=1, ψij(0,0)=ψij(0,1)=ψij(1,0)=1\psi_{ij}(0,0)=\psi_{ij}(0,1)=\psi_{ij}(1,0)=1, and setting wi=logψi(1)w_{i}=\log\psi_{i}(1), wij=logψij(1,1)w_{ij}=\log\psi_{ij}(1,1). This parameterization is sometimes called the fully visible Boltzmann machine. Under this representation, the MRF is associative whenever wij0w_{ij}\geq 0 for all ijEij\in E.

We have seen that inference in MRFs is tractable when we restrict the graph to a tree or require the potentials to encourage agreement. However, the repulsive potentials necessary to build MRFs exhibiting diversity are the opposite of associative potentials (since they imply wij<0w_{ij}<0), and lead to intractable inference for general graphs. Indeed, such negative potentials can create “frustrated cycles”, which have been used both as illustrations of common MRF inference algorithm failures (Kulesza and Pereira, 2008) and as targets for improving those algorithms (Sontag and Jaakkola, 2007). A wide array of (informally) approximate inference algorithms have been proposed to mitigate tractability problems, but none to our knowledge effectively and reliably handles the case where potentials exhibit strong repulsion.

2.2 Comparing DPPs and MRFs

Despite the computational issues outlined in the previous section, MRFs are popular models and, importantly, intuitive for practitioners, both because they are familiar and because their potential functions directly model simple, local relationships. We argue that DPPs have a similarly intuitive interpretation using the decomposition in Section 3.1. Here, we compare the distributions realizable by DPPs and MRFs to see whether the tractability of DPPs comes at a large expressive cost.

Consider a DPP over Y={1,2,,N}\mathcal{Y}=\{1,2,\dots,N\} with N×NN\times N kernel matrix LL decomposed as in Section 3.1; we have

Both of these models can capture negative correlations between indicator variables yiy_{i}. Both models also have N(N+1)2\frac{N(N+1)}{2} parameters: the DPP has quality scores qiq_{i} and similarity measures SijS_{ij}, and the MRF has node log-potentials wiw_{i} and edge log-potentials wijw_{ij}. The key representational difference is that, while wijw_{ij} are individually constrained to be nonpositive, the positive semidefinite constraint on the DPP kernel is global. One consequence is that, as a side effect, the MRF can actually capture certain limited positive correlations; for example, a 3-node MRF with w12,w13<0w_{12},w_{13}<0 and w23=0w_{23}=0 induces a positive correlation between nodes two and three by virtue of their mutual disagreement with node one. As we have seen in Section 2, the semidefinite constraint prevents the DPP from forming any positive correlations.

More generally, semidefiniteness means that the DPP diversity feature vectors must satisfy the triangle inequality, leading to

for all i,j,kYi,j,k\in\mathcal{Y} since ϕiϕj1Sij\|\phi_{i}-\phi_{j}\|\propto\sqrt{1-S_{ij}}. The similarity measure therefore obeys a type of transitivity, with large SijS_{ij} and SjkS_{jk} implying large SikS_{ik}.

Equation (101) is not, by itself, sufficient to guarantee that LL is positive semidefinite, since SS must also be realizable using unit length feature vectors. However, rather than trying to develop further intuition algebraically, we turn to visualization. While it is difficult to depict the feasible distributions of DPPs and MRFs in high dimensions, we can get a feel for their differences even with a small number of elements NN.

When N=2N=2, it is easy to show that the two models are equivalent, in the sense that they can both represent any distribution with negative correlations:

When N=3N=3, differences start to become apparent. In this setting both models have six parameters: for the DPP they are (q1,q2,q3,S12,S13,S23)(q_{1},q_{2},q_{3},S_{12},S_{13},S_{23}), and for the MRF they are (w1,w2,w3,w12,w13,w23)(w_{1},w_{2},w_{3},w_{12},w_{13},w_{23}). To place the two models on equal footing, we represent each as the product of unnormalized per-item potentials ψ1,ψ2,ψ3\psi_{1},\psi_{2},\psi_{3} and a single unnormalized ternary potential ψ123\psi_{123}. This representation corresponds to a factor graph with three nodes and a single, ternary factor (see Figure 10). The probability of a set YY is then given by

Figure 11a depicts four such slices of the realizable DPP distributions, and Figure 11b shows the same slices of the realizable MRF distributions. Points closer to the origin (on the lower left) correspond to “more repulsive” distributions, where the three elements of Y\mathcal{Y} are less likely to appear together. When ψ123(1,1,1)\psi_{123}(1,1,1) is large (gray surfaces), negative correlations are weak and the two models give rise to qualitatively similar distributions. As the value of the ψ123(1,1,1)\psi_{123}(1,1,1) shrinks to zero (red surfaces), the two models become quite different. MRFs, for example, can describe distributions where the first item is strongly anti-correlated with both of the others, but the second and third are not anti-correlated with each other. The transitive nature of the DPP makes this impossible.

To improve visibility, we have constrained S12,S13,S230S_{12},S_{13},S_{23}\geq 0 in Figure 11a. Figure 11c shows a single slice without this constraint; allowing negative similarity makes it possible to achieve strong three-way repulsion with less pairwise repulsion, closing the surface away from the origin. The corresponding MRF slice is shown in Figure 11d, and the two are overlaid in Figure 11e and Figure 11f. Even though there are relatively strong interactions in these plots (ψ123(1,1,1)=0.1\psi_{123}(1,1,1)=0.1), the models remain roughly comparable in terms of the distributions they can express.

As NN gets larger, we conjecture that the story is essentially the same. DPPs are primarily constrained by a notion of transitivity on the similarity measure; thus it would be difficult to use a DPP to model, for example, data where items repel “distant” items rather than similar items—if ii is far from jj and jj is far from kk we cannot necessarily conclude that ii is far from kk. One way of looking at this is that repulsion of distant items induces positive correlations between the selected items, which a DPP cannot represent.

MRFs, on the other hand, are constrained by their local nature and do not effectively model data that are “globally” diverse. For instance, a pairwise MRF we cannot exclude a set of three or more items without excluding some pair of those items. More generally, an MRF assumes that repulsion does not depend on (too much) context, so it cannot express that, say, there can be only a certain number of selected items overall. The DPP can naturally implement this kind of restriction though the rank of the kernel.

3 Dual representation

The standard inference algorithms for DPPs rely on manipulating the kernel LL through inversion, eigendecomposition, and so on. However, in situations where NN is large we may not be able to work efficiently with LL—in some cases we may not even have the memory to write it down. In this section, instead, we develop a dual representation of a DPP that shares many important properties with the kernel LL but is often much smaller. Afterwards, we will show how this dual representation can be applied to perform efficient inference.

Let BB be the D×ND\times N matrix whose columns are given by Bi=qiϕiB_{i}=q_{i}\phi_{i}, so that L=BBL=B^{\top}B. Consider now the matrix

By construction, CC is symmetric and positive semidefinite. In contrast to LL, which is too expensive to work with when NN is large, CC is only D×DD\times D, where DD is the dimension of the diversity feature function ϕ\phi. In many practical situations, DD is fixed by the model designer, while NN may grow without bound as new items become available; for instance, a search engine may continually add to its database of links. Furthermore, we have the following result.

The nonzero eigenvalues of CC and LL are identical, and the corresponding eigenvectors are related by the matrix BB. That is,

is an eigendecomposition of CC if and only if

In the forward direction, we assume that {(λn,v^n)}n=1D\{(\lambda_{n},\boldsymbol{\hat{v}}_{n})\}_{n=1}^{D} is an eigendecomposition of CC. We have

since v^n\boldsymbol{\hat{v}}_{n} are orthonormal by assumption. Furthermore, for any nn we have

using the fact that Cv^n=λnv^nC\boldsymbol{\hat{v}}_{n}=\lambda_{n}\boldsymbol{\hat{v}}_{n} since v^n\boldsymbol{\hat{v}}_{n} is an eigenvector of CC. Finally, for any distinct 1a,bD1\leq a,b\leq D, we have

Thus {(λn,1λnBv^n)}n=1D\left\{\left(\lambda_{n},\frac{1}{\sqrt{\lambda_{n}}}B^{\top}\boldsymbol{\hat{v}}_{n}\right)\right\}_{n=1}^{D} is an eigendecomposition of LL. In the other direction, an analogous argument applies once we observe that, since L=BBL=B^{\top}B, LL has rank at most DD and therefore at most DD nonzero eigenvalues. ∎

Proposition 3.1 shows that CC contains quite a bit of information about LL. In fact, CC is sufficient to perform nearly all forms of DPP inference efficiently, including normalization and marginalization in constant time with respect to NN, and sampling in time linear in NN.

Recall that the normalization constant for a DPP is given by det(L+I)\det(L+I). If λ1,λ2,,λN\lambda_{1},\lambda_{2},\dots,\lambda_{N} are the eigenvalues of LL, then the normalization constant is equal to n=1N(λn+1)\prod_{n=1}^{N}(\lambda_{n}+1), since the determinant is the product of the eigenvalues of its argument. By Proposition 3.1, the nonzero eigenvalues of LL are also the eigenvalues of the dual representation CC. Thus, we have

Computing the determinant of C+IC+I requires O(Dω)O(D^{\omega}) time.

3.2 Marginalization

Standard DPP marginalization makes use of the marginal kernel KK, which is of course as large as LL. However, the dual representation CC can be used to compute the entries of KK. We first eigendecompose the dual representation as C=n=1Dλnv^nv^nC=\sum_{n=1}^{D}\lambda_{n}\boldsymbol{\hat{v}}_{n}\boldsymbol{\hat{v}}_{n}^{\top}, which requires O(Dω)O(D^{\omega}) time. Then, we can use the definition of KK in terms of the eigendecomposition of LL as well as Proposition 3.1 to compute

That is, the diagonal entries of KK are computable from the dot products between the diversity features ϕi\phi_{i} and the eigenvectors of CC; we can therefore compute the marginal probability of a single item iYi\in\mathcal{Y} from an eigendecomposition of CC in O(D2)O(D^{2}) time. Similarly, given two items ii and jj we have

so we can compute arbitrary entries of KK in O(D2)O(D^{2}) time. This allows us to compute, for example, pairwise marginals P(i,jY)=KiiKjjKij2\mathcal{P}(i,j\in\boldsymbol{Y})=K_{ii}K_{jj}-K_{ij}^{2}. More generally, for a set AYA\in\mathcal{Y}, A=k|A|=k, we need to compute k(k+1)2\frac{k(k+1)}{2} entries of KK to obtain KAK_{A}, and taking the determinant then yields P(AY)\mathcal{P}(A\subseteq\boldsymbol{Y}). The process requires only O(D2k2+kω)O(D^{2}k^{2}+k^{\omega}) time; for small sets A|A| this is just quadratic in the dimension of ϕ\phi.

3.3 Sampling

Note that, when V^\hat{V} contains eigenvectors of CC, this is (up to scale) the relationship established by Proposition 3.1 between eigenvectors v^\boldsymbol{\hat{v}} of CC and eigenvectors v\boldsymbol{v} of LL.

The mapping in Equation (123) has several useful properties. If v1=Bv^1\boldsymbol{v}_{1}=B^{\top}\boldsymbol{\hat{v}}_{1} and v2=Bv^2\boldsymbol{v}_{2}=B^{\top}\boldsymbol{\hat{v}}_{2}, then v1+v2=B(v^1+v^2)\boldsymbol{v}_{1}+\boldsymbol{v}_{2}=B^{\top}(\boldsymbol{\hat{v}}_{1}+\boldsymbol{\hat{v}}_{2}), and likewise for any arbitrary linear combination. In other words, we can perform implicit scaling and addition of the vectors in VV using only their preimages in V^\hat{V}. Additionally, we have

so we can compute dot products of vectors in VV in O(D2)O(D^{2}) time. This means that, for instance, we can implicitly normalize v=Bv^\boldsymbol{v}=B^{\top}\boldsymbol{\hat{v}} by updating v^v^v^Cv^\boldsymbol{\hat{v}}\leftarrow\frac{\boldsymbol{\hat{v}}}{\sqrt{\boldsymbol{\hat{v}}^{\top}C\boldsymbol{\hat{v}}}}.

We now show how these operations allow us to efficiently implement key parts of the sampling algorithm. Because the nonzero eigenvalues of LL and CC are equal, the first loop of the algorithm, where we choose in index set JJ, remains unchanged. Rather than using JJ to construct orthonormal VV directly, however, we instead build the set V^\hat{V} by adding v^nv^nCv^n\frac{\boldsymbol{\hat{v}}_{n}}{\sqrt{\boldsymbol{\hat{v}}_{n}^{\top}C\boldsymbol{\hat{v}}_{n}}} for every nJn\in J.

In the last phase of the loop, we need to find an orthonormal basis VV_{\bot} for the subspace of VV orthogonal to a given ei\boldsymbol{e}_{i}. This requires two steps. In the first, we subtract a multiple of one of the vectors in VV from all of the other vectors so that they are zero in the iith component, leaving us with a set of vectors spanning the subspace of VV orthogonal to ei\boldsymbol{e}_{i}. In order to do this we must be able to compute the iith component of a vector vV\boldsymbol{v}\in V; since v=Bv^\boldsymbol{v}=B^{\top}\boldsymbol{\hat{v}}, this is easily done by computing the iith column of BB, and then taking the dot product with v^\boldsymbol{\hat{v}}. This takes only O(D)O(D) time. In the second step, we use the Gram-Schmidt process to convert the resulting vectors into an orthonormal set. This requires a series of dot products, sums, and scalings of vectors in VV; however, as previously argued all of these operations can be performed implicitly. Therefore the mapping in Equation (123) allows us to implement the final line of the second loop using only tractable computations on vectors in V^\hat{V}.

All that remains, then, is to efficiently choose an item ii according to the distribution

in the first line of the while loop. Simplifying, we have

Thus the required distribution can be computed in time O(NDk)O(NDk), where k=V^k=|\hat{V}|. The complete dual sampling algorithm is given in Algorithm 3; the overall runtime is O(NDk2+D2k3)O(NDk^{2}+D^{2}k^{3}).

4 Random projections

As we have seen, dual DPPs allow us to deal with settings where NN is too large to work efficiently with LL by shifting the computational focus to the dual kernel CC, which is only D×DD\times D. This is an effective approach when DND\ll N. Of course, in some cases DD might also be unmanageably large, for instance when the diversity features are given by word counts in natural language settings, or high-resolution image features in vision.

To address this problem, we describe a method for reducing the dimension of the diversity features while maintaining a close approximation to the original DPP model. Our approach is based on applying random projections, an extremely simple technique that nonetheless provides an array of theoretical guarantees, particularly with respect to preserving distances between points (Vempala, 2004). A classic result of Johnson and Lindenstrauss (1984), for instance, shows that high-dimensional points can be randomly projected onto a logarithmic number of dimensions while approximately preserving the distances between them. More recently, Magen and Zouzias (2008) extended this idea to the preservation of volumes spanned by sets of points. Here, we apply the connection between DPPs and spanned volumes to show that random projections allow us to reduce the dimensionality of ϕ\phi, dramatically speeding up inference, while maintaining a provably close approximation to the original, high-dimensional model. We begin by stating a variant of Magen and Zouzias’ result.

(Adapted from Magen and Zouzias (2008)) Let XX be a D×ND\times N matrix. Fix k<Nk<N and 0<ϵ,δ<1/20<\epsilon,\delta<1/2, and set the projection dimension

Let GG be a d×Dd\times D random projection matrix whose entries are independently sampled from N(0,1d)\mathcal{N}(0,\frac{1}{d}), and let XYX_{Y}, where Y{1,2,,N}Y\subseteq\{1,2,\dots,N\}, denote the D×YD\times|Y| matrix formed by taking the columns of XX corresponding to indices in YY. Then with probability at least 1δ1-\delta we have, for all YY with cardinality at most kk:

Lemma 3.2 says that, with high probability, randomly projecting to

dimensions is sufficient to approximately preserve all volumes spanned by kk columns of XX. We can use this result to bound the effectiveness of random projections for DPPs.

In order to obtain a result that is independent of DD, we will restrict ourselves to the portion of the distribution pertaining to subsets YY with cardinality at most a constant kk. This restriction is intuitively reasonable for any application where we use DPPs to model sets of relatively small size compared to NN, which is common in practice. However, formally it may seem a bit strange, since it implies conditioning the DPP on cardinality. In Section 5 we will show that this kind of conditioning is actually very practical and efficient, and Theorem 3.3, which we prove shortly, will apply directly to the kk-DPPs of Section 5 without any additional work.

For now, we will seek to approximate the distribution Pk(Y)=P(Y=Y  Yk)\mathcal{P}^{\leq k}(Y)=\mathcal{P}(\boldsymbol{Y}=Y\ |\ |\boldsymbol{Y}|\leq k), which is simply the original DPP conditioned on the cardinality of the modeled subset:

where ϕ(Y)\phi(Y) denotes the D×YD\times|Y| matrix formed from columns ϕi\phi_{i} for iYi\in Y. Our main result follows.

Let Pk(Y)\mathcal{P}^{\leq k}(Y) be the cardinality-conditioned DPP distribution defined by quality model qq and DD-dimensional diversity feature function ϕ\phi, and let

be the cardinality-conditioned DPP distribution obtained by projecting ϕ\phi with GG. Then for projection dimension dd as in Equation (129), we have

with probability at least 1δ1-\delta. Note that e6kϵ16kϵe^{6k\epsilon}-1\approx 6k\epsilon when kϵk\epsilon is small.

The theorem says that for dd logarithmic in NN and linear in kk, the L1L_{1} variational distance between the original DPP and the randomly projected version is bounded. In order to use Lemma 3.2, which bounds volumes of parallelepipeds, to prove this bound on determinants, we will make use of the following relationship:

In order to handle the conditional DPP normalization constant

we also must adapt Lemma 3.2 to sums of determinants. Finally, for technical reasons we will change the symmetry of the upper and lower bounds from the sign of ϵ\epsilon to the sign of the exponent. The following lemma gives the details.

Under the definitions and assumptions of Lemma 3.2, we have, with probability at least 1δ1-\delta,

where the first inequality holds with probability at least 1δ1-\delta by Lemma 3.2, and the third follows from the fact that (1ϵ)(1+2ϵ)1(1-\epsilon)(1+2\epsilon)\geq 1 (since ϵ<1/2\epsilon<1/2), thus (1ϵ)2k(1+2ϵ)2k(1-\epsilon)^{2k}\geq(1+2\epsilon)^{-2k}. The upper bound follows directly:

Recall the matrix BB, whose columns are given by Bi=qiϕiB_{i}=q_{i}\phi_{i}. We have

where the first inequality follows from Lemma 3.2 and Lemma 137, which hold simultaneously with probability at least 1δ1-\delta, and the second follows from (1+a)beab(1+a)^{b}\leq e^{ab} for a,b0a,b\geq 0. ∎

By combining the dual representation with random projections, we can deal simultaneously with very large NN and very large DD. In fact, in Section 6 we will show that NN can even be exponentially large if certain structural assumptions are met. These techniques vastly expand the range of problems to which DPPs can be practically applied.

5 Alternative likelihood formulas

Recall that, in an L-ensemble DPP, the likelihood of a particular set YYY\subseteq\mathcal{Y} is given by

This expression has some nice intuitive properties in terms of volumes, and, ignoring the normalization in the denominator, takes a simple and concise form. However, as a ratio of determinants on matrices of differing dimension, it may not always be analytically convenient. Minors can be difficult to reason about directly, and ratios complicate calculations like derivatives. Moreover, we might want the likelihood in terms of the marginal kernel K=L(L+I)1=I(L+I)1K=L(L+I)^{-1}=I-(L+I)^{-1}, but simply plugging in these identities yields a expression that is somewhat unwieldy.

As alternatives, we will derive some additional formulas that, depending on context, may have useful advantages. Our starting point will be the observation, used previously in the proof of Theorem 21, that minors can be written in terms of full matrices and diagonal indicator matrices; specifically, for positive semidefinite LL,

where IYI_{Y} is the diagonal matrix with ones in the diagonal positions corresponding to elements of YY and zeros everywhere else, and Yˉ=YY\bar{Y}=\mathcal{Y}-Y. These identities can be easily shown by examining the matrices blockwise under the partition Y=YYˉ\mathcal{Y}=Y\cup\bar{Y}, as in the proof of Theorem 21.

Applying Equation (149) to Equation (148), we get

Already, this expression, which is a single determinant of an N×NN\times N matrix, is in some ways easier to work with. We can also more easily write the likelihood in terms of KK:

Recall from Equation (35) that IKI-K is the marginal kernel of the complement DPP; thus, in an informal sense we can read Equation (155) as combining the marginal probability that YY is selected with the marginal probability that Yˉ\bar{Y} is not selected.

We can also make a similar derivation using Equation (150):

Note that Equation (155) involves asymmetric matrix products, but Equation (161) does not; on the other hand, KIYˉK-I_{\bar{Y}} is (in general) indefinite.

Learning

We have seen that determinantal point process offer appealing modeling intuitions and practical algorithms, capturing geometric notions of diversity and permitting computationally efficient inference in a variety of settings. However, to accurately model real-world data we must first somehow determine appropriate values of the model parameters. While an expert could conceivably design an appropriate DPP kernel from prior knowledge, in general, especially when dealing with large data sets, we would like to have an automated method for learning a DPP.

We first discuss how to parameterize DPPs conditioned on input data. We then define what we mean by learning, and, using the quality vs. diversity decomposition introduced in Section 3.1, we show how a parameterized quality model can be learned efficiently from a training set.

Suppose we want to use a DPP to model the seats in an auditorium chosen by students attending a class. (Perhaps we think students tend to spread out.) In this context each meeting of the class is a new sample from the empirical distribution over subsets of the (fixed) seats, so we merely need to collect enough samples and we should be able to fit our model, as desired.

For many problems, however, the notion of a single fixed base set Y\mathcal{Y} is inadequate. For instance, consider extractive document summarization, where the goal is to choose a subset of the sentences in a news article that together form a good summary of the entire article. In this setting Y\mathcal{Y} is the set of sentences in the news article being summarized, thus Y\mathcal{Y} is not fixed in advance but instead depends on context. One way to deal with this problem is to model the summary for each article as its own DPP with a separate kernel matrix. This approach certainly affords great flexibility, but if we have only a single sample summary for each article, there is little hope of getting good parameter estimates. Even more importantly, we have learned nothing that can be applied to generate summaries of unseen articles at test time, which was presumably our goal in the first place.

Alternatively, we could let Y\mathcal{Y} be the set of all sentences appearing in any news article; this allows us to learn a single model for all of our data, but comes with obvious computational issues and does not address the other concerns, since sentences are rarely repeated.

To solve this problem, we need a DPP that depends parametrically on the input data; this will enable us to share information across training examples in a justifiable and effective way. We first introduce some notation. Let X\mathcal{X} be the input space; for example, X\mathcal{X} might be the space of news articles. Let Y(X)\mathcal{Y}(X) denote the ground set of items implied by an input XXX\in\mathcal{X}, e.g., the set of all sentences in news article XX. We have the following definition.

A conditional DPP P(Y=YX)\mathcal{P}(\boldsymbol{Y}=Y|X) is a conditional probabilistic model which assigns a probability to every possible subset YY(X)Y\subseteq\mathcal{Y}(X). The model takes the form of an L-ensemble:

where L(X)L(X) is a positive semidefinite Y(X)×Y(X)|\mathcal{Y}(X)|\times|\mathcal{Y}(X)| kernel matrix that depends on the input.

As discussed in Section 2, the normalization constant for a conditional DPP can be computed efficiently and is given by det(L(X)+I)\det(L(X)+I). Using the quality/diversity decomposition introduced in Section 3.1, we have

In the following sections we will discuss application-specific parameterizations of the quality and diversity models qq and ϕ\phi in terms of the input. First, however, we review our learning setup.

The basic supervised learning problem is as follows. We receive a training data sample {(X(t),Y(t))}t=1T\{(X^{(t)},Y^{(t)})\}_{t=1}^{T} drawn independently and identically from a distribution DD over pairs (X,Y)X×2Y(X)(X,Y)\in\mathcal{X}\times 2^{\mathcal{Y}(X)}, where X\mathcal{X} is an input space and Y(X)\mathcal{Y}(X) is the associated ground set for input XX. We assume that the conditional DPP kernel L(X;θ)L(X;\theta) is parameterized in terms of a generic θ\theta, and let

denote the conditional probability of an output YY given input XX under parameter θ\theta. The goal of learning is to choose appropriate θ\theta based on the training sample so that we can make accurate predictions on unseen inputs.

While there are a variety of objective functions commonly used for learning, here we will focus on maximum likelihood learning (or maximum likelihood estimation, often abbreviated MLE), where the goal is to choose θ\theta to maximize the conditional log-likelihood of the observed data:

Optimizing L\mathcal{L} is consistent under mild assumptions; that is, if the training data are actually drawn from a conditional DPP with parameter θ\theta^{*}, then the learned θθ\theta\to\theta^{*} as TT\to\infty. Of course real data are unlikely to exactly follow any particular model, but in any case the maximum likelihood approach has the advantage of calibrating the DPP to produce reasonable probability estimates, since maximizing L\mathcal{L} can be seen as minimizing the log-loss on the training data.

To optimize the log-likelihood, we will use standard algorithms such as gradient ascent or L-BFGS (Nocedal, 1980). These algorithms depend on the gradient L(θ)\nabla\mathcal{L}(\theta), which must exist and be computable, and they converge to the optimum whenever L(θ)\mathcal{L}(\theta) is concave in θ\theta. Thus, our ability to optimize likelihood efficiently will depend fundamentally on these two properties.

2 Learning quality

We begin by showing how to learn a parameterized quality model qi(X;θ)q_{i}(X;\theta) when the diversity feature function ϕi(X)\phi_{i}(X) is held fixed (Kulesza and Taskar, 2011b). This setup is somewhat analogous to support vector machines (Vapnik, 2000), where a kernel is fixed by the practitioner and then the per-example weights are automatically learned. Here, ϕi(X)\phi_{i}(X) can consist of any desired measurements (and could even be infinite-dimensional, as long as the resulting similarity matrix SS is a proper kernel). We propose computing the quality scores using a log-linear model:

For ease of notation, going forward we will assume that the training set contains only a single instance (X,Y)(X,Y), and drop the instance index tt. All of the following results extend easily to multiple training examples. First, we show that under this parameterization the log-likelihood function is concave in θ\theta; then we will show that its gradient can be computed efficiently. With these results in hand we will be able to apply standard optimization techniques.

L(θ)\mathcal{L}(\theta) is concave in θ\theta.

With respect to θ\theta, the first term is linear, the second is constant, and the third is the composition of a concave function (negative log-sum-exp) and an affine function, so the overall expression is concave. ∎

We now derive the gradient L(θ)\nabla\mathcal{L}(\theta), using Equation (171) as a starting point.

Thus, as in standard maximum entropy modeling, the gradient of the log-likelihood can be seen as the difference between the empirical feature counts and the expected feature counts under the model distribution. The difference here, of course, is that Pθ\mathcal{P}_{\theta} is a DPP, which assigns higher probability to diverse sets. Compared with a standard independent model obtained by removing the diversity term from Pθ\mathcal{P}_{\theta}, Equation (174) actually emphasizes those training examples that are not diverse, since these are the examples on which the quality model must focus its attention in order to overcome the bias imposed by the determinant. In the experiments that follow we will see that this distinction is important in practice.

The sum over YY^{\prime} in Equation (174) is exponential in Y(X)|\mathcal{Y}(X)|; hence we cannot compute it directly. Instead, we can rewrite it by switching the order of summation:

Note that Y{i}Pθ(YX)\sum_{Y^{\prime}\supseteq\{i\}}\mathcal{P}_{\theta}(Y^{\prime}|X) is the marginal probability of item ii appearing in a set sampled from the conditional DPP. That is, the expected feature counts are computable directly from the marginal probabilities. Recall that we can efficiently marginalize DPPs; in particular, per-item marginal probabilities are given by the diagonal of K(X;θ)K(X;\theta), the marginal kernel (which now depends on the input and the parameters). We can compute K(X;θ)K(X;\theta) from the kernel L(X;θ)L(X;\theta) using matrix inversion or eigendecomposition. Algorithm 4 shows how we can use these ideas to compute the gradient of L(θ)\mathcal{L}(\theta) efficiently.

In fact, note that we do not need all of K(X;θ)K(X;\theta), but only its diagonal. In Algorithm 4 we exploit this in the main loop, using only O(N2)O(N^{2}) multiplications rather than the O(N3)O(N^{3}) we would need to construct the entire marginal kernel. (In the dual representation, this can be improved further to O(ND)O(ND) multiplications.) Unfortunately, these savings are asymptotically irrelevant since we still need to eigendecompose L(X;θ)L(X;\theta), requiring about O(N3)O(N^{3}) time (or O(D3)O(D^{3}) time for the corresponding eigendecomposition in the dual). It is conceivable that a faster algorithm exists for computing the diagonal of K(X;θ)K(X;\theta) directly, along the lines of ideas recently proposed by Tang and Saad (2011) (which focus on sparse matrices); however, we are not currently aware of a useful improvement over Algorithm 4.

We demonstrate learning for the conditional DPP quality model on an extractive multi-document summarization task using news text. The basic goal is to generate a short piece of text that summarizes the most important information from a news story. In the extractive setting, the summary is constructed by stringing together sentences found in a cluster of relevant news articles. This selection problem is a balancing act: on the one hand, each selected sentence should be relevant, sharing significant information with the cluster as a whole; on the other, the selected sentences should be diverse as a group so that the summary is not repetitive and is as informative as possible given its length (Dang, 2005; Nenkova et al., 2006). DPPs are a natural fit for this task, viewed through the decomposition of Section 3.1 (Kulesza and Taskar, 2011b).

As in Section 4.1, the input XX will be a cluster of documents, and Y(X)\mathcal{Y}(X) a set of candidate sentences from those documents. In our experiments Y(X)\mathcal{Y}(X) contains all sentences from all articles in the cluster, although in general preprocessing could also be used to try to improve the candidate set (Conroy et al., 2004). We will learn a DPP to model good summaries YY for a given input XX. Because DPPs model unordered sets while summaries are linear text, we construct a written summary from YY by placing the sentences it contains in the same order in which they appeared in the original documents. This policy is unlikely to give optimal results, but it is consistent with prior work (Lin and Bilmes, 2010) and seems to perform well. Furthermore, it is at least partially justified by the fact that modern automatic summary evaluation metrics like ROUGE, which we describe later, are mostly invariant to sentence order.

We experiment with data from the multi-document summarization task (Task 2) of the 2003 and 2004 Document Understanding Conference (DUC) (Dang, 2005). The article clusters used for these tasks are taken from the NIST TDT collection. Each cluster contains approximately 10 articles drawn from the AP and New York Times newswires, and covers a single topic over a short time span. The clusters have a mean length of approximately 250 sentences and 5800 words. The 2003 task, which we use for training, contains 30 clusters, and the 2004 task, which is our test set, contains 50 clusters. Each cluster comes with four reference human summaries (which are not necessarily formed by sentences from the original articles) for evaluation purposes. Summaries are required to be at most 665 characters in length, including spaces. Figure 12 depicts a sample cluster from the test set.

To measure performance on this task we follow the original evaluation and use ROUGE, an automatic evaluation metric for summarization (Lin, 2004). ROUGE measures nn-gram overlap statistics between the human references and the summary being scored, and combines them to produce various sub-metrics. ROUGE-1, for example, is a simple unigram recall measure that has been shown to correlate quite well with human judgments (Lin, 2004). Here, we use ROUGE’s unigram F-measure (which combines ROUGE-1 with a measure of precision) as our primary metric for development. We refer to this measure as ROUGE-1F. We also report ROUGE-1P and ROUGE-1R (precision and recall, respectively) as well as ROUGE-2F and ROUGE-SU4F, which include bigram match statistics and have also been shown to correlate well with human judgments. Our implementation uses ROUGE version 1.5.5 with stemming turned on, but without stopword removal. These settings correspond to those used for the actual DUC competitions (Dang, 2005); however, we use a more recent version of ROUGE.

Recall that our learning setup requires a training sample of pairs (X,Y)(X,Y), where YY(X)Y\subseteq\mathcal{Y}(X). Unfortunately, while the human reference summaries provided with the DUC data are of high quality, they are not extractive, thus they do not serve as examples of summaries that we can actually model. To obtain high-quality extractive “oracle” summaries from the human summaries, we employ a simple greedy algorithm (Algorithm 5). On each round the sentence that achieves maximal unigram F-measure to the human references, normalized by length, is selected and added to the extractive summary. Since high F-measure requires high precision as well as recall, we then update the references by removing the words “covered” by the newly selected sentence, and proceed to the next round.

We can measure the success of this approach by calculating ROUGE scores of our oracle summaries with respect to the human summaries. Table 2 shows the results for the DUC 2003 training set. For reference, the table also includes the ROUGE scores of the best automatic system from the DUC competition in 2003 (“machine”), as well as the human references themselves (“human”). Note that, in the latter case, the human summary being evaluated is also one of the four references used to compute ROUGE; hence the scores are probably significantly higher than a human could achieve in practice. Furthermore, it has been shown that extractive summaries, even when generated optimally, are by nature limited in quality compared with unconstrained summaries (Genest et al., 2010). Thus we believe that the oracle summaries make strong targets for training.

Under this definition of ϕ\phi, the similarity SijS_{ij} between sentences ii and jj is known as their cosine similarity:

Two sentences are cosine similar if they contain many of the same words, particularly words that are uncommon (and thus more likely to be salient).

We augment ϕi(X)\phi_{i}(X) with an additional constant feature taking the value ρ0\rho\geq 0, which is a hyperparameter. This has the effect of making all sentences more similar to one another, increasing repulsion. We set ρ\rho to optimize ROUGE-1F score on the training set; in our experiments, the best choice was ρ=0.7\rho=0.7.

We use the very standard cosine distance as our similarity metric because we need to be confident that it is sensible; it will remain fixed throughout the experiments. On the other hand, weights for the quality features are learned, so we can use a variety of intuitive measures and rely on training to find an appropriate combination. The quality features we use are listed below. For some of the features, we make use of cosine distances; these are computed using the same tf-idf vectors as the diversity features. When a feature is intrinsically real-valued, we produce a series of binary features by binning. The bin boundaries are determined either globally or locally. Global bins are evenly spaced quantiles of the feature values across all sentences in the training set, while local bins are quantiles of the feature values in the current cluster only.

Constant: A constant feature allows the model to bias towards summaries with a greater or smaller number of sentences.

Length: We bin the length of the sentence (in characters) into five global bins.

Document position: We compute the position of the sentence in its original document and generate binary features indicating positions 1–5, plus a sixth binary feature indicating all other positions. We expect that, for newswire text, sentences that appear earlier in an article are more likely to be useful for summarization.

Mean cluster similarity: For each sentence we compute the average cosine distance to all other sentences in the cluster. This feature attempts to measure how well the sentence reflects the salient words occurring most frequently in the cluster. We use the raw score, five global bins, and ten local bins.

LexRank: We compute continuous LexRank scores by finding the principal eigenvector of the row-normalized cosine similarity matrix. (See Erkan and Radev (2004) for details.) This provides an alternative measure of centrality. We use the raw score, five global bins, and five local bins.

Personal pronouns: We count the number of personal pronouns (“he”, “her”, “themselves”, etc.) appearing in each sentence. Sentences with many pronouns may be poor for summarization since they omit important entity names.

In total we have 40 quality features; including ρ\rho our model has 41 parameters.

At test time, we need to take the learned parameters θ\theta and use them to predict a summary YY for a previously unseen document cluster XX. One option is to sample from the conditional distribution, which can be done exactly and efficiently, as described in Section 2.4.4. However, sampling occasionally produces low-probability predictions. We obtain better performance on this task by applying two alternative inference techniques.

Greedy MAP approximation. One common approach to prediction in probabilistic models is maximum a posteriori (MAP) decoding, which selects the highest probability configuration. For practical reasons, and because the primary metrics for evaluation were recall-based, the DUC evaluations imposed a length limit of 665 characters, including spaces, on all summaries. In order to compare with prior work we also apply this limit in our tests. Thus, our goal is to find the most likely summary, subject to a budget constraint:

Algorithm 6 is closely related to those given by Krause and Guestrin (2005) and especially Lin and Bilmes (2010). As discussed in Section 2.4.5, algorithms of this type have formal approximation guarantees for monotone submodular problems. Our MAP problem is not generally monotone; nonetheless, Algorithm 6 seems to work well in practice, and is very fast (see Table 3).

Minimum Bayes risk decoding. The second inference technique we consider is minimum Bayes risk (MBR) decoding. First proposed by Goel and Byrne (2000) for automatic speech recognition, MBR decoding has also been used successfully for word alignment and machine translation (Kumar and Byrne, 2002, 2004). The idea is to choose a prediction that minimizes a particular application-specific loss function under uncertainty about the evaluation target. In our setting we use ROUGE-1F as a (negative) loss function, so we have

Combining these approximations, we have the following inference rule:

We train our model with a standard L-BFGS optimization algorithm. We place a zero-mean Gaussian prior on the parameters θ\theta, with variance set to optimize ROUGE-1F on a development subset of the 2003 data. We learn parameters θ\theta on the DUC 2003 corpus, and test them using DUC 2004 data. We generate predictions from the trained DPP using the two inference algorithms described in the previous section, and compare their performance to a variety of baseline systems.

Our first and simplest baseline merely returns the first 665 bytes of the cluster text. Since the clusters consist of news articles, this is not an entirely unreasonable summary in many cases. We refer to this baseline as begin.

We also compare against an alternative DPP-based model with identical similarity measure and quality features, but where the quality model has been trained using standard logistic regression. To learn this baseline, each sentence is treated as a unique instance to be classified as included or not included, with labels derived from our training oracle. Thus, it has the advantages of a DPP at test time, but does not take into account the diversity model while training; comparing to this baseline allows us to isolate the contribution of learning the model parameters in context. Note that MBR inference is impractical for this model because its training does not properly calibrate for overall summary length, so nearly all samples are either too long or too short. Thus, we report only the results obtained from greedy inference. We refer to this model as lr+dpp.

Next, we employ as baselines a range of previously proposed methods for multi-document summarization. Perhaps the simplest and most popular is Maximum Marginal Relevance (MMR), which uses a greedy selection process (Carbonell and Goldstein, 1998). MMR relies on a similarity measure between sentences, for which we use the cosine distance measure SS, and a measure of relevance for each sentence, for which we use the same logistic regression-trained quality model as above. Sentences are chosen iteratively according to

where YY is the set of sentences already selected (initially empty), qi(X)q_{i}(X) is the learned quality score, and SijS_{ij} is the cosine similarity between sentences ii and jj. The tradeoff α\alpha is optimized on a development set, and sentences are added until the budget is full. We refer to this baseline as lr+mmr.

We also compare against the three highest-scoring systems that actually competed in the DUC 2004 competition—peers 65, 104, and 35—as well as the submodular graph-based approach recently described by Lin and Bilmes (2010), which we refer to as submod1, and the improved submodular learning approach proposed by Lin and Bilmes (2012), which we denote submod2. We produced our own implementation of submod1, but rely on previously reported numbers for submod2, which include only ROUGE-1 scores.

Table 4 shows the results for all methods on the DUC 2004 test corpus. Scores for the actual DUC competitors differ slightly from the originally reported results because we use an updated version of the ROUGE package. Bold entries highlight the best performance in each column; in the case of MBR inference, which is stochastic, the improvements are significant at 99% confidence. The DPP models outperform the baselines in most cases; furthermore, there is a significant boost in performance due to the use of DPP maximum likelihood training in place of logistic regression. MBR inference performs best, assuming we take sufficiently many samples; on the other hand, greedy inference runs more quickly than dpp-mbr100 and produces superior results. Relative to most other methods, the DPP model with MBR inference seems to more strongly emphasize recall. Note that MBR inference was performed with respect to ROUGE-1F, but could also be run to optimize other metrics if desired.

Feature contributions. In Table 5 we report the performance of dpp-greedy when different groups of features from Section 4.2.1 are removed, in order to estimate their relative contributions. Length and position appear to be quite important; however, although individually similarity and LexRank scores have only a modest impact on performance, when both are omitted the drop is significant. This suggests, intuitively, that these two groups convey similar information—both are essentially measures of centrality—but that this information is important to achieving strong performance.

k𝑘k-DPPs

A determinantal point process assigns a probability to every subset of the ground set Y\mathcal{Y}. This means that, with some probability, a sample from the process will be empty; with some probability, it will be all of Y\mathcal{Y}. In many cases this is not desirable. For instance, we might want to use a DPP to model the positions of basketball players on a court, under the assumption that a team tends to spread out for better coverage. In this setting, we know that with very high probability each team will have exactly five players on the court. Thus, if our model gives some probability of seeing zero or fifty players, it is not likely to be a good fit.

We showed in Section 2.4.4 that there exist elementary DPPs having fixed cardinality kk; however, this is achieved only by focusing exclusively (and equally) on kk specific “aspects” of the data, as represented by eigenvectors of the kernel. Thus, for DPPs, the notions of size and content are fundamentally intertwined. We cannot change one without affecting the other. This is a serious limitation on the types of distributions than can be expressed; for instance, a DPP cannot even capture the uniform distribution over sets of cardinality kk.

More generally, even for applications where the number of items is unknown, the size model imposed by a DPP may not be a good fit. We have seen that the cardinality of a DPP sample has a simple distribution: it is the number of successes in a series of Bernoulli trials. But while this distribution characterizes certain types of data, other cases might look very different. For example, picnickers may tend to stake out diverse positions in a park, but on warm weekend days there might be hundreds of people, and on a rainy Tuesday night there are likely to be none. This bimodal distribution is quite unlike the sum of Bernoulli variables imposed by DPPs.

Perhaps most importantly, in some cases we do not even want to model cardinality at all, but instead offer it as a parameter. For example, a search engine might need to deliver ten diverse results to its desktop users, but only five to its mobile users. This ability to control the size of a DPP “on the fly” can be crucial in real-world applications.

In this section we introduce kk-DPPs, which address the issues described above by conditioning a DPP on the cardinality of the random set Y\boldsymbol{Y}. This simple change effectively divorces the DPP content model, with its intuitive diversifying properties, from the DPP size model, which is not always appropriate. We can then use the DPP content model with a size model of our choosing, or simply set the desired size based on context. The result is a significantly more expressive modeling approach (which can even have limited positive correlations) and increased control.

We begin by defining kk-DPPs. The conditionalization they require, though simple in theory, necessitates new algorithms for inference problems like normalization and sampling. Naively, these tasks require exponential time, but we show that through recursions for computing elementary symmetric polynomials we can solve them exactly in polynomial time. Finally, we demonstrate the use of kk-DPPs on an image search problem, where the goal is to show users diverse sets of images that correspond to their query.

A kk-DPP on a discrete set Y={1,2,,N}\mathcal{Y}=\{1,2,\dots,N\} is a distribution over all subsets YYY\subseteq\mathcal{Y} with cardinality kk (Kulesza and Taskar, 2011a). In contrast to the standard DPP, which models both the size and content of a random subset Y\boldsymbol{Y}, a kk-DPP is concerned only with the content of a random kk-set. Thus, a kk-DPP is obtained by conditioning a standard DPP on the event that the set Y\boldsymbol{Y} has cardinality kk. Formally, the kk-DPP PLk\mathcal{P}^{k}_{L} gives probabilities

where Y=k|Y|=k and LL is a positive semidefinite kernel. Compared to the standard DPP, the only changes are the restriction on YY and the normalization constant. While in a DPP every kk-set YY competes with all other subsets of Y\mathcal{Y}, in a kk-DPP it competes only with sets of the same cardinality. This subtle change has significant implications.

For instance, consider the seemingly simple distribution that is uniform over all sets YYY\subseteq\mathcal{Y} with cardinality kk. If we attempt to build a DPP capturing this distribution we quickly run into difficulties. In particular, the marginal probability of any single item is kN\frac{k}{N}, so the marginal kernel KK, if it exists, must have kN\frac{k}{N} on the diagonal. Likewise, the marginal probability of any pair of items is k(k1)N(N1)\frac{k(k-1)}{N(N-1)}, and so by symmetry the off diagonal entries of KK must be equal to a constant. As a result, any valid marginal kernel has to be the sum of a constant matrix and a multiple of the identity matrix. Since a constant matrix has at most one nonzero eigenvalue and the identity matrix is full rank, it is easy to show that, except in the special cases k=0,1,N1k=0,1,N-1, the resulting kernel has full rank. But we know that a full rank kernel implies that the probability of seeing all NN items together is nonzero. Thus the desired process cannot be a DPP unless k=0,1,N1,k=0,1,N-1, or NN. On the other hand, a kk-DPP with the identity matrix as its kernel gives the distribution we are looking for. This improved expressive power can be quite valuable in practice.

2 Inference

Of course, increasing the expressive power of the DPP causes us to wonder whether, in doing so, we might have lost some of the convenient computational properties that made DPPs useful in the first place. Naively, this seems to be the case; for instance, while the normalizing constant for a DPP can be written in closed form, the sum in Equation (181) is exponential and seems hard to simplify. In this section, we will show how kk-DPP inference can in fact be performed efficiently, using recursions for computing the elementary symmetric polynomials.

Recall that the kkth elementary symmetric polynomial on λ1,λ2,λN\lambda_{1},\lambda_{2}\dots,\lambda_{N} is given by

The normalization constant for a kk-DPP is

where λ1,λ2,,λN\lambda_{1},\lambda_{2},\dots,\lambda_{N} are the eigenvalues of LL.

One way to see this is to examine the characteristic polynomial of LL, det(LλI)\det(L-\lambda I) (Gel’fand, 1989). We can also show it directly using properties of DPPs. Recalling that

where PL\mathcal{P}_{L} is the DPP with kernel LL. Applying Lemma 2.6, which expresses any DPP as a mixture of elementary DPPs, we have

where we use Lemma 2.7 in the last two steps to conclude that PVJ(Y)=0\mathcal{P}^{V_{J}}(Y^{\prime})=0 unless J=Y|J|=|Y^{\prime}|. (Recall that VJV_{J} is the set of eigenvectors of LL associated with λn\lambda_{n} for nJn\in J.) ∎

To compute the kkth elementary symmetric polynomial, we can use the recursive algorithm given in Algorithm 7, which is based on the observation that every set of kk eigenvalues either omits λN\lambda_{N}, in which case we must choose kk of the remaining eigenvalues, or includes λN\lambda_{N}, in which case we get a factor of λN\lambda_{N} and choose only k1k-1 of the remaining eigenvalues. Formally, letting ekNe_{k}^{N} be a shorthand for ek(λ1,λ2,,λN)e_{k}(\lambda_{1},\lambda_{2},\dots,\lambda_{N}), we have

Note that a variety of recursions for computing elementary symmetric polynomials exist, including Newton’s identities, the Difference Algorithm, and the Summation Algorithm (Baker and Harwell, 1996). Algorithm 7 is essentially the Summation Algorithm, which is both asymptotically faster and numerically more stable than the other two, since it uses only sums and does not rely on precise cancellation of large numbers.

Algorithm 7 runs in time O(Nk)O(Nk). Strictly speaking, the inner loop need only iterate up to Nk+lN-k+l in order to obtain ekNe_{k}^{N} at the end; however, by going up to NN we compute all of the preceding elementary symmetric polynomials elNe_{l}^{N} along the way. Thus, by running Algorithm 7 with k=Nk=N we can compute the normalizers for kk-DPPs of every size in time O(N2)O(N^{2}). This can be useful when kk is not known in advance.

2.2 Sampling

Since a kk-DPP is just a DPP conditioned on size, we could sample a kk-DPP by repeatedly sampling the corresponding DPP and rejecting the samples until we obtain one of size kk. To make this more efficient, recall from Section 2.4.4 that the standard DPP sampling algorithm proceeds in two phases. First, a subset VV of the eigenvectors of LL is selected at random, and then a set of cardinality V|V| is sampled based on those eigenvectors. Since the size of a sample is fixed in the first phase, we could reject the samples before the second phase even begins, waiting until we have V=k|V|=k. However, rejection sampling is likely to be slow. It would be better to directly sample a set VV conditioned on the fact that its cardinality is kk. In this section we show how sampling kk eigenvectors can be done efficiently, yielding a sampling algorithm for kk-DPPs that is asymptotically as fast as sampling standard DPPs.

We can formalize the intuition above by rewriting the kk-DPP distribution in terms of the corresponding DPP:

whenever Y=k|Y|=k, where we replace the DPP normalization constant with the kk-DPP normalization constant using Proposition 5.1. Applying Lemma 2.6 and Lemma 2.7 to decompose the DPP into elementary parts yields

Therefore, a kk-DPP is also a mixture of elementary DPPs, but it only gives nonzero weight to those of dimension kk. Since the second phase of DPP sampling provides a means for sampling from any given elementary DPP, we can sample from a kk-DPP if we can sample index sets JJ according to the corresponding mixture components. Like normalization, this is naively an exponential task, but we can do it efficiently using the recursive properties of elementary symmetric polynomials.

Let J\boldsymbol{J} be the desired random variable, so that Pr(J=J)=1ekNnJλn\Pr(\boldsymbol{J}=J)=\frac{1}{e_{k}^{N}}\prod_{n\in J}\lambda_{n} when J=k|J|=k, and zero otherwise. Then Algorithm 8 yields a sample for J\boldsymbol{J}.

If k=0k=0, then Algorithm 8 returns immediately at the first iteration of the loop with J=J=\emptyset, which is the only possible value of J\boldsymbol{J}.

If N=1N=1 and k=1k=1, then J\boldsymbol{J} must contain the single index 11. We have e11=λ1e_{1}^{1}=\lambda_{1} and e00=1e_{0}^{0}=1, thus λ1e00e11=1\lambda_{1}\frac{e_{0}^{0}}{e_{1}^{1}}=1, and Algorithm 8 returns J={1}J=\{1\} with probability 1.

We proceed by induction and compute the probability that Algorithm 8 returns JJ for N>1N>1 and 1kN1\leq k\leq N. By inductive hypothesis, if an iteration of the loop in Algorithm 8 begins with n<Nn<N and 0ln0\leq l\leq n, then the remainder of the algorithm adds to JJ a set of elements JJ^{\prime} with probability

Now suppose that JJ contains NN, J=J{N}J=J^{\prime}\cup\{N\}. Then NN must be added to JJ in the first iteration of the loop, which occurs with probability λNek1N1ekN\lambda_{N}\frac{e_{k-1}^{N-1}}{e_{k}^{N}}. The second iteration then begins with n=N1n=N-1 and l=k1l=k-1. If ll is zero, we have the immediate base case; otherwise we have 1ln1\leq l\leq n. By the inductive hypothesis, the remainder of the algorithm selects JJ^{\prime} with probability

if J=k1|J^{\prime}|=k-1, and zero otherwise. Thus Algorithm 8 returns JJ with probability

On the other hand, if JJ does not contain NN, then the first iteration must add nothing to JJ; this happens with probability

where we use the fact that ekNλNek1N1=ekN1e_{k}^{N}-\lambda_{N}e_{k-1}^{N-1}=e_{k}^{N-1}. The second iteration then begins with n=N1n=N-1 and l=kl=k. We observe that if N1<kN-1<k, then Equation (199) is equal to zero, since eln=0e_{l}^{n}=0 whenever l>nl>n. Thus almost surely the second iteration begins with knk\leq n, and we can apply the inductive hypothesis. This guarantees that the remainder of the algorithm chooses JJ with probability

whenever J=k|J|=k. The overall probability that Algorithm 8 returns JJ is therefore

Algorithm 8 precomputes the values of e11,,ekNe_{1}^{1},\dots,e_{k}^{N}, which requires O(Nk)O(Nk) time using Algorithm 7. The loop then iterates at most NN times and requires only a constant number of operations, so Algorithm 8 runs in O(Nk)O(Nk) time overall. By Equation (195), selecting JJ with Algorithm 8 and then sampling from the elementary DPP PVJ\mathcal{P}^{V_{J}} generates a sample from the kk-DPP. As shown in Section 2.4.4, sampling an elementary DPP can be done in O(Nk3)O(Nk^{3}) time (see the second loop of Algorithm 1), so sampling kk-DPPs is O(Nk3)O(Nk^{3}) overall, assuming we have an eigendecomposition of the kernel in advance. This is no more expensive than sampling a standard DPP.

2.3 Marginalization

Since kk-DPPs are not DPPs, they do not in general have marginal kernels. However, we can still use their connection to DPPs to compute the marginal probability of a set AA, Ak|A|\leq k:

where LAL^{A} is the kernel, given in Equation (50), of the DPP conditioned on the inclusion of AA, and

is the normalization constant for the (kA)(k-|A|)-DPP with kernel LAL^{A}. That is, the marginal probabilities for a kk-DPP are just the marginal probabilities for a DPP with the same kernel, but with an appropriate change of normalizing constants. We can simplify Equation (205) by observing that

since the left hand side is the probability (under the DPP with kernel LL) that AA occurs by itself, and the right hand side is the marginal probability of AA multiplied by the probability of observing nothing else conditioned on observing AA: 1/det(LA+I)1/\det(L^{A}+I). Thus we have

That is, the marginal probability of AA is the probability of observing exactly AA times the normalization constant when conditioning on AA. Note that a version of this formula also holds for standard DPPs, but there it can be rewritten in terms of the marginal kernel.

Equations (205) and (209) are general but require computing large determinants and elementary symmetric polynomials, regardless of the size of AA. Moreover, those quantities (for example, det(LA+I)\det(L^{A}+I)) must be recomputed for each unique AA whose marginal probability is desired. Thus, finding the marginal probabilities of many small sets is expensive compared to a standard DPP, where we need only small minors of KK. However, we can derive a more efficient approach in the special but useful case where we want to know all of the singleton marginals for a kk-DPP—for instance, in order to implement quality learning as described in Section 4.2.

We start by using Equation (195) to write the marginal probability of an item ii in terms of a combination of elementary DPPs:

Because the marginal kernel of the elementary DPP PVJ\mathcal{P}^{V_{J}} is given by nJvnvn\sum_{n\in J}\boldsymbol{v}_{n}\boldsymbol{v}_{n}^{\top}, we have

where ek1n=ek1(λ1,λ2,,λn1,λn+1,,λN)e^{-n}_{k-1}=e_{k-1}(\lambda_{1},\lambda_{2},\dots,\lambda_{n-1},\lambda_{n+1},\dots,\lambda_{N}) denotes the (k1)(k-1)-order elementary symmetric polynomial for all eigenvalues of LL except λn\lambda_{n}. Note that λnek1n/ekN\lambda_{n}e_{k-1}^{-n}/e^{N}_{k} is exactly the marginal probability that nJn\in J when JJ is chosen using Algorithm 8; in other words, the marginal probability of item ii is the sum of the contributions (vnei)2(\boldsymbol{v}_{n}^{\top}\boldsymbol{e}_{i})^{2} made by each eigenvector scaled by the respective probabilities that the eigenvectors are selected. The contributions are easily computed from the eigendecomposition of LL, thus we need only ekNe^{N}_{k} and ek1ne^{-n}_{k-1} for each value of nn in order to calculate the marginals for all items in O(N2)O(N^{2}) time, or O(ND)O(ND) time if the rank of LL is D<ND<N.

Algorithm 7 computes ek1N1=ek1Ne^{N-1}_{k-1}=e^{-N}_{k-1} in the process of obtaining ekNe^{N}_{k}, so naively we could run Algorithm 7 NN times, repeatedly reordering the eigenvectors so that each takes a turn at the last position. To compute all of the required polynomials in this fashion would require O(N2k)O(N^{2}k) time. However, we can improve this (for small kk) to O(Nlog(N)k2)O(N\log(N)k^{2}); to do so we will make use of a binary tree on NN leaves. Each node of the tree corresponds to a set of eigenvalues of LL; the leaves represent single eigenvalues, and an interior node of the tree represents the set of eigenvalues corresponding to its descendant leaves. (See Figure 13.) We will associate with each node the set of elementary symmetric polynomials e1(Λ),e2(Λ),,ek(Λ)e_{1}(\Lambda),e_{2}(\Lambda),\dots,e_{k}(\Lambda), where Λ\Lambda is the set of eigenvalues represented by the node.

These polynomials can be computed directly for leaf nodes in constant time, and the polynomials of an interior node can be computed given those of its children using a simple recursion:

Thus, we can compute the polynomials for the entire tree in O(Nlog(N)k2)O(N\log(N)k^{2}) time; this is sufficient to obtain ekNe^{N}_{k} at the root node.

However, if we now remove a leaf node corresponding to eigenvalue nn, we invalidate the polynomials along the path from the leaf to the root; see Figure 13. This leaves logN\log N disjoint subtrees which together represent all of the eigenvalues of LL, leaving out λn\lambda_{n}. We can now apply Equation (214) logN\log N times to the roots of these trees in order to obtain ek1ne^{-n}_{k-1} in O(log(N)k2)O(\log(N)k^{2}) time. If we do this for each value of nn, the total additional time required is O(Nlog(N)k2)O(N\log(N)k^{2}).

The algorithm described above thus takes O(Nlog(N)k2)O(N\log(N)k^{2}) time to produce the necessary elementary symmetric polynomials, which in turn allow us to compute all of the singleton marginals. This is a dramatic improvement over applying Equation (205) to each item separately.

2.4 Conditioning

Suppose we want to condition a kk-DPP on the inclusion of a particular set AA. For A+B=k|A|+|B|=k we have

Thus the conditional kk-DPP is a kAk-|A|-DPP whose kernel is the same as that of the associated conditional DPP. The normalization constant is ZkAAZ_{k-|A|}^{A}. We can condition on excluding AA in the same manner.

2.5 Finding the mode

Unfortunately, although kk-DPPs offer the efficient versions of DPP inference algorithms presented above, finding the most likely set YY remains intractable. It is easy to see that the reduction from Section 2.4.5 still applies, since the cardinality of the YY corresponding to an exact 3-cover, if it exists, is known. In practice we can utilize greedy approximations, like we did for standard DPPs in Section 4.2.1.

3 Experiments: image search

We demonstrate the use of kk-DPPs on an image search task (Kulesza and Taskar, 2011a). The motivation is as follows. Suppose that we run an image search engine, where our primary goal is to deliver the most relevant possible images to our users. Unfortunately, the query strings those users provide are often ambiguous. For instance, a user searching for “philadelphia” might be looking for pictures of the city skyline, street-level shots of buildings, or perhaps iconic sights like the Liberty Bell or the Love sculpture. Furthermore, even if we know the user is looking for a skyline photograph, he or she might specifically want a daytime or nighttime shot, a particular angle, and so on. In general, we cannot expect users to provide enough information in a textual query to identify the best image with any certainty.

For this reason search engines typically provide a small array of results, and we argue that, to maximize the probability of the user being happy with at least one image, the results should be relevant to the query but also diverse with respect to one another. That is, if we want to maximize the proportion of users searching “philadelphia” who are satisfied by our response, each image we return should satisfy a large but distinct subset of those users, thus maximizing our overall coverage. Since we want diverse results but also require control over the number of results we provide, a kk-DPP is a natural fit.

Of course, we do not actually run a search engine and do not have real users. Thus, in order to be able to evaluate our model using real human feedback, we define the task in a manner that allows us to obtain inexpensive human supervision via Amazon Mechanical Turk. We do this by establishing a simple binary decision problem, where the goal is to choose, given two possible sets of image search results, the set that is more diverse. Formally, our labeled training data comprises comparative pairs of image sets {(Yt+,Yt)}t=1T\{(Y^{+}_{t},Y^{-}_{t})\}_{t=1}^{T}, where set Yt+Y_{t}^{+} is preferred over set YtY_{t}^{-}, Yt+=Yt=k|Y_{t}^{+}|=|Y_{t}^{-}|=k. We can measure performance on this classification task using the zero-one loss, which is zero whenever we choose the correct set from a given pair, and one otherwise.

For this task we employ a simple method for learning a combination of kk-DPPs that is convex and seems to work well in practice. Given a set L1,L2,,LDL_{1},L_{2},\dots,L_{D} of “expert” kernel matrices, which are fixed in advance, define the combination model

where l=1Dθl=1\sum_{l=1}^{D}\theta_{l}=1. Note that this is a combination of distributions, rather than a combination of kernels. We will learn θ\theta to optimize a logistic loss measure on the binary task:

where γ\gamma is a hyperparameter that controls how aggressively we penalize mistakes. Intuitively, the idea is to find a combination of kk-DPPs where the positive sets Yt+Y^{+}_{t} receive higher probability than the corresponding negative sets YtY^{-}_{t}. By using the logistic loss (Figure 14), which acts like a smooth hinge loss, we focus on making fewer mistakes.

Because Equation (220) is convex in θ\theta (it is the composition of the convex logistic loss function with a linear function of θ\theta), we can optimize it efficiently using projected gradient descent, where we alternate taking gradient steps and projecting on the constraint l=1Dθl=1\sum_{l=1}^{D}\theta_{l}=1. The gradient is given by

where δt\delta^{t} is a vector with entries

Projection onto the simplex is achieved using standard algorithms (Bertsekas, 1999).

3.2 Data

We create datasets for three broad image search categories, using 8–12 hand-selected queries for each category. (See Table 6.) For each query, we retrieve the top 64 results from Google Image Search, restricting the search to JPEG files that pass the strictest level of Safe Search filtering. Of those 64 results, we eliminate any that are no longer available for download. On average this leaves us with 63.0 images per query, with a range of 59–64.

We then use the downloaded images to generate 960 training instances for each category, spread evenly across the different queries. In order to compare kk-DPPs directly against baseline heuristic methods that do not model probabilities of full sets, we generate only instances where Yt+Y_{t}^{+} and YtY_{t}^{-} differ by a single element. That is, the classification problem is effectively to choose which of two candidate images it+,itii_{t}^{+},i_{t}^{i} is a less redundant addition to a given partial result set YtY_{t}:

In our experiments YtY_{t} contains five images, so k=Yt+=Yt=6k=|Y_{t}^{+}|=|Y_{t}^{-}|=6. We sample partial result sets using a kk-DPP with a SIFT-based kernel (details below) to encourage diversity. The candidates are then selected uniformly at random from the remaining images, except for 10% of instances that are reserved for measuring the performance of our human judges. For those instances, one of the candidates is a duplicate image chosen uniformly at random from the partial result set, making it the obviously more redundant choice. The other candidate is chosen as usual.

In order to decide which candidate actually results in the more diverse set, we collect human diversity judgments using Amazon’s Mechanical Turk. Annotators are drawn from the general pool of Turk workers, and are able to label as many instances as they wish. Annotators are paid $0.01 USD for each instance that they label. For practical reasons, we present the images to the annotators at reduced scale; the larger dimension of an image is always 250 pixels. The annotators are instructed to choose the candidate that they feel is “less similar” to the images in the partial result set. We do not offer any specific guidance on how to judge similarity, since dealing with uncertainty in human users is central to the task. The candidate images are presented in random order. Figure 15 shows a sample instance from each category.

Overall, we find that workers choose the correct image for 80.8% of the calibration instances (that is, they choose the one not belonging to the partial result set). This suggests only moderate levels of noise due to misunderstanding, inattention or robot workers. However, for non-calibration instances the task is inherently difficult and subjective. To keep noise in check, we have each instance labeled by five independent judges, and keep only those instances where four or more judges agree. In the end this leaves us with 408–482 labeled instances per category, or about half of the original instances.

3.3 Kernels

We define a set of 55 “expert” similarity kernels for the collected images, which form the building blocks of our combination model and baseline methods. Each kernel LfL^{\boldsymbol{f}} is the Gram matrix of some feature function f\boldsymbol{f}; that is, Lijf=f(i)f(j)L^{\boldsymbol{f}}_{ij}=\boldsymbol{f}(i)\cdot\boldsymbol{f}(j) for images ii and jj. We therefore specify the kernels through the feature functions used to generate them. All of our feature functions are normalized so that f(i)2=1\|\boldsymbol{f}(i)\|^{2}=1 for all ii; this ensures that no image is a priori more likely than any other. Implicitly, thinking in terms of the decomposition in Section 3.1, we are assuming that all of the images in our set are equally relevant in order to isolate the modeling of diversity. This assumption is at least partly justified by the fact that our images come from actual Google searches, and are thus presumably relevant to the query.

We use the following feature functions, which derive from standard image processing and feature extraction methods:

Color (2 variants): Each pixel is assigned a coordinate in three-dimensional Lab color space. The colors are then sorted into axis-aligned bins, producing a histogram of either 8 or 64 dimensions.

SIFT (2 variants): The images are processed with the vlfeat toolbox to obtain sets of 128-dimensional SIFT descriptors (Lowe, 1999; Vedaldi and Fulkerson, 2008). The descriptors for a given category are combined, subsampled to a set of 25,000, and then clustered using kk-means into either 256 or 512 clusters. The feature vector for an image is the normalized histogram of the nearest clusters to the descriptors in the image.

GIST: The images are processed using code from Oliva and Torralba (2006) to yield 960-dimensional GIST feature vectors characterizing properties like “openness,” “roughness,” “naturalness,” and so on.

In addition to the five feature functions described above, we include another five that are identical but focus only on the center of the image, defined as the centered rectangle with dimensions half those of the original image. This gives our first ten kernels. We then create 45 pairwise combination kernels by concatenating every possible pair of the 10 basic feature vectors. This technique produces kernels that synthesize more than one source of information, offering greater flexibility.

Finally, we augment our kernels by adding a constant hyperparameter ρ\rho to each entry. ρ\rho acts a knob for controlling the overall preference for diversity; as ρ\rho increases, all images appear more similar, thus increasing repulsion. In our experiments, ρ\rho is chosen independently for each method and each category to optimize performance on the training set.

3.4 Methods

We test four different methods. Two use kk-DPPs, and two are derived from Maximum Marginal Relevance (MMR) (Carbonell and Goldstein, 1998). For each approach, we test both the single best expert kernel on the training data and a learned combination of kernels. All methods were tuned separately for each of the three query categories. On each run a random 25% of the labeled examples are reserved for testing, and the remaining 75% form the training set used for setting hyperparameters and training. Recall that YtY_{t} is the five-image partial result set for instance tt, and let Ct={it+,it}C_{t}=\{i^{+}_{t},i^{-}_{t}\} denote the set of two candidates images, where it+i^{+}_{t} is the candidate preferred by the human judges.

Given a single kernel LL, the kk-DPP prediction is

We select the kernel with the best zero-one accuracy on the training set, and apply it to the test set.

We apply our learning method to the full set of 55 kernels, optimizing Equation (220) on the training set to obtain a 55-dimensional mixture vector θ\theta. We set γ\gamma to minimize the zero-one training loss. We then take the learned θ\theta and apply it to making predictions on the test set:

Recall that MMR is a standard, heuristic technique for generating diverse sets of search results. The idea is to build a set iteratively by adding on each round a result that maximizes a weighted combination of relevance (with respect to the query) and diversity, measured as the maximum similarity to any of the previously selected results. (See Section 4.2.1 for more details about MMR.) For our experiments, we assume relevance is uniform; hence we merely need to decide which of the two candidates has the smaller maximum similarity to the partial result set. Thus, for a given kernel LL, the MMR prediction is

As for the kk-DPP, we select the single best kernel on the training set, and apply it to the test set.

We can also attempt to learn a mixture of similarity kernels for MMR. We use the same training approach as for kk-DPPs, but replace the probability score Pθk(Yy{i})P^{k}_{\theta}(Y_{y}\cup\{i\}) with the negative cost

which is just the negative similarity of item ii to the set YtY_{t} under the combined kernel metric. Significantly, this substitution makes the optimization non-smooth and non-convex, unlike the kk-DPP optimization. In practice this means that the global optimum is not easily found. However, even a local optimum may provide advantages over the single best kernel. In our experiments we use the local optimum found by projected gradient descent starting from the uniform kernel combination.

3.5 Results

Table 7 shows the mean zero-one accuracy of each method for each query category, averaged over 100 random train/test splits. Statistical significance is computed by bootstrapping. Regardless of whether we learn a mixture, kk-DPPs outperform MMR on two of the three categories, significant at 99% confidence. In all cases, the learned mixture of kk-DPPs achieves the best performance. Note that, because the decision being made for each instance is binary, 50% is equivalent to random performance. Thus the numbers in Table 7 suggest that this is a rather difficult task, a conclusion supported by the rates of noise exhibited by the human judges. However, the changes in performance due to learning and the use of kk-DPPs are more obviously significant when measured as improvements above this baseline level. For example, in the cars category our mixture of kk-DPPs performs 14.58 percentage points better than random, versus 9.59 points for MMR with a mixture of kernels. Figure 16 shows some actual samples drawn using the kk-DPP sampling algorithm.

Table 8 shows, for the kk-DPP mixture model, the kernels receiving the highest weights for each search category (on average over 100 train/test splits). Combined-feature kernels appear to be useful, and the three categories exhibit significant differences in what annotators deem diverse, as we might expect.

We can also return to our original motivation and try to measure how well each method “covers” the space of likely user intentions. Since we do not have access to real users who are searching for the queries in our dataset, we instead simulate them by imagining that each is looking for a particular target image drawn randomly from the images in our collection. For instance, given the query “philadelphia” we might draw a target image of the Love sculpture, and then evaluate each method on whether it selects an image of the Love sculpture, i.e., whether it satisfies that virtual user. More generally, we will simply record the maximum similarity of any image in the result set to the target image. We expect better methods to show higher similarity when averaged over a large number of such users.

Structured DPPs

We have seen in the preceding sections that DPPs offer polynomial-time inference and learning with respect to NN, the number of items in the ground set Y\mathcal{Y}. This is important since DPPs model an exponential number of subsets YYY\subseteq\mathcal{Y}, so naive algorithms would be intractable. And yet, we can imagine DPP applications for which even linear time is too slow. For example, suppose that after modeling the positions of basketball players, as proposed in the previous section, we wanted to take our analysis one step further. An obvious extension is to realize that a player does not simply occupy a single position, but instead moves around the court over time. Thus, we might want to model not just diverse sets of positions on the court, but diverse sets of paths around the court during a game. While we could reasonably discretize the possible court positions to a manageable number MM, the number of paths over, say, 100 time steps would be M100M^{100}, making it almost certainly impossible to enumerate them all, let alone build an M100×M100M^{100}\times M^{100} kernel matrix.

However, in this combinatorial setting we can take advantage of the fact that, even though there are exponentially many paths, they are structured; that is, every path is built from a small number of the same basic components. This kind of structure has frequently been exploited in machine learning, for example, to find the best translation of a sentence, or to compute the marginals of a Markov random field. In such cases structure allows us to factor computations over exponentially many possibilities in an efficient way. And yet, the situation for structured DPPs is even worse: when the number of items in Y\mathcal{Y} is exponential, we are actually modeling a distribution over the doubly exponential number of subsets of an exponential Y\mathcal{Y}. If there are M100M^{100} possible paths, there are 2M1002^{M^{100}} subsets of paths, and a DPP assigns a probability to every one. This poses an extreme computational challenge.

In order to develop efficient structured DPPs (SDPPs), we will therefore need to combine the dynamic programming techniques used for standard structured prediction with the algorithms that make DPP inference efficient. We will show how this can be done by applying the dual DPP representation from Section 3.3, which shares spectral properties with the kernel LL but is manageable in size, and the use of second-order message passing, where the usual sum-product or min-sum semiring is replaced with a special structure that computes quadratic quantities over a factor graph (Li and Eisner, 2009). In the end, we will demonstrate that it is possible to normalize and sample from an SDPP in polynomial time.

Structured DPPs open up a large variety of new possibilities for applications; they allow us to model diverse sets of essentially any structured objects. For instance, we could find not only the best translation but a diverse set of high-quality translations for a sentence, perhaps to aid a human translator. Or, we could study the distinct proteins coded by a gene under alternative RNA splicings, using the diversifying properties of DPPs to cover the large space of possibilities with a small representative set. Later, we will apply SDPPs to three real-world tasks: identifying multiple human poses in images, where there are combinatorially many possible poses, and we assume that the poses are diverse in that they tend not to overlap; identifying salient lines of research in a corpus of computer science publications, where the structures are citation chains of important papers, and we want to find a small number of chains that covers the major topic in the corpus; and building threads from news text, where the goal is to extract from a large corpus of articles the most significant news stories, and for each story present a sequence of articles covering the major developments of that story through time.

We begin by defining SDPPs and stating the structural assumptions that are necessary to make inference efficient; we then show how these assumptions give rise to polynomial-time algorithms using second order message passing. We discuss how sometimes even these polynomial algorithms can be too slow in practice, but demonstrate that by applying the technique of random projections (Section 3.4) we can dramatically speed up computation and reduce memory use while maintaining a close approximation to the original model (Kulesza and Taskar, 2010). Finally, we show how SDPPs can be applied to the experimental settings described above, yielding improved results compared with a variety of standard and heuristic baseline approaches.

In Section 2.4 we saw that DPPs remain tractable on modern computers for NN up to around 10,000. This is no small feat, given that the number of subsets of 10,000 items is roughly the number of particles in the observable universe to the 40th power. Of course, this is not magic but simply a consequence of a certain type of structure; that is, we can perform inference with DPPs because the probabilities of these subsets are expressed as combinations of only a relatively small set of O(N2)O(N^{2}) parameters. In order to make the jump now to ground sets Y\mathcal{Y} that are exponentially large, we will need to make an similar assumption about the structure of Y\mathcal{Y} itself. Thus, a structured DPP (SDPP) is a DPP in which the ground set Y\mathcal{Y} is given implicitly by combinations of a set of parts. For instance, the parts could be positions on the court, and an element of Y\mathcal{Y} a sequence of those positions. Or the parts could be rules of a context-free grammar, and then an element of Y\mathcal{Y} might be a complete parse of a sentence. This assumption of structure will give us the algorithmic leverage we need to efficiently work with a distribution over a doubly exponential number of possibilities.

Because elements of Y\mathcal{Y} are now structures, we will no longer think of Y={1,2,,N}\mathcal{Y}=\{1,2,\dots,N\}; instead, each element yY\boldsymbol{y}\in\mathcal{Y} is a structure given by a sequence of RR parts (y1,y2,,yR)(y_{1},y_{2},\dots,y_{R}), each of which takes a value from a finite set of MM possibilities. For example, if y\boldsymbol{y} is the path of a basketball player, then RR is the number of time steps at which the player’s position is recorded, and yry_{r} is the player’s discretized position at time rr. We will use yi\boldsymbol{y}_{i} to denote the iith structure in Y\mathcal{Y} under an arbitrary ordering; thus Y={y1,y2,,yN}\mathcal{Y}=\{\boldsymbol{y}_{1},\boldsymbol{y}_{2},\dots,\boldsymbol{y}_{N}\}, where N=MRN=M^{R}. The parts of yi\boldsymbol{y}_{i} are denoted yiry_{ir}.

An immediate challenge is that the kernel LL, which has N2N^{2} entries, can no longer be written down explicitly. We therefore define its entries using the quality/diversity decomposition presented in Section 3.1. Recall that this decomposition gives the entries of LL as follows:

where q(yi)q(\boldsymbol{y}_{i}) is a nonnegative measure of the quality of structure yi\boldsymbol{y}_{i}, and ϕ(yi)\phi(\boldsymbol{y}_{i}) is a DD-dimensional vector of diversity features so that ϕ(yi)ϕ(yj)\phi(\boldsymbol{y}_{i})^{\top}\phi(\boldsymbol{y}_{j}) is a measure of the similarity between structures yi\boldsymbol{y}_{i} and yj\boldsymbol{y}_{j}. We cannot afford to specify qq and ϕ\phi for every possible structure, but we can use the assumption that structures are built from parts to define a factorization, analogous to the factorization over cliques that gives rise to Markov random fields.

Specifically, we assume that the model decomposes over a set of factors FF, where a factor αF\alpha\in F is a small subset of the parts of a structure. (Keeping the factors small will ensure that the model is tractable.) We denote by yα\boldsymbol{y}_{\alpha} the collection of parts of y\boldsymbol{y} that are included in factor α\alpha; then the factorization assumption is that the quality score decomposes multiplicatively over parts, and the diversity features decompose additively:

We argue that these are quite natural factorizations. For instance, in our player tracking example we might have a positional factor for each time rr, allowing the quality model to prefer paths that go through certain high-traffic areas, and a transitional factor for each pair of times (r1,r)(r-1,r), allowing the quality model to enforce the smoothness of a path over time. More generally, if the parts correspond to cliques in a graph, then the quality scores can be given by a standard log-linear Markov random field (MRF), which defines a multiplicative distribution over structures that give labelings of the graph. Thus, while in Section 3.2 we compared DPPs and MRFs as alternative models for the same binary labeling problems, SDPPs can also be seen as an extension to MRFs, allowing us to take a model of individual structures and use it as a quality measure for modeling diverse sets of structures.

Diversity features, on the other hand, decompose additively, so we can think of them as global feature functions defined by summing local features, again as done in standard structured prediction. For example, ϕr(yr)\phi_{r}(y_{r}) could track the coarse-level position of a player at time rr, so that paths passing through similar positions at similar times are less likely to co-occur. Note that, in contrast to the unstructured case, we do not generally have ϕ(y)=1\|\phi(\boldsymbol{y})\|=1, since there is no way to enforce such a constraint under the factorization in Equation (231). Instead, we simply set the factor features ϕα(yα)\phi_{\alpha}(\boldsymbol{y}_{\alpha}) to have unit norm for all α\alpha and all possible values of yα\boldsymbol{y}_{\alpha}. This slightly biases the model towards structures that have the same (or similar) features at every factor, since such structures maximize ϕ\|\phi\|. However, the effect of this bias seems to be minor in practice.

As for unstructured DPPs, the quality and diversity models combine to produce balanced, high-quality, diverse results. However, in the structured case the contribution of the diversity model can be especially significant due to the combinatorial nature of the items in Y\mathcal{Y}. For instance, imagine taking a particular high-quality path and perturbing it slightly, say by shifting the position at each time step by a small random amount. This process results in a new and distinct path, but is unlikely to have a significant effect on the overall quality: the path remains smooth and goes through roughly the same positions. Of course, this is not unique to the structured case; we can have similar high-quality items in any DPP. What makes the problem especially serious here is that there is a combinatorial number of such slightly perturbed paths; the introduction of structure dramatically increases not only the number of items in Y\mathcal{Y}, but also the number of subtle variations that we might want to suppress. Furthermore, factored distributions over structures are often very peaked due to the geometric combination of quality scores across many factors, so variations of the most likely structure can be much more probable than any real alternative. For these reasons independent samples from an MRF can often look nearly identical; a sample from an SDPP, on the other hand, is much more likely to contain a truly diverse set of structures.

Before describing the technical details needed to make SDPPs computationally efficient, we first develop some intuition by studying the results of the model as applied to a synthetic motion tracking task, where the goal is to follow a collection of particles as they travel in a one-dimensional space over time. This is essentially a simplified version of our player tracking example, but with the motion restricted to a line. We will assume that a path y\boldsymbol{y} has 50 parts, where each part yr{1,2,,50}y_{r}\in\{1,2,\dots,50\} is the particle’s position at time step rr discretized into one of 50 locations. The total number of possible trajectories in this setting is 505050^{50}, and we will be modeling 250502^{50^{50}} possible sets of trajectories. We define positional and transitional factors

While a real tracking problem would involve quality scores q(y)q(\boldsymbol{y}) that depend on some observations—for example, measurements over time from a set of physical sensors, or perhaps a video feed from a basketball game—for simplicity we determine the quality of a trajectory here using only its starting position and a measure of smoothness over time. Specifically, we have

where the initial quality score q1(y1)q_{1}(y_{1}) is given by a smooth trimodal function with a primary mode at position 25 and secondary modes at positions 10 and 40, depicted by the blue curves on the left side of Figure 17, and the quality scores for all other positional factors are fixed to one and have no effect. The transition quality is the same at all time steps, and given by q(yr1,yr)=fN(yr1yr)q(y_{r-1},y_{r})=f_{\mathcal{N}}(y_{r-1}-y_{r}), where fNf_{\mathcal{N}} is the density function of the normal distribution; that is, the quality of a transition is maximized when the particle does not change location, and decreases as the particle moves further and further from its previous location. In essence, high quality paths start near the central position and move smoothly through time.

We want trajectories to be considered similar if they travel through similar positions, so we define a 50-dimensional diversity feature vector as follows:

Intuitively, feature ll is activated when the trajectory passes near position ll, so trajectories passing through nearby positions will activate the same features and thus appear similar in the diversity model. Note that for simplicity, the time at which a particle reaches a given position has no effect on the diversity features. The diversity features for the transitional factors are zero and have no effect.

We use the quality and diversity models specified above to define our SDPP. In order to obtain good results for visualization, we scale the kernel so that the expected number of trajectories in a sample from the SDPP is five. We then apply the algorithms developed later to draw samples from the model. The first row of Figure 17 shows the results, and for comparison each corresponding panel on the second row shows an equal number of trajectories sampled independently, with probabilities proportional to their quality scores. As evident from the figure, trajectories sampled independently tend to cluster in the middle region due to the strong preference for this starting position. The SDPP samples, however, are more diverse, tending to cover more of the space while still respecting the quality scores—they are still smooth, and still tend to start near the center.

2 Second-order message passing

The central computational challenge for SDPPs is the fact that N=MRN=M^{R} is exponentially large, making the usual inference algorithms intractable. However, we showed in Section 3.3 that DPP inference can be recast in terms of a smaller dual representation CC; recall that, if BB is the D×ND\times N matrix whose columns are given by Byi=q(yi)ϕ(yi)B_{\boldsymbol{y}_{i}}=q(\boldsymbol{y}_{i})\phi(\boldsymbol{y}_{i}), then L=BBL=B^{\top}B and

Of course, for the dual representation to be of any use we must be able to efficiently compute CC. If we think of qα2(yα)q^{2}_{\alpha}(\boldsymbol{y}_{\alpha}) as the factor potentials of a graphical model p(y)αFqα2(yα)p(\boldsymbol{y})\propto\prod_{\alpha\in F}q^{2}_{\alpha}(\boldsymbol{y}_{\alpha}), then computing CC is equivalent to computing second moments of the diversity features under pp (up to normalization). Since the diversity features factor additively, CC is quadratic in the local diversity features ϕα(yα)\phi_{\alpha}(\boldsymbol{y}_{\alpha}). Thus, we could naively calculate CC by computing the pairwise marginals p(yα,yα)p(\boldsymbol{y}_{\alpha},\boldsymbol{y}_{\alpha^{\prime}}) for all realizations of the factors α,α\alpha,\alpha^{\prime} and, by linearity of expectations, adding up their contributions:

where the proportionality is due to the normalizing constant of p(y)p(\boldsymbol{y}). However, this sum is quadratic in the number of factors and their possible realizations, and can therefore be expensive when structures are large.

Instead, we can substitute the factorization from Equation (231) into Equation (237) to obtain

It turns out that this expression is computable in linear time using a second-order message passing algorithm.

Second-order message passing was first introduced by Li and Eisner (2009). The main idea is to compute second-order statistics over a graphical model by using the standard belief propagation message passing algorithm, but with a special semiring in place of the usual sum-product or max-product. This substitution makes it possible to compute quantities of the form

where pαp_{\alpha} are nonnegative and aαa_{\alpha} and bαb_{\alpha} are arbitrary functions. Note that we can think of pαp_{\alpha} as defining a multiplicatively decomposed function

and aαa_{\alpha} and bαb_{\alpha} as defining corresponding additively decomposed functions aa and bb.

We begin by defining the notion of a factor graph, which provides the structure for all message passing algorithms. We then describe standard belief propagation on factor graphs, and show how it can be defined in a general way using semirings. Finally we demonstrate that belief propagation using the semiring proposed by Li and Eisner (2009) computes quantities of the form in Equation (240).

Message passing operates on factor graphs. A factor graph is an undirected bipartite graph with two types of vertices: variable nodes and factor nodes. Variable nodes correspond to the parts of the structure being modeled; for the SDPP setup described above, a factor graph contains RR variable nodes, each associated with a distinct part rr. Similarly, each factor node corresponds to a distinct factor αF\alpha\in F. Every edge in the graph connects a variable node to a factor node, and an edge exists between variable node rr and factor node α\alpha if and only if rαr\in\alpha. Thus, the factor graph encodes the relationships between parts and factors. Figure 18 shows an example factor graph for the tracking problem from Section 6.1.1.

It is obvious that the computation of Equation (240) cannot be efficient when factors are allowed to be arbitrary, since in the limit a factor could contain all parts and we could assign arbitrary values to every configuration y\boldsymbol{y}. Thus we will assume that the degree of the factor nodes is bounded by a constant cc. (In Figure 18, as well as all of the experiments we run, we have c=2c=2.) Furthermore, message passing algorithms are efficient whenever the factor graph has low treewidth, or, roughly, when only small sets of nodes need to be merged to obtain a tree. Going forward we will assume that the factor graph is a tree, since any low-treewidth factor graph can be converted into an equivalent factor tree with bounded factors using the junction tree algorithm (Lauritzen and Spiegelhalter, 1988).

2.2 Belief propagation

We now describe the basic belief propagation algorithm, first introduced by Pearl (1982). Suppose each factor has an associated real-valued weight function wα(yα)w_{\alpha}(\boldsymbol{y}_{\alpha}), giving rise to the multiplicatively decomposed global weight function

Then the goal of belief propagation is to efficiently compute sums of w(y)w(\boldsymbol{y}) over combinatorially large sets of structures y\boldsymbol{y}.

We will refer to a structure y\boldsymbol{y} as an assignment to the variable nodes of the factor graph, since it defines a value yry_{r} for every part. Likewise we can think of yα\boldsymbol{y}_{\alpha} as an assignment to the variable nodes adjacent to α\alpha, and yry_{r} as an assignment to a single variable node rr. We use the notation yαyr\boldsymbol{y}_{\alpha}\sim y_{r} to indicate that yα\boldsymbol{y}_{\alpha} is consistent with yry_{r}, in the sense that it assigns the same value to variable node rr. Finally, denote by F(r)F(r) the set of factors in which variable rr participates.

The belief propagation algorithm defines recursive message functions mm to be passed along edges of the factor graph; the formula for the message depends on whether it is traveling from a variable node to a factor node, or vice versa:

From a variable rr to a factor α\alpha:

From a factor α\alpha to a variable rr:

Intuitively, an outgoing message summarizes all of the messages arriving at the source node, excluding the one coming from the target node. Messages from factor nodes additionally incorporate information about the local weight function.

Belief propagation passes these messages in two phases based on an arbitrary orientation of the factor tree. In the first phase, called the forward pass, messages are passed upwards from the leaves to the root. In the second phase, or backward pass, the messages are passed downward, from the root to the leaves. Upon completion of the second phase one message has been passed in each direction along every edge in the factor graph, and it is possible to prove using an inductive argument that, for every yry_{r},

If we think of the wαw_{\alpha} as potential functions, then Equation (245) gives the (unnormalized) marginal probability of the assignment yry_{r} under a Markov random field.

Note that the algorithm passes two messages per edge in the factor graph, and each message requires considering at most McM^{c} assignments, therefore its running time is O(McR)O(M^{c}R). The sum on the right-hand side of Equation (245), however, is exponential in the number of parts. Thus belief propagation offers an efficient means of computing certain combinatorial quantities that would naively require exponential time.

2.3 Semirings

In fact, the belief propagation algorithm can be easily generalized to operate over an arbitrary semiring, thus allowing the same basic algorithm to perform a variety of useful computations. Recall that a semiring W,,,0,1\langle W,\oplus,\otimes,\boldsymbol{0},\boldsymbol{1}\rangle comprises a set of elements WW, an addition operator \oplus, a multiplication operator \otimes, an additive identity 0\boldsymbol{0}, and a multiplicative identity 1\boldsymbol{1} satisfying the following requirements for all a,b,cWa,b,c\in W:

Addition is associative and commutative, with identity 0\boldsymbol{0}:

Multiplication is associative, with identity 1\boldsymbol{1}:

Multiplication distributes over addition:

0\boldsymbol{0} is absorbing under multiplication:

We can rewrite the messages defined by belief propagation in terms of these more general operations. For wα(yα)Ww_{\alpha}(\boldsymbol{y}_{\alpha})\in W, we have

As before, we can pass messages forward and then backward through the factor tree. Because the properties of semirings are sufficient to preserve the inductive argument, we then have the following analog of Equation (245):

We have seen that Equation (256) computes marginal probabilities under the sum-product semiring, but other semirings give rise to useful results as well. Under the max-product semiring, for instance, Equation (256) is the so-called max-marginal—the maximum unnormalized probability of any single assignment y\boldsymbol{y} consistent with yry_{r}. In the next section we take this one step further, and show how a carefully designed semiring will allow us to sum second-order quantities across exponentially many structures y\boldsymbol{y}.

2.4 Second-order semiring

It is easy to verify that the semiring properties hold for these operations. Now, suppose that the weight function for a factor α\alpha is given by

where pαp_{\alpha}, aαa_{\alpha}, and bαb_{\alpha} are as before. Then wα(yα)Ww_{\alpha}(\boldsymbol{y}_{\alpha})\in W, and we can get some intuition about the multiplication operator by observing that the fourth component of wα(yα)wα(yα)w_{\alpha}(\boldsymbol{y}_{\alpha})\otimes w_{\alpha^{\prime}}(\boldsymbol{y}_{\alpha^{\prime}}) is

In other words, multiplication in the second-order semiring combines the values of pp multiplicatively and the values of aa and bb additively, leaving the result in the fourth component. It is not hard to extend this argument inductively and show that the fourth component of αFwα(yα)\bigotimes_{\alpha\in F}w_{\alpha}(\boldsymbol{y}_{\alpha}) is given in general by

Thus, by Equation (256) and the definition of \oplus, belief propagation with the second-order semiring yields messages that satisfy

Note that multiplication and addition remain constant-time operations in the second-order semiring, thus belief propagation can still be performed in time linear in the number of factors. In the following section we will show that the dual representation CC, as well as related quantities needed to perform inference in SDPPs, takes the form of Equation (265); thus second-order message passing will be an important tool for efficient SDPP inference.

3 Inference

The factorization proposed in Equation (231) gives a concise definition of a structured DPP for an exponentially large Y\mathcal{Y}; remarkably, under suitable conditions it also gives rise to tractable algorithms for normalizing the SDPP, computing marginals, and sampling. The only restrictions necessary for efficiency are the ones we inherit from belief propagation: the factors must be of bounded size so that we can enumerate all of their possible configurations, and together they must form a low-treewidth graph on the parts of the structure. These are precisely the same conditions needed for efficient graphical model inference (Koller and Friedman, 2009), which is generalized by inference in SDPPs.

As we saw in Section 3.3, the dual representation CC is sufficient to normalize and marginalize an SDPP in time constant in NN. Recall from Equation (239) that the dual representation of an SDPP can be written as

which is of the form required to apply second-order message passing. Specifically, we can compute for each pair of diversity features (a,b)(a,b) the value of

by summing Equation (265) over the possible assignments yry_{r}, and then simply assemble the results into the matrix CC. Since there are D(D+1)2\frac{D(D+1)}{2} unique entries in CC and message passing runs in time O(McR)O(M^{c}R), computing CC in this fashion requires O(D2McR)O(D^{2}M^{c}R) time.

It is easy to verify that computations in this vectorized semiring are identical to those obtained by repeated use of the scalar semiring.

Given CC, we can now normalize and compute marginals for an SDPP using the formulas in Section 3.3; for instance

where C=n=1Dλnv^nv^nC=\sum_{n=1}^{D}\lambda_{n}\boldsymbol{\hat{v}}_{n}\boldsymbol{\hat{v}}_{n}^{\top} is an eigendecomposition of CC.

The introduction of structure offers an alternative type of marginal probability, this time not of structures yY\boldsymbol{y}\in\mathcal{Y} but of single part assignments. More precisely, we can ask how many of the structures in a sample from the SDPP can be expected to make the assignment y^r\hat{y}_{r} to part rr:

The sum is exponential, but we can compute it efficiently using second-order message passing. We apply Equation (273) to get

The result is a sum of DD terms, each of which takes the form of Equation (265), and therefore is efficiently computable by message passing. The desired part marginal probability simply requires DD separate applications of belief propagation, one per eigenvector v^n\boldsymbol{\hat{v}}_{n}, for a total runtime of O(D2McR)O(D^{2}M^{c}R). (It is also possible to vectorize this computation and use a single run of belief propagation.) Note that if we require the marginal for only a single part μr(y^r)\mu_{r}(\hat{y}_{r}), we can run just the forward pass if we root the factor tree at part node rr. However, by running both passes we obtain everything we need to compute the part marginals for any rr and y^r\hat{y}_{r}; the asymptotic time required to compute all part marginals is the same as the time required to compute just one.

3.2 Sampling

While the dual representation provides useful algorithms for normalization and marginals, the dual sampling algorithm is linear in NN; for SDPPs, this is too slow to be useful. In order to make SDPP sampling practical, we need to be able to efficiently choose a structure yi\boldsymbol{y}_{i} according to the distribution

in the first line of the while loop in Algorithm 3. We can use the definition of BB to obtain

Thus, the desired distribution has the familiar form of Equation (265). For instance, the marginal probability of part rr taking the assignment y^r\hat{y}_{r} is given by

which we can compute with k=V^k=|\hat{V}| runs of belief propagation (or a single vectorized run), taking only O(DMcRk)O(DM^{c}Rk) time. More generally, the message-passing computation of these marginals offers an efficient algorithm for sampling individual full structures yi\boldsymbol{y}_{i}. We will first show a naive method based on iterated computation of conditional marginals, and then use it to derive a more efficient algorithm by integrating the sampling of parts into the message-passing process.

Returning to the factor graph used for belief propagation (see Section 6.2.1), we can force a part rr^{\prime} to take a certain assignment yry_{r^{\prime}} by adding a new singleton factor containing only rr^{\prime}, and setting its weight function to 1\boldsymbol{1} for yry_{r^{\prime}} and 0\boldsymbol{0} otherwise. (In practice, we do not need to actually create a new factor; we can simply set outgoing messages from variable rr^{\prime} to 0\boldsymbol{0} for all but the desired assignment yry_{r^{\prime}}.) It is easy to see that Equation (256) becomes

where the sum is now doubly constrained, since any assignment y\boldsymbol{y} that is not consistent with yry_{r^{\prime}} introduces a 0\boldsymbol{0} into the product. If αFwα(yα)\bigotimes_{\alpha\in F}w_{\alpha}(\boldsymbol{y}_{\alpha}) gives rise to a probability measure over structures y\boldsymbol{y}, then Equation (283) can be seen as the unnormalized conditional marginal probability of the assignment yry_{r} given yry_{r^{\prime}}. For example, using the second-order semiring with p=q2p=q^{2} and a=b=v^ϕa=b=\boldsymbol{\hat{v}}^{\top}\phi, we have

Summing these values for all v^V^\boldsymbol{\hat{v}}\in\hat{V} and normalizing the result yields the conditional distribution of yry_{r} given fixed assignment yry_{r^{\prime}} under Equation (281). Going forward we will assume for simplicity that V^\hat{V} contains a single vector v^\boldsymbol{\hat{v}}; however, the general case is easily handled by maintaining V^|\hat{V}| messages in parallel, or by vectorizing the computation.

The observation that we can compute conditional probabilities with certain assignments held fixed gives rise to a naive algorithm for sampling a structure according to Pr(yi)\Pr(\boldsymbol{y}_{i}) in Equation (281), shown in Algorithm 9. While polynomial, Algorithm 9 requires running belief propagation RR times, which might be prohibitively expensive for large structures. We can do better by weaving the sampling steps into a single run of belief propagation. We discuss first how this can be done for linear factor graphs, where the intuition is simpler, and then extend it to general factor trees.

Suppose that the factor graph is a linear chain arranged from left to right. Each node in the graph has at most two neighbors—one to the left, and one to the right. Assume the belief propagation forward pass proceeds from left to right, and the backward pass from right to left. To send a message to the right, a node needs only to receive its message from the left. Conversely, to send a message to the left, only the message from the right is needed. Thus, the forward and backward passes can be performed independently.

Consider now the execution of Algorithm 9 on this factor graph. Assume the variable nodes are numbered in decreasing order from left to right, so the variable sampled in the first iteration is the rightmost variable node. Observe that on iteration rr, we do not actually need to run belief propagation to completion; we need only the messages incoming to variable node rr, since those suffice to compute the (conditional) marginals for part rr. To obtain those messages, we must compute all of the forward messages sent from the left of variable rr, and the backward messages from the right. Call this set of messages m(r)\boldsymbol{m}(r).

Note that m(1)\boldsymbol{m}(1) is just a full, unconstrained forward pass, which can be computed in time O(DMcR)O(DM^{c}R). Now compare m(r)\boldsymbol{m}(r) to m(r1)\boldsymbol{m}(r-1). Between iteration r1r-1 and rr, the only change to SS is that variable r1r-1, to the right of variable rr, has been assigned. Therefore the forward messages in m(r)\boldsymbol{m}(r), which come from the left, do not need to be recomputed, as they are a subset of the forward messages in m(r1)\boldsymbol{m}(r-1). Likewise, the backward messages sent from the right of variable r1r-1 are unchanged, so they do not need to be recomputed. The only new messages in m(r)\boldsymbol{m}(r) are those backward messages traveling from r1r-1 to rr. These can be computed, using m(r1)\boldsymbol{m}(r-1) and the sampled assignment yr1y_{r-1}, in constant time. See Figure 19 for an illustration of this process.

Thus, rather than restarting belief propagation on each loop of Algorithm 9, we have shown that we need only compute a small number of additional messages. In essence we have threaded the sampling of parts rr into the backward pass. After completing the forward pass, we sample y1y_{1}; we then compute the backward messages from y1y_{1} to y2y_{2}, sample y2y_{2}, and so on. When the backward pass is complete, we sample the final assignment yRy_{R} and are finished. Since the initial forward pass takes O(DMcR)O(DM^{c}R) time and each of the O(R)O(R) subsequent iterations takes at most O(DMc)O(DM^{c}) time, we can sample from Pr(yi)\Pr(\boldsymbol{y}_{i}) over a linear graph in O(DMcR)O(DM^{c}R) time.

The algorithm described above for linear graphs can be generalized to arbitrary factor trees. For standard graphical model sampling using the sum-product semiring, the generalization is straightforward—we can simply pass messages up to the root and then sample on the backward pass from the root to the leaves. However, for arbitrary semirings this is algorithm is incorrect, since an assignment to one node can affect the messages arriving at its siblings even when the parent’s assignment is fixed.

Let mba(S)m_{b\to a}(\cdot|S) be the message function sent from node bb to node aa during a run of belief propagation where the assignments in SS have been held fixed. Imagine that we re-root the factor tree with aa as the root; then define Ta(b)T_{a}(b) to be the subtree rooted at bb (see Figure 20). Several useful observations follow.

If b1b_{1} and b2b_{2} are distinct neighbors of aa, then Ta(b1)T_{a}(b_{1}) and Ta(b2)T_{a}(b_{2}) are disjoint.

The claim is immediate, since the underlying graph is a tree. ∎

mba(S)m_{b\to a}(\cdot|S) can be computed given only the messages mcb(S)m_{c\to b}(\cdot|S) for all neighbors cac\neq a of bb and either the weight function wbw_{b} (if bb is a factor node) or the assignment to bb in SS (if bb is a variable node and such an assignment exists).

Follows from the message definitions in Equations (254) and (255). ∎

mba(S)m_{b\to a}(\cdot|S) depends only on the assignments in SS that give values to variables in Ta(b)T_{a}(b).

If bb is a leaf (that is, its only neighbor is aa), the lemma holds trivially. If bb is not a leaf, then assume inductively that incoming messages mcb(S)m_{c\to b}(\cdot|S), cac\neq a, depend only on assignments to variables in Tb(c)T_{b}(c). By Lemma 6.2, the message mba(S)m_{b\to a}(\cdot|S) depends only on those messages and (possibly) the assignment to bb in SS. Since bb and Tb(c)T_{b}(c) are subgraphs of Ta(b)T_{a}(b), the claim follows. ∎

To sample a structure, we begin by initializing S0=S_{0}=\emptyset and setting messages m^ba=mba(S0)\hat{m}_{b\to a}=m_{b\to a}(\cdot|S_{0}) for all neighbor pairs (a,b)(a,b). This can be done in O(DMcR)O(DM^{c}R) time via belief propagation.

Now we walk the graph, sampling assignments and updating the current messages m^ba\hat{m}_{b\to a} as we go. Step tt from node bb to aa proceeds in three parts as follows:

Check whether bb is a variable node without an assignment in St1S_{t-1}. If so, sample an assignment yby_{b} using the current incoming messages m^cb\hat{m}_{c\to b}, and set St=St1{yb}S_{t}=S_{t-1}\cup\{y_{b}\}. Otherwise set St=St1S_{t}=S_{t-1}.

Recompute and update m^ba\hat{m}_{b\to a} using the current messages and Equations (254) and (255), taking into account any assignment to bb in StS_{t}.

This simple algorithm has the following useful invariant.

Following step tt from bb to aa, for every neighbor dd of aa we have

By design, the theorem holds at the outset of the walk. Suppose inductively that the claim is true for steps 1,2,,t11,2,\dots,t-1. Let tt^{\prime} be the most recent step prior to tt at which we visited aa, or if step tt was our first visit to aa. Since the graph is a tree, we know that between steps tt^{\prime} and tt the walk remained entirely within Ta(b)T_{a}(b). Hence the only assignments in StStS_{t}-S_{t^{\prime}} are to variables in Ta(b)T_{a}(b). As a result, for all neighbors dbd\neq b of aa we have m^da=mda(St)=mda(St)\hat{m}_{d\to a}=m_{d\to a}(\cdot|S_{t^{\prime}})=m_{d\to a}(\cdot|S_{t}) by the inductive hypothesis, Lemma 6.1, and Lemma 6.3.

It remains to show that m^ba=mba(Si)\hat{m}_{b\to a}=m_{b\to a}(\cdot|S_{i}). For all neighbors cac\neq a of bb, we know that m^cb=mcb(Si1)=mcb(St)\hat{m}_{c\to b}=m_{c\to b}(\cdot|S_{i-1})=m_{c\to b}(\cdot|S_{t}) due to the inductive hypothesis and Lemma 6.3 (since bb is not in Tb(c)T_{b}(c)). By Lemma 6.2, then, we have m^ba=mba(St)\hat{m}_{b\to a}=m_{b\to a}(\cdot|S_{t}). ∎

Theorem 285 guarantees that whenever we sample an assignment for the current variable node in the first part of step tt, we sample from the conditional marginal distribution Pr(ybSt1)\Pr(y_{b}|S_{t-1}). Therefore, we can sample a complete structure from the distribution Pr(y)\Pr(\boldsymbol{y}) if we walk the entire tree. This can be done, for example, by starting at the root and proceeding in depth-first order. Such a walk takes O(R)O(R) steps, and each step requires computing only a single message. Thus, allowing now for k=V^>1k=|\hat{V}|>1, we can sample a structure in time O(DMcRk)O(DM^{c}Rk), a significant improvement over Algorithm 9. The procedure is summarized in Algorithm 10.

Algorithm 10 is the final piece of machinery needed to replicate the DPP sampling algorithm using the dual representation. The full SDPP sampling process is given in Algorithm 11 and runs in time O(D2k3+DMcRk2)O(D^{2}k^{3}+DM^{c}Rk^{2}), where kk is the number of eigenvectors selected in the first loop. As in standard DPP sampling, the asymptotically most expensive operation is the orthonormalization; here we require O(D2)O(D^{2}) time to compute each of the O(k2)O(k^{2}) dot products.

4 Experiments: pose estimation

To demonstrate that SDPPs effectively model characteristics of real-world data, we apply them to a multiple-person pose estimation task (Kulesza and Taskar, 2010). Our input will be a still image depicting multiple people, and our goal is to simultaneously identify the poses—the positions of the torsos, heads, and left and right arms—of all the people in the image. A pose y\boldsymbol{y} is therefore a structure with four parts, in this case literally body parts. To form a complete structure, each part rr is assigned a position/orientation pair yry_{r}. Our quality model will be based on “part detectors” trained to estimate the likelihood of a particular body part at a particular location and orientation; thus we will focus on identifying poses that correspond well to the image itself. Our similarity model, on the other hand, will focus on the location of a pose within the image. Since the part detectors often have uncertainty about the precise location of a part, there may be many variations of a single pose that outscore the poses of all the other, less detectable people. An independent model would thus be likely to choose many similar poses. By encouraging the model to choose a spatially diverse set of poses, we hope to improve the chance that the model predicts a single pose for each person.

Our dataset consists of 73 still frames taken from various TV shows, each approximately 720 by 540 pixels in size (Sapp et al., 2010)The images and code were obtained from http://www.vision.grasp.upenn.edu/video. As much as possible, the selected frames contain three or more people at similar scale, all facing the camera and without serious occlusions. Sample images from the dataset are shown in Figure 22. Each person in each image is annotated by hand; each of the four parts (head, torso, right arm, and left arm) is labeled with the pixel location of a reference point (e.g., the shoulder) and an orientation selected from among 24 discretized angles.

There are approximately 75,000 possible values for each part, so there are about 475,0004^{75,000} possible poses, and thus we cannot reasonably use a standard DPP for this problem. Instead, we build a factorized SDPP. Our factors are given by the standard pictorial structure model (Felzenszwalb and Huttenlocher, 2005; Fischler and Elschlager, 1973), treating each pose as a two-level tree with the torso as the root and the head and arms as leaves. Each node (body part) has a singleton factor, and each edge has a corresponding pairwise factor.

Our quality function derives from the model proposed by Sapp et al. (2010), and is given by

where EE is the set of edges in the part tree, γ\gamma is a scale parameter that will control the expected number of poses in an SDPP sample, and β\beta is a sharpness parameter that controls the dynamic range of the quality scores. We set the values of the hyperparameters γ\gamma and β\beta using a held-out training set, as discussed below. The per-part quality scores qr(yr)q_{r}(y_{r}) are provided by the customized part detectors trained by Sapp et al. (2010) on similar images; they assign a value to every proposed location and orientation yry_{r} of part rr. The pairwise quality scores qr,r(yr,yr)q_{r,r^{\prime}}(y_{r},y_{r^{\prime}}) are defined according to a Gaussian “spring” that encourages, for example, the left arm to begin near the left shoulder of the torso. Full details of the model are provided in Sapp et al. (2010).

In order to encourage the model not to choose overlapping poses, our diversity features reflect the locations of the constituent parts:

4.2 Methods

We compare samples from the SDPP defined above to those from two baseline methods. The first, which we call the independent model, draws poses independently according to the distribution obtained by normalizing the quality scores, which is essentially the graphical model used by Sapp et al. (2010). For this model the number of poses to be sampled must be supplied by the user, so to create a level playing field we choose the number of poses in an SDPP sample YY. Since this approach does not incorporate a notion of diversity (or any correlations between selected poses whatsoever), we expect that we will frequently see multiple poses that correspond to the same person.

The second baseline is a simple non-maximum suppression model (Canny, 1986), which incorporates a heuristic for encouraging diversity. The first pose is drawn from the normalized quality model in the same manner as for the independent method. Subsequent poses, however, are constrained so that they cannot overlap with the previously selected poses, but otherwise drawn according to the quality model. We consider poses overlapping if they cover any of the same pixels when rendered. Again, the number of poses must be provided as an argument, so we use the number of poses from a sample of the SDPP. While the non-max approach can no longer generate multiple poses in the same location, it achieves this using a hard, heuristic constraint. Thus, we might expect to perform poorly when multiple people actually do overlap in the image, for example if one stands behind the other.

The SDPP, on the other hand, generates samples that prefer, but do not require poses to be spatially diverse. That is, strong visual information in the image can override our prior assumptions about the separation of distinct poses. We split our data randomly into a training set of 13 images and a test set of 60 images. Using the training set, we select values for γ\gamma, β\beta, and σ\sigma that optimize overall F1F_{1} score at radius 100 (see below), as well as distinct optimal values of β\beta for the baselines. (γ\gamma and σ\sigma are irrelevant for the baselines.) We then use each model to sample 10 sets of poses for each test image, for a total of 600 samples per model.

4.3 Results

For each sample from each of the three tested methods, we compute measures of precision and recall as well as an F1F_{1} score. In our tests, precision is measured as the fraction of predicted parts for which both endpoints are within a given radius of the endpoints of an expert-labeled part of the same type (head, left arm, and so on). We report results across a range of radii. Correspondingly, recall is the fraction of expert-labeled parts with endpoints within a given radius of a predicted part of the same type. Since the SDPP model encourages diversity, we expect to see improvements in recall at the expense of precision, compared to the independent model. F1F_{1} score is the harmonic mean of precision and recall. We compute all metrics separately for each sample, and then average the results across samples and images in the test set.

The results are shown in Figure 21a. At tight tolerances, when the radius of acceptance is small, the SDPP performs comparably to the independent and non-max samples, perhaps because the quality scores are simply unreliable at this resolution, thus diversity has little effect. As the radius increases, however, the SDPP obtains better results, significantly outperforming both baselines. Figure 21b shows the curves for just the arm parts, which tend to be more difficult to locate accurately and exhibit greater variance in orientation. Figure 21c shows the precision/recall obtained by each model. As expected, the SDPP model achieves its improved F1F_{1} score by increasing recall at the cost of precision.

For illustration, we show the SDPP sampling process for some sample images from the test set in Figure 22. The SDPP part marginals are visualized as a “cloud”, where brighter colors correspond to higher probability. From left to right, we can see how the marginals change as poses are selected during the main loop of Algorithm 11. As we saw for simple synthetic examples in Figure 8a, the SDPP discounts but does not entirely preclude poses that are similar to those already selected.

5 Random projections for SDPPs

It is quite remarkable that we can perform polynomial-time inference for SDPPs given their extreme combinatorial nature. Even so, in some cases the algorithms presented in Section 6.3 may not be fast enough. Eigendecomposing the dual representation CC, for instance, requires O(D3)O(D^{3}) time, while normalization, marginalization, and sampling, even when an eigendecomposition has been precomputed, scale quadratically in DD, both in terms of time and memory. In practice, this limits us to relatively low-dimensional diversity features ϕ\phi; for example, in our pose estimation experiments we built ϕ\phi from a fairly coarse grid of 32 points mainly for reasons of efficiency. As we move to textual data, this will become an even bigger problem, since there is no natural low-dimensional analog for feature vectors based on, say, word occurrences. In the following section we will see data where natural vectors ϕ\phi have dimension DD\approx 30,000; without dimensionality reduction, storing even a single belief propagation message would require over 200 terabytes of memory.

To address this problem, we will make use of the random projection technique described in Section 3.4, reducing the dimension of the diversity features without sacrificing the accuracy of the model. Because Theorem 3.3 depends on a cardinality condition, we will focus on kk-SDPPs. As described in Section 5, a kk-DPP is simply a DPP conditioned on the cardinality of the modeled subset Y\boldsymbol{Y}:

where ϕ(Y)\phi(Y) denotes the D×YD\times|Y| matrix formed from columns ϕ(y)\phi(\boldsymbol{y}) for yY\boldsymbol{y}\in Y. When qq and ϕ\phi factor over parts of a structure, as in Section 6.1, we will refer to this distribution as a kk-SDPP. We note in passing that the algorithms for normalization and sampling in Section 5 apply equally well to kk-SDPPs, since they depend mainly on the eigenvalues of LL, which we can obtain from CC.

Recall that Theorem 3.3 requires projection dimension

In the structured setting, N=MRN=M^{R}, thus dd must be logarithmic in the number of labels and linear in the number of parts. Under this condition, we have, with probability at least 1δ1-\delta,

In order to empirically study the effects of random projections, we test them on a simple toy application where DD is small enough that the exact model is tractable. The goal is to identify diverse, high-quality sets of travel routes between U.S. cities, where diversity is with respect to geographical location, and quality is optimized by short paths visiting the most populous or most touristy cities. Such sets of paths could be used, for example, by a politician to plan campaign routes, or by a traveler organizing a series of vacations.

We model the problem as a kk-SDPP over path structures having R=4R=4 parts, where each part is a stop along the path and can take any of M=200M=200 city values. The quality and diversity functions are factored, with a singleton factor for every individual stop and pairwise factors for consecutive pairs of stops. The quality of a singleton factor is based on the Google hit count for the assigned city, so that paths stopping in popular cities are preferred. The quality of a pair of consecutive stops is based on the distance between the assigned cities, so that short paths are preferred. In order to disallow paths that travel back and forth between the same cities, we augment the stop assignments to include arrival direction, and assign a quality score of zero to paths that return in the direction from which they came. The diversity features are only defined on the singleton factors; for a given city assignment yry_{r}, ϕr(yr)\phi_{r}(y_{r}) is just the vector of inverse distances between yry_{r} and all of the 200 cities. As a result, paths passing through the same or nearby cities appear similar, and the model prefers paths that travel through different regions of the country. We have D=200D=200.

Figure 23 shows sets of paths sampled from the kk-SDPP for various values of kk. For k=2k=2, the model tends to choose one path along the east coast and another along the west coast. As kk increases, a variety of configurations emerge; however, they continue to emphasize popular cities and the different paths remain geographically diverse.

We can now investigate the effects of random projections on this model. Figure 24 shows the L1L_{1} variational distance between the original model and the projected model (estimated by sampling), as well as the memory required to sample a set of paths for a variety of projection dimensions dd. As predicted by Theorem 3.3, only a relatively small number of projection dimensions are needed to obtain a close approximation to the original model. Past d25d\approx 25, the rate of improvement due to increased dimension falls off dramatically; meanwhile, the required memory and running time start to become significant. Figure 24 suggests that aggressive use of random projections, like those we employ in the following section, is not only theoretically but also empirically justified.

6 Experiments: threading graphs

In this section we put together many of the techniques introduced in this paper in order to complete a novel task that we refer to as graph threading (Gillenwater et al., 2012). The goal is to extract from a large directed graph a set of diverse, salient threads, or singly-connected chains of nodes. Depending on the construction of the graph, such threads can have various semantics. For example, given a corpus of academic literature, high-quality threads in the citation graph might correspond to chronological chains of important papers, each building on the work of the last. Thus, graph threading could be used to identify a set of significant lines of research. Or, given a collection of news articles from a certain time period, where each article is a node connected to previous, related articles, we might want to display the most significant news stories from that period, and for each story provide a thread that contains a timeline of its major events. We experiment on data from these two domains in the following sections. Other possibilities might include discovering trends on social media sites, for example, where users can post image or video responses to each other, or mining blog entries for important conversations through trackback links. Figure 25 gives an overview of the graph threading task for document collections.

Generally speaking, graph threading offers a means of gleaning insights from collections of interrelated objects—for instance, people, documents, images, events, locations, and so on—that are too large and noisy for manual examination. In contrast to tools like search, which require the user to specify a query based on prior knowledge, a set of threads provides an immediate, concise, high-level summary of the collection, not just identifying a set of important objects but also conveying the relationships between them. As the availability of such datasets continues to grow, this kind of automated analysis will be key in helping us to efficiently and effectively navigate and understand the information they contain.

Research from to the Topic Detection and Tracking (TDT) program (Wayne, 2000) has led to useful methods for tasks like link detection, topic detection, and topic tracking that can be seen as subroutines for graph threading on text collections. Graph threading with kk-SDPPs, however, addresses these tasks jointly, using a global probabilistic model with a tractable inference algorithm.

Other work in the topic tracking literature has addressed related tasks (Mei and Zhai, 2005; Blei and Lafferty, 2006; Leskovec et al., 2009). In particular, Blei and Lafferty (2006) proposed dynamic topic models (DTMs), which, given a division of text documents into time slices, attempt to fit a generative model where topics evolve over time, and documents are drawn from the topics available at the time slice during which they were published. The evolving topics found by a DTM can be seen as threads of a sort, but in contrast to graph threading they are not composed of actual items in the dataset (in this case, documents). In Section 6.6.4 we will return to this distinction when we compare kk-SDPP threading to a DTM baseline.

The information retrieval community has produced other methods for extracting temporal information from document collections. Swan and Jensen (2000) proposed a system for finding temporally clustered named entities in news text and presenting them on a timeline. Allan et al. (2001) introduced the task of temporal summarization, which takes as input a stream of news articles related to a particular topic, and then seeks to extract sentences describing important events as they occur. Yan et al. (2011) evaluated methods for choosing sentences from temporally clustered documents that are relevant to a query. In contrast, graph threading seeks not to extract grouped entities or sentences, but instead to organize a subset of the objects (documents) themselves into threads, with topic identification as a side effect.

Some prior work has also focused more directly on threading. Shahaf and Guestrin (2010) and Chieu and Lee (2004) proposed methods for selecting individual threads, while Shahaf et al. (2012) recently proposed metro maps as alternative structured representations of related news stories. Metro maps are effectively sets of non-chronological threads that are encouraged to intersect and, in doing so, generate a map of events and topics. However, these approaches assume some prior knowledge about content. Shahaf and Guestrin (2010), for example, assume the thread endpoints are specified, and Chieu and Lee (2004) require a set of query words. Likewise, because they build metro maps individually, Shahaf et al. (2012) implicitly assume that the collection is filtered to a single topic, perhaps from a user query. These inputs make it possible to quickly pare down the document graph. In contrast, we will apply graph threading to very large graphs, and consider all possible threads.

6.2 Setup

In order to be as useful as possible, the threads we extract from a data graph need to be both high quality, reflecting the most important parts of the collection, and diverse, so that they cover distinct aspects of the data. In addition, we would like to be able to directly control both the length and the number of threads that we return, since different contexts might necessitate different settings. Finally, to be practical our method must be efficient in both time and memory use. kk-SDPPs with random projections allow us to simultaneously achieve all of these goals.

Given a directed graph on MM vertices with edge set EE and a real-valued weight function w()w(\cdot) on nodes and edges, define the weight of a thread y=(y1,y2,,yR),\boldsymbol{y}=(y_{1},y_{2},\dots,y_{R}), (yr,yr+1)E(y_{r},y_{r+1})\in E by

We can use ww to define a simple log-linear quality model for our kk-SDPP:

where β\beta is a hyperparameter controlling the dynamic range of the quality scores. We fix the value of β\beta on a validation set in our experiments.

In some cases it might also be convenient to have diversity features on edges of the graph as well as nodes. If so, they can be accommodated without much difficulty; however, for simplicity we proceed with the setup above.

We assume that RR, kk, and the projection dimension dd are provided; the first two depend on application context, and the third, as discussed in Section 6.5, is a tradeoff between computational efficiency and faithfulness to the original model. To generate diverse thread samples, we first project the diversity features ϕ\phi by a random d×Dd\times D matrix GG whose entries are drawn independently and identically from N(0,1d)\mathcal{N}(0,\frac{1}{d}). We then apply second-order message passing to compute the dual representation CC, as in Section 6.3.1. After eigendecomposing CC, which is only d×dd\times d due to the projection, we can run the first phase of the kk-DPP sampling algorithm from Section 5.2.2 to choose a set V^\hat{V} of eigenvectors, and finally complete the SDPP sampling algorithm in Section 6.3.2 to obtain a set of kk threads YY. We now apply this model to two datasets; one is a citation graph of computer science papers, and the other is a large corpus of news text.

6.3 Academic citation data

The Cora dataset comprises a large collection of approximately 200,000 academic papers on computer science topics, including citation information (McCallum et al., 2000). We construct a directed graph with papers as nodes and citations as edges, and then remove papers with missing metadata or zero outgoing citations, leaving us with 28,155 papers. The average out-degree is 3.263.26 citations per paper, and 0.0110.011% of the total possible edges are present in the graph.

To obtain useful threads, we set edge weights to reflect the degree of textual similarity between the citing and the cited paper, and node weights to correspond with a measure of paper “importance”. Specifically, the weight of edge (a,b)(a,b) is given by the cosine similarity metric, which for two documents aa and bb is the dot product of their normalized tf-idf vectors, as defined in Section 4.2.1:

Here WW is a subset of the words found in the documents. We select WW by filtering according to document frequency; that is, we remove words that are too common, appearing in more than 10% of papers, or too rare, appearing in only one paper. After filtering, there are 50,912 unique words.

The node weights are given by the LexRank score of each paper (Erkan and Radev, 2004). The LexRank score is the stationary distribution of the thresholded, binarized, row-normalized matrix of cosine similarities, plus a damping term, which we fix to 0.150.15. LexRank is a measure of centrality, so papers that are closely related to many other papers will receive a higher score.

Finally, we design the diversity feature function ϕ\phi to encourage topical diversity. Here we apply cosine similarity again, representing a document by the 1,000 documents to which it is most similar. This results in binary ϕ\phi of dimension D=M=28,155D=M=28,155 with exactly 1,000 non-zeros; ϕl(yr)=1\phi_{l}(y_{r})=1 implies that ll is one of the 1,000 most similar documents to yry_{r}. Correspondingly, the dot product between the diversity features of two documents is proportional to the fraction of top-1,000 documents they have in common. In order to make kk-SDPP inference efficient, we project ϕ\phi down to d=50d=50 dimensions.

Figure 26 illustrates the behavior of the model when we set k=4k=4 and R=5R=5. Samples from the model, like the one presented in the figure, not only offer some immediate intuition about the types of papers contained in the collection, but also, upon examining individual threads, provide a succinct illustration of the content and development of each area. Furthermore, the sampled threads cover distinct topics, standing apart visually in Figure 26 and exhibiting diverse salient terms.

6.4 News articles

Our news dataset comprises over 200,000 articles from the New York Times, collected from 2005-2007 as part of the English Gigaword corpus (Graff and Cieri, 2009). We split the articles into six groups, with six months’ worth of articles in each group. Because the corpus contains a significant amount of noise in the form of articles that are short snippets, lists of numbers, and so on, we filter the results by discarding articles more than two standard deviations longer than the mean article, articles less than 400 words, and articles whose fraction of non-alphabetic words is more than two standard deviations above the mean. On average, for each six-month period we are left with 34,504 articles.

For each time period, we generate a graph with articles as nodes. As for the citations dataset, we use cosine similarity to define edge weights. The subset of words WW used to compute cosine similarity contains all words that appear in at least 20 articles and at most 15% of the articles. Across the six time periods, this results in an average of 36,356 unique words. We include in our graph only those edges with cosine similarity of at least 0.1; furthermore, we require that edges go forward in time to enforce the chronological ordering of threads. The resulting graphs have an average of 0.320.32% of the total possible edges, and an average degree of 107. As before, we use LexRank for node weights, and the top-1000 similar documents to define a binary feature function ϕ\phi. We add a constant feature ρ\rho to ϕ\phi, which controls the overall degree of repulsion; large values of ρ\rho make all documents more similar to one another. We set ρ\rho and the quality model hyperparameters to maximize a cosine similarity evaluation metric (see Section 6.6.4), using the data from the first half of 2005 as a development set. Finally, we randomly project the diversity features from D34,500D\approx 34,500 to d=50d=50 dimensions. For all of the following experiments, we use k=10k=10 and R=8R=8. All evaluation metrics we report are averaged over 100100 random samples from the model.

In order to convey the scale and content of the graphs built from news data, we provide some high-resolution renderings. Figure 27 shows the graph neighborhood of a single article node from our development set. Each node represents an article and is annotated with the corresponding headline; the size of each node reflects its weight, as does the thickness of an edge. The horizontal position of a node corresponds to the time at which the article was published, from left to right; the vertical positions are optimized for readability. In the digital version of this paper, Figure 27 can be zoomed in order to read the headlines; in hardcopy, however, it is likely to be illegible. As an alternative, an online, zoomable version of the figure is available at http://zoom.it/GUCR.

Visualizing the entire graph is quite challenging since it contains tens of thousands of nodes and millions of edges; placing such a figure in the paper would be impractical since the computational demands of rendering it and the zooming depth required to explore it would exceed the abilities of modern document viewers. Instead, we provide an online, zoomable version based upon a high-resolution (540 megapixel) rendering, available at http://zoom.it/jOKV. Even at this level of detail, only 1% of the edges are displayed; otherwise they become visually indistinct. As in Figure 27, each node represents an article and is sized according to its weight and overlaid with its headline. The horizontal position corresponds to time, ranging from January 2005 (on the left) to June 2005 (on the right). The vertical positions are determined by similarity to a set of threads sampled from the kk-SDPP, which are rendered in color.

We will compare the kk-SDPP model to two natural baselines.

kk-means baseline. A simple method for this task is to split each six-month period of articles into RR equal-sized time slices, and then apply kk-means clustering to each slice, using cosine similarity at the clustering metric. We can then select the most central article from each cluster to form the basis of a set of threads. The kk articles chosen from time slice rr are matched one-to-one with those from slice r1r-1 by computing the pairing that maximizes the average cosine similarity of the pairs—that is, the coherence of the threads. Repeating this process for all rr yields a set of kk threads of length RR, where no two threads will contain the same article. However, because clustering is performed independently for each time slice, it is likely that the threads will sometimes exhibit discontinuities when the articles chosen at successive time slices do not naturally align.

DTM baseline. A natural extension, then, is the dynamic topic model (DTM) of Blei and Lafferty (2006), which explicitly attempts to find topics that are smooth through time. We use publicly available codehttp://code.google.com/p/princeton-statistical-learning/ to fit DTMs with the number of topics set to kk and with the data split into RR equal time slices. We set the hyperparameters to maximize the cosine similarity metric (see Section 6.6.4) on our development set. We then choose, for each topic at each time step, the document with the highest per-word probability of being generated by that topic. Documents from the same topic form a single thread.

Figure 28 shows some of the threads sampled randomly from the kk-SDPP for our development set, and Figure 29 shows the same for threads produced by the DTM baseline. An obvious distinction is that topic model threads always span nearly the entire time period, selecting one article per time slice as required by the form of the model, while the DPP can select threads covering only the relevant span. Furthermore, the headlines in the figures suggest that the kk-SDPP produces more tightly focused, narrative threads due to its use of the data graph, while the DTM threads, though topically related, tend not to describe a single continuous news story. This distinction, which results from the fact that topic models are not designed with threading in mind, and so do not take advantage of the explicit relation information given by the graph, means that kk-SDPP threads often form a significantly more coherent representation of the news collection.

We provide a quantitative evaluation of the threads generated by our baselines and sampled from the kk-SDPP by comparing them to a set of human-generated news summaries. The human summaries are not threaded; they are flat, approximately daily news summaries found in the Agence France-Presse portion of the Gigaword corpus, distinguished by their “multi” type tag. The summaries generally cover world news, which is only a subset of the contents of our dataset. Nonetheless, they allow us to provide an extrinsic evaluation for this novel task without generating gold standard timelines manually, which is a difficult task given the size of the corpus. We compute four metrics:

Cosine similarity. We concatenate the human summaries over each six-month period to obtain a target tf-idf vector, concatenate the set of threads to be evaluated to obtain a predicted tf-idf vector, and then compute the cosine similarity (in percent) between the target and predicted vectors. All hyperparameters are chosen to optimize this metric on a validation set.

ROUGE-1, 2, and SU4. As described in Section 4.2.1, ROUGE is an automatic evaluation metric for text summarization based on nn-gram overlap statistics (Lin, 2004). We report three standard variants.

Table 10 shows the results of these comparisons, averaged over all six half-year intervals. Under each metric, the kk-SDPP produces threads that more closely resemble human summaries.

An important distinction between the baselines and the kk-SDPP is that the former are topic-oriented, choosing articles that relate to broad subject areas, while the kk-SDPP approach is story-oriented, chaining together articles with direct individual relationships. An example of this distinction can be seen in Figures 28 and 29.

To obtain a large-scale evaluation of this type of thread coherence, we employ Mechanical Turk, on online marketplace for inexpensively and efficiently completing tasks requiring human judgment. We asked Turkers to read the headlines and first few sentences of each article in a timeline and then rate the overall narrative coherence of the timeline on a scale of 1 (“the articles are totally unrelated”) to 5 (“the articles tell a single clear story”). Five separate Turkers rated each timeline. The average ratings are shown in the left column of Table 11; the kk-SDPP timelines are rated as significantly more coherent, while kk-means does poorly since it has no way to ensure that clusters are similar between time slices.

In addition, we asked Turkers to evaluate threads implicitly by performing a second task. (This also had the side benefit of ensuring that Turkers were engaged in the rating task and did not enter random decisions.) We displayed timelines into which two additional “interloper” articles selected at random had been inserted, and asked users to remove the two articles that they thought should be removed to improve the flow of the timeline. A screenshot of the task is provided in Figure 30. Intuitively, the true interlopers should be selected more often when the original timeline is coherent. The average number of interloper articles correctly identified is shown in the right column of Table 11.

Finally, assuming that tf-idf and feature values have been computed in advance (this process requires approximately 160 seconds), we report in Table 12 the time required to produce a set of threads on the development set. This measurement includes clustering for the kk-means baseline, model fitting for the DTM baseline, and random projections, computation of the covariance matrix, and sampling for the kk-SDPP. The tests were run on a machine with eight Intel Xeon E5450 cores and 32G of memory. Thanks to the use of random projections, the kk-SDPP is not only the most faithful to human news summaries, but also the fastest by a large margin.

Conclusion

We believe that DPPs offer exciting new possibilities for a wide range of practical applications. Unlike heuristic diversification techniques, DPPs provide coherent probabilistic semantics, and yet they do not suffer from the computational issues that plague existing models when negative correlations arise. Before concluding, we briefly mention two open technical questions, as well as some possible directions for future research.

The Shannon entropy of the DPP with marginal kernel KK is given by

While numerical simulation strongly suggests that the conjecture is true, to our knowledge no proof currently exists.

2 Open question: higher-order sums

In order to calculate, for example, the Hellinger distance between a pair of DPPs, it would be useful to be able to compute quantities of the form

for p>1p>1. To our knowledge it is not currently known whether it is possible to compute these quantities efficiently.

3 Research directions

A variety of interesting machine learning questions remain for future research.

Would DPPs based on Hermitian or asymmetric kernels offer worthwhile modeling advantages?

Is there a simple characterization of the conditional independence relations encoded by a DPP?

Can we perform DPP inference under more complicated constraints on allowable sets? (For instance, if the items correspond to edges in a graph, we might only consider sets that comprise a valid matching.)

How can we learn the similarity kernel for a DPP (in addition to the quality model) from labeled training data?

How can we efficiently (perhaps approximately) work with SDPPs over loopy factor graphs?

Can SDPPs be used to diversify nn-best lists and improve reranking performance, for instance in parsing or machine translation?

References