Density Estimation in Infinite Dimensional Exponential Families
Bharath Sriperumbudur, Kenji Fukumizu, Arthur Gretton, Aapo Hyvärinen, Revant Kumar
Introduction
In this paper, we consider an infinite dimensional generalization (Canu and Smola, 2005; Fukumizu, 2009) of (1),
where the function space is defined as
being the cumulant generating function, and a reproducing kernel Hilbert space (RKHS) (Aronszajn, 1950) with as its reproducing kernel. While various generalizations are possible for different choices of (e.g., an Orlicz space as in Pistone and Sempi, 1995), the connection of to the natural exponential family in (1) is particularly enlightening when is an RKHS. This is due to the reproducing property of the kernel, , through which takes the role of the sufficient statistic. In fact, it can be shown (see Section 3 and Example 1 for more details) that every is generated by induced by a finite dimensional RKHS , and therefore the family with being an infinite dimensional RKHS is a natural infinite dimensional generalization of . Furthermore, this generalization is particularly interesting as in contrast to , it can be shown that is a rich class of densities (depending on the choice of and therefore ) that can approximate a broad class of probability densities arbitrarily well (see Propositions 1, 13 and Corollary 2). This generalization is not only of theoretical interest, but also has implications for statistical and machine learning applications. For example, in Bayesian non-parametric density estimation, the densities in are chosen as prior distributions on a collection of probability densities (e.g., see van der Vaart and van Zanten, 2008). has also found applications in nonparametric hypothesis testing (Gretton et al., 2012; Fukumizu et al., 2008) and dimensionality reduction (Fukumizu et al., 2004, 2009) through the mean and covariance operators, which are obtained as the first and second Fréchet derivatives of (see Fukumizu, 2009, Section 1.2.3). Recently, the infinite dimensional exponential family, has been used to develop a gradient-free adaptive MCMC algorithm based on Hamiltonian Monte Carlo (Strathmann et al., 2015) and also has been used in the context of learning the structure of graphical models (Sun et al., 2015).
Motivated by the richness of the infinite dimensional generalization and its statistical applications, it is of interest to model densities by , and therefore the goal of this paper is to estimate unknown densities by elements in when is an infinite dimensional RKHS. Formally, given i.i.d. random samples drawn from an unknown density , the goal is to estimate through . Throughout the paper, we refer to case of as well-specified, in contrast to the misspecified case where . The setting is useful because is a rich class of densities that can approximate a broad class of probability densities arbitrarily well, hence it may be widely used in place of non-parametric density estimation methods (e.g., kernel density estimation (KDE)). In fact, through numerical simulations, we show in Section 6 that estimating through performs better than KDE, and that the advantage of the proposed estimator grows with increasing dimensionality.
A related work was carried out by Barron and Sheu (1991)—also see references therein—where the goal is to estimate a density, by approximating its logarithm as an expansion in terms of basis functions, such as polynomials, splines or trigonometric series. Similar to Fukumizu (2009), Barron and Sheu proposed the ML estimator , where
and is the linear space of dimension spanned by the chosen basis functions. Under the assumption that has square-integrable derivatives up to order , they showed that with for each of the approximating families, where is the KL divergence between and . Similar work was carried out by Gu and Qiu (1993), who assumed that lies in an RKHS, and proposed an estimator based on penalized MLE, with consistency and rates established in Jensen-Shannon divergence. Though these results are theoretically interesting, these estimators are obtained via a procedure similar to that in Fukumizu (2009), and therefore suffers from the practical drawbacks discussed above.
The discussion so far shows that the MLE approach to learning results in estimators that are of limited practical interest. To alleviate this, one can treat the problem of estimating in a completely non-parametric fashion by using KDE, which is well-studied (Tsybakov, 2009, Chapter 1) and easy to implement. This approach ignores the structure of , however, and is known to perform poorly for moderate to large (Wasserman, 2006, Section 6.5) (see also Section 6 of this paper).
To understand the advantages associated with the score matching method, let us consider the problem of density estimation where the data generating distribution (say ) belongs to in (1). In other words, given random samples drawn i.i.d. from , the goal is to estimate as , and use as an estimator of . While the MLE approach is well-studied and enjoys nice statistical properties in asymptopia (i.e., asymptotically unbiased, efficient, and normally distributed), the computation of can be intractable in many situations as discussed above. In particular, this is the case for where for all , , and the functional form of is known (as a function of and ); yet we do not know how to easily compute , which is often analytically intractable. In this setting (which is exactly the setting of this paper), assuming to be differentiable (w.r.t. ), and , in (3) reduces to
through integration by parts (see Hyvärinen, 2005, Theorem 1), under appropriate regularity conditions on and for all . Here . The main advantage of the objective in (3) (and also (4)) is that when it is applied to the situation discussed above where , is independent of , and an estimate of can be obtained by simply minimizing the empirical counterpart of , given by
Since is also independent of , may be easily computable, unlike the MLE. We would like to highlight that while the score matching approach may have computational advantages over MLE, it only estimates up to the scaling factor , and therefore requires the approximation or computation of through numerical integration to estimate . Note that this issue (of computing through numerical integration) exists even with MLE, but not with KDE. In score matching, however, numerical integration is needed only once, while MLE would typically require a functional form of the log-partition function which is approximated through numerical integration at every step of an iterative optimization algorithm (for example, see (2)), thus leading to major computational savings. An important application that does not require the computation of is in finding modes of the distribution, which has recently become very popular in image processing (Comaniciu and Meer, 2002), and has already been investigated in the score matching framework (Sasaki et al., 2014). Similarly, in sampling methods such as sequential Monte Carlo (Doucet et al., 2001), it is often the case that the evaluation of unnormalized densities is sufficient to calculate required importance weights.
2 Contributions
(i) We present an estimate of in the well-specified case through the minimization of Fisher divergence, in Section 4. First, we show that estimating using the score matching method reduces to estimating by solving a simple finite-dimensional linear system (Theorems 4 and 5). Hyvärinen (2007) obtained a similar result for where the estimator is obtained by solving a linear system, which in the case of Gaussian family matches the MLE (Hyvärinen, 2005). The estimator obtained in the infinite dimensional case is not a simple extension of its finite-dimensional counterpart, however, as the former requires an appropriate regularizer (we use ) to make the problem well-posed. We would like to highlight that to the best of our knowledge, the proposed estimator is the first practically computable estimator of with consistency guarantees (see below). (ii) In contrast to Hyvärinen (2007) where no guarantees on consistency or convergence rates are provided for the density estimator in , we establish in Theorem 6 the consistency and rates of convergence for the proposed estimator of , and use these to prove consistency and rates of convergence for the corresponding plug-in estimator of (Theorems 7 and B.2), even when is infinite dimensional. Furthermore, while the estimator of (and therefore ) is obtained by minimizing the Fisher divergence, the resultant density estimator is also shown to be consistent in KL divergence (and therefore in Hellinger and total-variation distances) and we provide convergence rates in all these distances.
Formally, we show that the proposed estimator is converges as
if for some , where denotes the range or image of an operator , , and is a Hilbert-Schmidt operator on (see Theorem 4) with being the reproducing kernel and denoting the tensor product. When is a finite-dimensional RKHS, we show that the estimator enjoys parametric rates of convergence, i.e.,
Note that the convergence rates are obtained under a non-classical smoothness assumption on , namely that it lies in the image of certain fractional power of , which reduces to a more classical assumption if we choose to be a Matérn kernel (see Section 2 for its definition), as it induces a Sobolev space. In Section 4.2, we discuss in detail the smoothness assumption on for the Gaussian (Example 2) and Matérn (Example 3) kernels. Another interesting point to observe is that unlike in the classical function estimation methods (e.g., kernel density estimation and regression), the rates presented above for the proposed estimator tend to saturate for ( w.r.t. ), with the best rate attained at ( w.r.t. ), which means the smoothness of is not fully captured by the estimator. Such a saturation behavior is well-studied in the inverse problem literature (Engl et al., 1996) where it has been attributed to the choice of regularizer. In Section 4.3, we discuss alternative regularization strategies using ideas from Bauer et al. (2007), which covers non-parametric least squares regression: we show that for appropriately chosen regularizers, the above mentioned rates hold for any , and do not saturate for the aforementioned ranges of (see Theorem 9). (iii) In Section 5, we study the problem of density estimation in the misspecified setting, i.e., , which is not addressed in Hyvärinen (2007) and Fukumizu (2009). Using a more sophisticated analysis than in the well-specified case, we show in Theorem 12 that as . Under an appropriate smoothness assumption on (see the statement of Theorem 12 for details), we show that as along with a rate for this convergence, even though . However, unlike in the well-specified case, where the consistency is obtained not only in but also in other distances, we obtain convergence only in for the misspecified case. Note that while Barron and Sheu (1991) considered the estimation of in the misspecified setting, the results are restricted to the approximating families consisting of polynomials, splines, or trigonometric series. Our results are more general, as they hold for abstract RKHSs. (iv) In Section 6, we present preliminary numerical results comparing the proposed estimator with KDE in estimating a Gaussian and mixture of Gaussians, with the goal of empirically evaluating performance as gets large for a fixed sample size. In these two estimation problems, we show that the proposed estimator outperforms KDE, and the advantage grows as increases. Inspired by this preliminary empirical investigation, our proposed estimator (or computationally efficient approximations) has been used by Strathmann et al. (2015) in a gradient-free adaptive MCMC sampler, and by Sun et al. (2015) for graphical model structure learning. These applications demonstrate the practicality and performance of the proposed estimator.
Finally, we would like to make clear that our principal goal is not to construct density estimators that improve uniformly upon KDE, but to provide a novel flexible modeling technique for approximating an unknown density by a rich parametric family of densities, with the parameter being infinite dimensional, in contrast to the classical approach of finite dimensional approximation.
Various notations and definitions that are used throughout the paper are collected in Section 2. The proofs of the results are provided in Section 8, along with some supplementary results in an appendix.
Definitions & Notation
where denotes the imaginary unit .
where is the Gamma function, and is the modified Bessel function of the third kind of order ( controls the smoothness of ).
Approximation of Densities by 𝒫𝒫\mathcal{P}
In this section, we first show that every finite dimensional exponential family, is generated by the family induced by a finite dimensional RKHS, which naturally leads to the infinite dimensional generalization of when is an infinite dimensional RKHS. Next, we investigate the approximation properties of in Proposition 1 and Corollary 2 when is an infinite dimensional RKHS.
The following are some popular examples of probability distributions that belong to . Here we show the corresponding RKHSs that generate these distributions. In some of these examples, we choose and ignore the fact that is a probability distribution as assumed in the definition of .
Beta: , .
Binomial: , , .
While Example 1 shows that all popular probability distributions are contained in for an appropriate choice of finite-dimensional , it is of interest to understand the richness of (i.e., what class of distributions can be approximated arbitrarily well by ?) when is an infinite dimensional RKHS. This is addressed by the following result, which is proved in Section 8.1.
Then is dense in w.r.t. Kullback-Leibler divergence, total variation ( norm) and Hellinger distances. In addition, if for some , then is also dense in w.r.t. norm.
Suppose and (5) holds. Then is dense in w.r.t. KL divergence, TV and Hellinger distances. Moreover, if for some , then is also dense in w.r.t. norm.
By choosing to be compact and to be a uniform distribution on , Corollary 2 reduces to an easily interpretable result that any continuous density on can be approximated arbitrarily well by densities in in KL, Hellinger and () distances.
Similar to the results so far, an approximation result for can also be obtained w.r.t. Fisher divergence (see Proposition 13). Since this result is heavily based on the notions and results developed in Section 5, we defer its presentation until that section. Briefly, this result states that if is sufficiently rich (i.e., dense in an appropriate class of functions), then any with can be approximated arbitrarily well by elements in w.r.t. Fisher divergence, where .
Density Estimation in 𝒫𝒫\mathcal{P}: Well-specified Case
In this section, we present our score matching estimator for an unknown density (well-specified case) from i.i.d. random samples drawn from it. This involves choosing the minimizer of the (empirical) Fisher divergence between and as the estimator, which we show in Theorem 5 to be obtained by solving a simple finite-dimensional linear system. In contrast, we would like to remind the reader that the MLE is infeasible in practice due to the difficulty in handling . The consistency and convergence rates of and the plug-in estimator are provided in Section 4.1 (see Theorems 6 and 7). Before we proceed, we list the assumptions on , and that we need in our analysis.
is continuously extendible to . is twice continuously differentiable on with continuous extension of to for .
for and as , for all .
(-Integrability) For some and , and where
Under these assumptions, the following result—proved in Section 8.3—shows that the problem of estimating through the minimization of Fisher divergence reduces to the problem of estimating through a weighted least squares minimization in (see parts (i) and (ii)). This motivates the minimization of the regularized empirical weighted least squares (see part (iv)) to obtain an estimator of , which is then used to construct the plug-in estimate of .
Suppose (A)–(D) hold with . Then for all . In addition, the following hold. (i) For all ,
where , is a trace-class positive operator with
and satisfies . (iii) For any , a unique minimizer of over exists and is given by
(iv) (Estimator of ) Given samples drawn i.i.d. from , for any , the unique minimizer of over exists and is given by
where , and
An advantage of the alternate formulation of in Theorem 4(ii) over (6) is that it provides a simple way to obtain an empirical estimate of —by replacing and by their empirical estimators, and respectively—from finite samples drawn i.i.d. from , which is then used to obtain an estimator of . Note that the empirical estimate of , i.e., depends only on and which in turn depend on the known quantities, and , and therefore in Theorem 4(iv) should in principle be computable. In practice, however, it is not easy to compute the expression for as it involves solving an infinite dimensional linear system. In Theorem 5 (proved in Section 8.4), we provide an alternative expression for as a solution of a simple finite-dimensional linear system (see (7) and (8)), using the general representer theorem (see Theorem A.2). It is interesting to note that while the solution to in Theorem 4(ii) is obtained by solving a non-linear system, (the system is non-linear as depends on which in turn depends on ), its estimator proposed in Theorem 4, is obtained by solving a simple linear system. In addition, we would like to highlight the fact that the proposed estimator, is precisely the Tikhonov regularized solution (which is well-studied in the theory of linear inverse problems) to the ill-posed linear system . We further discuss the choice of regularizer in Section 4.3 using ideas from the inverse problem literature.
An important remark we would like to make about Theorem 4 is that though in (6) is valid only for , as it is obtained from where , the expression is valid for any , as it is finite under the assumption that (D) holds with . Therefore, in Theorem 4(iii, iv), and are obtained by minimizing and over instead of over , as the latter does not yield a nice expression (unlike and , respectively). However, there is no guarantee that , and so the density estimator may not be valid. While this is not an issue when studying the convergence of (see Theorem 6), the convergence of to (in various distances) needs to be handled slightly differently depending on whether the kernel is bounded or not (see Theorems 7 and B.2). Note that when the kernel is bounded, we obtain , which implies is valid.
Let , where is defined in Theorem 4(iv) and . Then
where is defined in Theorem 4(iv) and is obtained by solving
with
We would like to highlight that though requires solving a simple linear system in (8), it can still be computationally intensive when and are large as is a matrix. This is still a better scenario than that of MLE, however, since computationally efficient methods exist to solve large linear systems such as (8), whereas MLE can be intractable due to the difficulty in handling the log-partition function (though it can be approximated). On the other hand, MLE is statistically well-understood, with consistency and convergence rates established in general for the problem of density estimation (van de Geer, 2000) and in particular for the problem at hand (Fukumizu, 2009). In order to ensure that and are statistically useful, in the following section, we investigate their consistency and convergence rates under some smoothness conditions on .
Suppose (A)–(D) with hold. (i) If , then (ii) If for some , then for ,
(iii) If , then for , as .
(i) While Theorem 6 (proved in Section 8.5) provides an asymptotic behavior for under conditions that depend on (and are therefore not easy to check in practice), a non-asymptotic bound on that holds for all can be obtained under stronger assumptions through an application of Bernstein’s inequality in separable Hilbert spaces. For the sake of simplicity, we provided asymptotic results which are obtained through an application of Chebyshev’s inequality. (ii) The proof of Theorem 6(i) involves decomposing into an estimation error part, , and an approximation error part, , where . While as , and without any assumptions on (see the proof in Section 8.5 for details), it is not reasonable to expect as without assuming . This is because, if lies in the null space of , then is zero irrespective of and therefore cannot approximate . (iii) The condition is difficult to check in practice as it depends on (which in turn depends on ). However, since the null space of is just constant functions if the kernel is bounded and (see Lemma 14 in Section 8.6 for details), assuming yields that and therefore consistency can be attained under conditions that are easy to impose in practice. As mentioned in Remark 3(iii), the condition ensures identifiability and a sufficient condition for it to hold is , which is satisfied by Gaussian, Matérn and inverse multiquadric kernels. (iv) It is well known that convergence rates are possible only if the quantity of interest (here ) satisfies some additional conditions. In function estimation, this additional condition is classically imposed by assuming to be sufficiently smooth, e.g., lies in a Sobolev space of certain smoothness. By contrast, the smoothness condition in Theorem 6(ii) is imposed in an indirect manner by assuming for some —so that the results hold for abstract RKHSs and not just Sobolev spaces—which then provides a rate, with the best rate being that is attained when . While such a condition has already been used in various works (Caponnetto and Vito, 2007; Smale and Zhou, 2007; Fukumizu et al., 2013) in the context of non-parametric least squares regression, we explore it in more detail in Proposition 8, and Examples 2 and 3. Note that this condition is common in the inverse problem theory (see Engl, Hanke, and Neubauer, 1996), and it naturally arises here through the connection of being a Tikhonov regularized solution to the ill-posed linear system . An interesting observation about the rate is that it does not improve with increasing (for ), in contrast to the classical results in function estimation (e.g., kernel density estimation and kernel regression) where the rate improves with increasing smoothness. This issue is discussed in detail in Section 4.3. (v) Since only if is finite-dimensional, we recover the parametric rate of in a finite-dimensional situation with an automatic choice for as .
While Theorem 6 provides statistical guarantees for parameter convergence, the question of primary interest is the convergence of to . This is guaranteed by the following result, which is proved in Section 8.6.
Suppose (A)–(D) with hold and . Assume . Then the following hold: (i) For any with ,
In addition, if for some , then for ,
as where . (ii) In addition, if for some , then for ,
(iii) If , then and with .
While Theorem 7 addresses the case of bounded kernels, the case of unbounded kernels requires a technical modification. The reason for this modification, as alluded to in the discussion following Theorem 4, is due to the fact that may not be in when is unbounded, and therefore the corresponding density estimator, may not be well-defined. In order to keep the main ideas intact, we discuss the unbounded case in detail in Section B.2 in Appendix B.
2 Range Space Assumption
While Theorems 6 and 7 are satisfactory from the point of view of consistency, we believe the presented rates are possibly not minimax optimal since these rates are valid for any RKHS that satisfies the conditions (A)–(D) and does not capture the smoothness of (and therefore the corresponding ). In other words, the rates presented in Theorems 6 and 7 should depend on the decay rate of the eigenvalues of which in turn effectively captures the smoothness of . However, we are not able to obtain such a result—see the remark following the proof of Theorem 6 for a discussion. While these rates do not reflect the intrinsic smoothness of , they are obtained under the smoothness assumption, i.e., range space condition that for some . This condition is quite different from the classical smoothness conditions that appear in non-parametric function estimation. While the range space assumption has been made in various earlier works (e.g., Caponnetto and Vito (2007); Smale and Zhou (2007); Fukumizu et al. (2013) in the context of non-parametric least square regression), in the following, we investigate the implicit smoothness assumptions that it makes on in our context. To this end, first it is easy to show (see the proof of Proposition B.3 in Section B.3) that
Then implies .
In the following, we apply the above result in two examples involving Gaussian and Matérn kernels to get insights into the range space assumption.
Let with as its corresponding RKHS (see Section 2 for its definition). By Proposition 8, it is easy to verify that implies for . Since for (i.e., Gaussian RKHSs are nested), ensures that lies in for arbitrary small .
While Example 3 provides some understanding about the minimax optimality of under additional assumptions on , the problem is not completely resolved. In the following section, however, we show that the rate in Theorem 6 is not optimal for , and that improved rates can be obtained by choosing the regularizer appropriately.
3 Choice of Regularizer
We understand from the characterization of in (9) that larger values yield smoother functions in . However, the smoothness of for is not captured in the rates in Theorem 6(ii), where the rate saturates at providing the best possible rate of (irrespective of the size of ). This is unsatisfactory on the part of the estimator, as it does not effectively capture the smoothness of , i.e., the estimator is not adaptive to the smoothness of . We remind the reader that the estimator is obtained by minimizing the regularized empirical Fisher divergence (see Theorem 4(iv)) yielding , which can be seen as a heuristic to solve the (non-linear) inverse problem (see Theorem 4(ii)) from finite samples, by replacing and with their empirical counterparts. This heuristic, which ensures that the finite sample inverse problem is well-posed, is popular in inverse problem literature under the name of Tikhonov regularization (Engl et al., 1996, Chapter 5). Note that Tikhonov regularization helps to make the ill-posed inverse problem a well-posed one by approximating by , , where appears as the inverse of the eigenvalues of while computing . In other words, if is invertible, then an estimate of can be obtained as , i.e., where and are the eigenvalues and eigenvectors of respectively. However, being a rank operator defined on (which can be infinite dimensional) is not invertible and therefore the regularized estimator is constructed as where is defined through functional calculus (see Engl, Hanke, and Neubauer, 1996, Section 2.3) as
The constant is called the qualification of which is what determines the point of saturation of . We show in Theorem 9 that if has a finite qualification, then the resultant estimator cannot fully exploit the smoothness of and therefore the rate of convergence will suffer for . Given that satisfies (E), we construct our estimator of as
Note that the above estimator can be obtained by using the data dependent regularizer, in the minimization of defined in Theorem 4(iv), i.e.,
However, unlike for which a simple form is available in Theorem 5 by solving a linear system, we are not able to obtain such a nice expression for . The following result (proved in Section 8.8) presents an analog of Theorems 6 and 7 for the new estimators, and .
Suppose (A)–(E) hold with . (i) If for some , then for any ,
where with . In addition, if , then for any with ,
(ii) If for some , then for any ,
with . (iii) If , then for any ,
with and .
Theorem 9 shows that if has infinite qualification, then smoothness of is fully captured in the rates and as , we attain rate for in contrast to (similar improved rates are also obtained for in various distances) in Theorem 6. In the following example, we present two choices of that improve on Tikhonov regularization. We refer the reader to Rosasco et al. (2005, Section 3.1) for more examples of .
(i) Tikhonov regularization involves for which it is easy to verify that and therefore the rates saturate at , leading to the results in Theorems 6 and 7. (ii) Showalter’s method and spectral cut-off use
respectively for which it is easy to verify that (see Engl, Hanke, and Neubauer, 1996, Examples 4.7 & 4.8 for details) and therefore improved rates are obtained for in Theorem 9 compared to that of Tikhonov regularization.
Density Estimation in 𝒫𝒫\mathcal{P}: Misspecified Case
In this section, we analyze the misspecified case where , which is a more reasonable case than the well-specified one, as in practice it is not easy to check whether . To this end, we consider the same estimator as considered in the well-specified case where is obtained from Theorem 5. The following result shows that as , and under the assumption that there exists such that . We present the result for bounded kernels although it can be easily extended to unbounded kernels as in Theorem B.2. Also, the presented result for Tikhonov regularization extends easily to using the ideas in the proof of Theorem 9. Note that unlike in the well-specified case where convergence in other distances can be shown even though the estimator is constructed from , it is difficult to show such a result in the misspecified case.
Let be probability densities such that where satisfies (A). Assume that (B), (C) and (D) with hold. Suppose , and there exists such that
Then for an estimator constructed from random samples drawn i.i.d. from , where is defined in (7)—also see Theorem 4(iv)—with , we have
In addition, if for some , then
with . If , then for ,
While the above result is useful and interesting, the assumption about the existence of is quite restrictive. This is because if (which is not in ) belongs to a family where is dense in w.r.t. , then there is no that attains the infimum, i.e., does not exist and therefore the proof technique employed in Theorem 10 will fail. In the following, we present a result (Theorem 12) that does not require the existence of but attains the same result as in Theorem 10, but requiring a more complicated proof. Before we present Theorem 12, we need to introduce some notation.
To this end, let us return to the objective function under consideration,
where and . Define
This is a reasonable class of functions to consider as under the condition , it is clear that . Endowed with a semi-norm,
is a vector space of functions, from which a normed space can be constructed as follows. Let us define to be equivalent, i.e., , if . In other words, if and only if and differ by a constant -almost everywhere. Now define the quotient space where denotes the equivalence class of . Defining , it is easy to verify that defines a norm on . In addition, endowing the following bilinear form on
makes it a pre-Hilbert space. Let be the Hilbert space obtained by completion of . As shown in Proposition 11 below, under some assumptions, a continuous mapping can be defined, which is injective modulo constant functions. Since addition of a constant does not contribute to , the space can be regarded as a parameter space extended from . In addition to , Proposition 11 (proved in Section 8.10) describes the adjoint of and relevant self-adjoint operators, which will be useful in analyzing in Theorem 12.
In addition, and are Hilbert-Schmidt and therefore compact. Also, and are compact, positive and self-adjoint operators on and respectively where
and the restriction of to is given by
Note that for , the derivatives do not depend on the choice of a representative element almost surely w.r.t. , and thus the above integrals are well defined. Having constructed , it is clear that , which means estimating is equivalent to estimating by . With all these preparations, we are now ready to present a result (proved in Section 8.11) on consistency and convergence rate for without assuming the existence of .
Let be probability densities such that . Assume that (A)–(D) hold with and . Then the following hold. (i) As (ii) Define . If , then
In addition, if for some , then for
. (iii) If and , then with .
Based on the observation (i) in the above remark that if is dense in w.r.t. , it is possible to obtain an approximation result for (similar to those discussed in Section 3) w.r.t. Fisher divergence as shown below, whose proof is provided in Section 8.12.
Numerical Simulations
We have proposed an estimator of that is obtained by minimizing the regularized empirical Fisher divergence and presented its consistency along with convergence rates. As discussed in Section 1, however one can simply ignore the structure of and estimate in a completely non-parametric fashion, for example using the kernel density estimator (KDE). In fact, consistency and convergence rates of KDE are also well-studied (Tsybakov, 2009, Chapter 1) and the kernel density estimator is very simple to compute—requiring only computations—compared to the proposed estimator, which is obtained by solving a linear system of size . This raises questions about the applicability of the proposed estimator in practice, though it is very well known that KDE performs poorly for moderate to large (Wasserman, 2006, Section 6.5). In this section, we numerically demonstrate that the proposed score matching estimator performs significantly better than the KDE, and in particular, that the advantage with the proposed estimator grows as gets large. Note further that the maximum likelihood approach of Barron and Sheu (1991) and Fukumizu (2009) does not yield estimators that are practically feasible, and therefore to the best of our knowledge, the proposed estimator is the only viable estimator for estimating densities through .
In the following, we consider two simple scenarios of estimating a multivariate normal and mixture of normals using the proposed estimator and demonstrate the superior performance of the proposed estimator over KDE. Inspired by this preliminary empirical investigation, recently, the proposed estimator has been explored in two concrete applications of gradient-free adaptive MCMC sampler (Strathmann et al., 2015) and graphical model structure learning (Sun et al., 2015) where the superiority of working with the infinite dimensional family is demonstrated. We would like to again highlight that the goal of this work is not to construct density estimators that improve upon KDE but to provide a novel modeling technique of approximating an unknown density by a rich parametric family of densities with the parameter being infinite dimensional in contrast to the classical approach of finite dimensional approximation.
through the score matching approach and KDE, and compare their estimation accuracies. Here is the p.d.f. of . By choosing the kernel, which is a Gaussian plus polynomial of degree 2, it is easy to verify that Gaussian distributions lie in , and therefore the first problem considers the well-specified case while the second problem deals with the misspecified case. In our simulations, we chose , , and . The base measure of the exponential family is . The bandwidth parameter is chosen by cross-validation (CV) of the objective function (see Theorem 4(iv)) within the parameter set , where is the median of pairwise distances of data, and the regularization parameter is set as with sample size . For KDE, the Gaussian kernel is used for the smoothing kernel, and the bandwidth parameter is chosen by CV from ; where for both the methods, 5-fold CV is applied.
Since it is difficult to accurately estimate the normalization constant in the proposed method, we use two methods to evaluate the accuracy of estimation. One is the objective function for the score matching method,
and the other is correlation of the estimator with the true density function,
where is a probability distribution. For , we use the empirical distribution based on 10000 random samples drawn i.i.d. from .
Summary & Discussion
We have considered an infinite dimensional generalization, , of the finite-dimensional exponential family, where the densities are indexed by functions in a reproducing kernel Hilbert space (RKHS), . We showed that is a rich object that can approximate a large class of probability densities arbitrarily well in Kullback-Leibler divergence, and addressed the main question of estimating an unknown density, from finite samples drawn i.i.d. from it, in well-specified () and misspecified () settings. We proposed a density estimator based on minimizing the regularized version of the empirical Fisher divergence, which results in solving a simple finite-dimensional linear system. Our estimator provides a computationally efficient alternative to maximum likelihood based estimators, which suffer from the computational intractability of the log-partition function. The proposed estimator is also shown to empirically outperform the classical kernel density estimator, with advantage increasing as the dimension of the space increases. In addition to these computational and empirical results, we have established the consistency and convergence rates under certain smoothness assumptions (e.g., ) for both well-specified and misspecified scenarios.
Proofs
We provide proofs of the results presented in Sections 3–5.
Sriperumbudur et al. (2011, Proposition 5) showed that is dense in w.r.t. uniform norm if and only if satisfies (5). Therefore, the denseness in , KL and Hellinger distances follow trivially from Lemma A.1. For norm (), the denseness follows by using the bound obtained from Lemma A.1(i) with and .
2 Proof of Corollary 2
For any , define . Note that for all and , implying that for any . This means, for any , such that for any , we have , where for all .
We now prove the denseness in KL divergence by noting that
which implies . This implies, for any , such that for any , . Arguing as above, we have , i.e., there exists such that . Since is dense in , for any and any , there exists such that . For , since we have
3 Proof of Theorem 4
(i) By the reproducing property of , since for all , it is easy to verify that
where in the second line, we used for with being a Hilbert space and
Observe that for all , is a Hilbert-Schmidt operator as and is also Hilbert-Schmidt as . Therefore, (11) is equivalent to
Since the first condition in (D) implies , is -integrable in the Bochner sense (see Diestel and Uhl, 1977, Definition 1 and Theorem 2), and therefore it follows from Diestel and Uhl (1977, Theorem 6) that
where is the Bochner integral of , thereby yielding (6).
which means is trace-class and therefore compact. Here, we used monotone convergence theorem in and Parseval’s identity in . Note that is positive since (ii) From (6), we have . Using for all , we obtain that for any ,
where follows from integration by parts under (C) and the equality in is valid as is Bochner -integrable under (D) with . Therefore . For the third term, and the result follows. (iii) Define . For any , it is easy to verify that
Clearly, is minimized if and only if and therefore is the unique minimizer of . (iv) Since is similar to with replaced by and replaced by , we obtain .
4 Proof of Theorem 5
We prove the result based on the general representer theorem (Theorem A.2). From Theorem 4(iv), we have
where , and . Therefore, it follows from Theorem A.2 that
with Since , (15) reduces to and yielding and .
Instead of using the general representer theorem (Theorem A.2), it is possible to see that the standard representer theorem (Kimeldorf and Wahba, 1971; Schölkopf et al., 2001) gives a similar, but slightly different linear system, and the solutions are the same if is non-singular. The general representer theorem yields that and are solution to , where . On the other hand, by using the standard representer theorem, it is easy to show that has the form in (14) with and being solution to . Clearly, both the solutions match if is invertible while the latter has many solutions if is not invertible.
5 Proof of Theorem 6
where we used in (). Define , and so that
where . We now bound , and using Proposition A.4. Note that where is defined in (12) is a positive, self-adjoint, trace-class operator and (D) (with ) implies that
Using the bounds in , and in (16), we obtain
(i) By Proposition A.3(i), we have that as if . Therefore, it follows from (20) that as , and . (ii) If for , it follows from Proposition A.3(ii) that
and therefore the result follows by choosing . (iii) Note that
It follows from Proposition A.4(v) that for where is a sufficiently large constant that depends on but not on and . Using the bounds on , and from part (i) and the bound on from Proposition A.3(ii), we therefore obtain
as and the result follows.
Under slightly strong assumptions on the kernel, the bound on in (17) can be improved to obtain while the one on in (19) can be refined to obtain where is the intrinsic dimension of . Using the fact that , it is easy to verify that the latter is an improved bound than the one in (19). In addition dominates . However, if in (18) is not improved, then dominates , thereby resulting in a bound that does not capture the smoothness of (or the corresponding ). Unfortunately, even with a refined analysis (not reported here), we are not able to improve the bound on wherein the difficulty lies with handling .
6 Proof of Theorem 7
Before we prove the result, we present a lemma.
Since , it implies that, for every , and hence . Also, under the assumptions on and , it is easy to verify that , which implies
In the following, we obtain a bound on . While one can trivially use the bound to obtain a rate on through the result in Theorem 6(ii), a better rate can be obtained by carefully bounding as shown below. Consider
7 Proof of Proposition 8
Observation 1: By (Wendland, 2005, Theorem 10.12), we have
where is defined in sense. Since
Observation 3: For , we have for all ,
Observation 4: For any , we have
which implies , i.e., .
We now use these observations to prove the result. Since , there exists such that , which means
where we have invoked generalized Young’s inequality (Folland, 1999, Proposition 8.9) in (), Hausdorff-Young inequality (Folland, 1999, p. 253) in (), and observation 3 combined with in (). This shows that , i.e., .
8 Proof of Theorem 9
To prove Theorem 9, we need the following lemma (De Vito et al., 2012, Lemma 5), which is due to Andreas Maurer.
Proof of Theorem 9. (i) The proof follows the ideas in the proof of Theorem 10 in Bauer et al. (2007), which is a more general result dealing with the smoothness condition, where is operator monotone. Recall that is operator monotone on if for any pair of self-adjoint operators , with spectra in such that , we have , where “” is the partial ordering for self-adjoint operators on some Hilbert space , which means for any , . In our case, we adapt the proof for . Define . Since , there exists such that , which yields
We now bound –. Since , we have where we used in (E) and the bound on from the proof of Theorem 6(i). Similarly, implies where in (E) and Proposition A.4(i) are invoked. Also, in (E) implies that
We now consider two cases: : Since is operator monotone on for , by Theorem 1 in Bauer et al. (2007), there exists a constant such that . We now obtain a bound on . To this end, consider
which by Chebyshev’s inequality implies that
and therefore . Since , we have . : Since is Lipschitz on for , by Lemma 15, and therefore .
Collecting all the above bounds, we obtain
and the result follows. The proofs of the claims involving , and follow exactly the same ideas as in the proof of Theorem 7 by using the above bound on in Lemma A.1. (ii) We now bound as follows. Note that
We bound as
where we used the fact that is operator monotone along with . Using (26), can be bounded as
with and bounded as in part (i) above. Here . Combining and , we obtain the required result. (iii) The proof follows the ideas in the proof of Theorems 6 and 7. Consider so that
Therefore where we used the fact that and the result follows.
9 Proof of Theorem 10
Before we analyze , we need a small calculation for notational convenience. For any probability densities , it is clear that . We generalize this by defining
Clearly, if , then matches with . Therefore, for probability densities ,
where . The result simply follows from the proof of Theorem 7, where we showed that and if for as , . When , we bound in (8.9) as where is in turn bounded as in (21).
10 Proof of Proposition 11
which means and therefore . Since where is some constant, it is clear that is a continuous map from to . The adjoint of is defined by the relation . If , then
For and , this yields
We now show that is Hilbert-Schmidt. Since is separable, let be an ONB of . Then we have
which proves that is Hilbert-Schmidt (hence compact) and therefore is also Hilbert-Schmidt and compact. The other assertions about and are straightforward.
11 Proof of Theorem 12
By slight abuse of notation, is used to denote in the proof for simplicity. For , we have
Since satisfies (C) it is easy to verify that (see proof of Theorem 4(ii)). This implies and
where is defined in Theorem 4(ii), and is precisely the operator defined in Theorem 4(ii). Following the proof of Theorem 4(ii), for , it is easy to show that the unique minimizer of the regularized objective, exists and is given by
We would like to reiterate that (29) and (30) also match with their counterparts in Theorem 4 and therefore as in Theorem 4(iv), an estimator of is given by . In other words, this is the same as in Theorem 4(iv) since , and can be solved by a simple linear system provided in Theorem 5. Here is the empirical estimator of . Now consider
where . The proof now proceeds using the following decomposition, equivalent to the one used in the proof of Theorem 6(i), i.e.,
where we used (30) in . is well-defined as it is the empirical version of the restriction of to . Since and , we have
for where is a sufficiently large constant that does not depend on and . Following the proof of Proposition A.4(i), we have
wherein the first term is zero as and since
the integral in the second term is finite because of (D) and . Therefore, an application of Chebyshev’s inequality yields
We now show that . To this end, define and consider
which therefore yields the claim through an application of Chebyshev’s inequality. Using this along with (33) and (34) in (32), and using the resulting bound in (31) yields
(i) We bound as follows. First note that
and so for any , we have
From and , we have
where the inequality follows from Proposition A.3(ii). Using (37) and (38) in (36), we obtain , using which in (35) yields
Since the above inequality holds for any , we therefore have
Since , we have that as , and . (ii) Recall from (i). From Proposition A.3(i) it follows that as if . Therefore, (35) reduces to and the consistency result follows. If for some , then the rates follow from Proposition A.3 by noting that and choosing . (iii) This simply follows from an analysis similar to the one used in the proof of Theorem 6(iii).
12 Proof of Proposition 13
For any , define , which implies that . Since is dense in , we have for any , there exists such that . For a given , pick . Therefore,
Appendix A Appendix: Technical Results
In this appendix, we present some technical results that are used in the proofs.
In the following result, claims (iii) and (iv) are quoted from Lemma 3.1 of van der Vaart and van Zanten (2008).
for any ;
;
where is a universal constant;
(i) Define . Consider
Since and , from (A.2) we obtain
where we used for in the last line above. (ii) This simply follows from (A.2) by using .
A.2 General Representer Theorem
The following is the general representer theorem for abstract Hilbert spaces.
and thus . Therefore , and so for every , and hence .
The following result is quite well-known in the linear inverse problem theory (Engl et al., 1996).
Let be a bounded, self-adjoint compact operator on a separable Hilbert space . For and , define and for . Then the following hold.
For any , as and if , then as .
If for and , then
by the dominated convergence theorem. For any , we have
Let where , if and , if . Then
as . (ii) If , then there exists such that . This yields
On the other hand, for , we have
Using the above in (A.3) yields the result.
A.4 Bound on the Norm of Certain Operators and Functions
The following result is used in many places throughout the paper. We would like to highlight that special cases of this result are known, e.g., see the proof of Theorem 4 in Caponnetto and Vito (2007) where concentration inequalites are obtained for the quantities in Proposition A.4 using Bernstein’s inequality. Here, we provide asymptotic statements using Chebyshev’s inequality.
.
.
where the last inequality follows from (iii). Using (A.5) in (A.4), we obtain
The result therefore follows by an application of Chebyshev’s inequality. (v) We use the idea in Step 2.1 of the proof of Theorem 4 in Caponnetto and Vito (2007), where is written equivalently as follows: Note that , which implies
and so . Using the Von Neumann series representation, we have
A.5 Interpolation Space
In this section, we briefly recall the definition of interpolation spaces of the real method. To this end, let and be two arbitrary Banach spaces that are continuously embedded in some topological (Hausdorff) vector space . Then, for and , the -functional of the real interpolation method (see Bennett and Sharpley, 1988, Definition 1.1, p. 293) is defined by
Suppose and are two Banach spaces that satisfy (i.e., and the inclusion operator is continuous), then the -functional reduces to
The -functional can be used to define interpolation norms, for , and , as
Moreover, the corresponding interpolation spaces (Bennett and Sharpley, 1988, Definition 1.7, p. 299) are defined as
Appendix B Appendix: Miscellaneous Results
In this appendix, we present the proofs of some claims that we made in Sections 1, 4 and 5.
The following result provides a relationship between Fisher and Kullback-Leibler divergences.
as for all where . Then
Under the conditions mentioned on and , it can be shown that
See Theorem 1 in Lyu (2009) for a proof. The above identity is a simple generalization of de Bruijn’s identity that relates the Fisher information to the derivative of the Shannon entropy (see Cover and Thomas, 1991, Theorem 16.6.2). Integrating w.r.t. on both sides of (B.2), we obtain KL(p_{t}\|q_{t})\Big{|}^{\infty}_{t=0}=-\int^{\infty}_{0}J(p_{t}\|q_{t})\,dt which yields the equality in (B.1) as as and as .
To handle the case of unbounded , in the following, we assume that there exists a positive constant such that , so that an estimator of can be constructed as
where is defined in Theorem 4(iv). This modification yields a valid estimator as long as satisfies since this implies . The construction of requires the knowledge of , however, which we assume is known a priori. Using the representer theorem in RKHS, it can be shown (see Section B.2.1) that
where and are obtained by solving the following quadratically constrained quadratic program (QCQP),
with , and , being defined in the proof of Theorem 5 and the remark following it. The following result investigates the consistency and convergence rates for .
Let be a fixed constant, and be a clipped estimator given by (B.3). Suppose (A)–(D) with hold. Let and . Define . Then, as ,
, if ;
for , if and ;
if ;
.
In addition, if for some , then where with assuming the respective conditions in (i)-(iii) above hold.
For any , since and , we have
where we used the fact that for and . In the following, we obtain bounds for \bigl{\|}p_{\breve{f}_{\lambda,n}}-p_{0}\bigr{\|}_{L^{r}(\Omega)} for any , and in terms of \|\breve{f}_{\lambda,n}-f_{0}\bigr{\|}_{\mathcal{H}}. Define . Since satisfies , then it is clear that as since
Similarly, it is easy to verify that . (i) Recalling (A.1), we have
Using (B.4), we bound as
Also for any with , we have , where . Again using (B.4), we have
and . Therefore,
where the above inequality is obtained by carrying out and simplifying the decomposition as in (A.1). Using (B.4), we therefore have
(iv) As , by Theorem 4, we obtain Note that we have bounded the various distances between and in terms of . Since with probability converging to 1, the assertions on consistency are proved by Theorem 6(i) in combination with Lemma 14—as we did not explicitly assume —and the rates follow from Theorem 6(iii).
The following observations can be made while comparing the scenarios of using bounded vs. unbounded kernels in the problem of estimating through Theorems 7 and B.2. First, the consistency results in , Hellinger and KL distances are the same but for additional integrability conditions on and . The additional integrability conditions are not too difficult to hold in practice as they involve and which can be chosen appropriately. However, the unbounded situation in Theorem B.2 requires the knowledge of which is usually not known. On the other hand, the consistency result in in Theorem B.2 is slightly weaker than in Theorem 7. This may be an artifact of our analysis as we are not able to adapt the bounding technique used in the proof of Theorem 7 to bound as it critically depends on the boundedness of . Therefore, we used a trivial bound of , which yields the result through Theorem 6(i). Due to the same reason, we also obtain a slower rate of convergence in . Second, the rate of convergence in KL is slower than in Theorem B.2, which again may be an artifact of our analysis. The convergence rate for KL in Theorem 7 is based on the application of Theorem 6(ii) in Lemma A.1, where the bound on KL in Lemma A.1 critically uses the boundedness to upper bound KL in terms of squared Hellinger distance.
Any can be decomposed as where
which is a closed subset of and so that . Since the objective function in (B.3) matches with the one in Theorem 5, using the above decomposition in (B.3), it is easy to verify that depends only on so that (B.3) reduces to
and . Since is of the form in (14), using it in (B.5), it is easy to show that . Similarly, it can be shown that . Since appears in (B.5) only through , (B.5) reduces to
where is constructed as in (14) using and is such that . The necessary and sufficient conditions for the optimality of is given by the following Karush-Kuhn-Tucker conditions,
Combining the dual feasibility and stationary conditions, we have , i.e., . Using this in the complementary slackness involving and , it follows that . Since , we have , i.e., is completely determined by . Therefore is of the form in (14) and (B.6) reduces to a quadratically constrained quadratic program.
where , and the spaces and have equivalent norms.
To prove Proposition B.3, we need the following result which we quote from Steinwart and Scovel (2012, Lemma 6.3) (also see Tartar, 2007, Lemma 23.1) that interpolates -spaces whose underlying measures are absolutely continuous with respect to a measure .
Let be a measure on a measurable space and and be measurable functions. For , define . Then we have
and the norms on these two spaces are equivalent. Moreover, this result still holds for weights and , if one uses the convention in the definition of the weighted spaces.
Proof of Proposition B.3. The proof is based on the ideas used in the proof of Theorem 4.6 in Steinwart and Scovel (2012). Recall that by the Hilbert-Schmidt theorem, has the following representation,
where if and if with . Let . By definition, is equivalent to such that , i.e.,
where . Let us equip this space with the bilinear form
It is easy to verify that is an ONB of . Also since for and is continuous, i.e., for any ,
and so . Similarly, we can show that . In the following, we first prove the result for and then for .
(a) : For any and , we have
where we define for . For , we find
From this we immediately obtain the equivalence
where . Applying the second part of Lemma B.4 to the counting measure on yields
from which we obtain the following equivalence
For , this implies a.e. and so .
Acknowledgments
Theorem A.2 and the proof are due to an anonymous reviewer. We thank Dougal Sutherland for a careful reading of the paper which helped to fix minor errors. A part of the work was carried out while BKS was a Research Fellow in the Statistical Laboratory, Department of Pure Mathematics and Mathematical Statistics, University of Cambridge. BKS thanks Richard Nickl for many valuable comments and insightful discussions. KF is supported in part by MEXT Grant-in-Aid for Scientific Research on Innovative Areas 25120012.