Projected normal distribution

In directional statistics, the projected normal distribution (also known as offset normal distribution, angular normal distribution or angular Gaussian distribution)[1][2] is a probability distribution over directions that describes the radial projection of a random variable with n-variate normal distribution over the unit (n-1)-sphere.

Projected normal distribution
Notation
Parameters (___location)
(scale)
Support

Unit n-sphere, with angular or Cartesian coordinates:

PDF complicated, see text

Definition and properties

edit

Given a random variable   that follows a multivariate normal distribution  , the projected normal distribution   represents the distribution of the random variable   obtained projecting   over the unit sphere. In the general case, the projected normal distribution can be asymmetric and multimodal. In case   is parallel to an eigenvector of  , the distribution is symmetric.[3] The first version of such distribution was introduced in Pukkila and Rao (1988).[4]

Support

edit

The support of this distribution is the unit (n-1)-sphere, which can be variously given in terms of a set of  -dimensional angular spherical cooordinates:

 

or in terms of  -dimensional Cartesian coordinates:

 

The two are linked via the embedding function,  , with range   This function is defined by the formula for spherical coordinates at  

Density function

edit

The density of the projected normal distribution   can be constructed from the density of its generator n-variate normal distribution   by re-parametrising to n-dimensional spherical coordinates and then integrating over the radial coordinate.

In full spherical coordinates with radial component   and angles  , a point   can be written as  , with  . To be clear,  , as given by the above-defined embedding function. The joint density becomes

 

where the factor   is due to the change of variables  . The density of   can then be obtained via marginalization over   as[5]

 

The same density had been previously obtained in Pukkila and Rao (1988, Eq. (2.4))[4] using a different notation.

Note on density definition

edit

This subsection gives some clarification lest the various forms of probability density used in this article be misunderstood. Take for example a random variate  , with uniform density,  . If  , it has density,  . This works if both densities are defined with respect to Lebesgue measure on the real line. By default convention:

  • Density functions are Lebesgue-densities, defined with respect to Lebesgue measure, applied in the space where the argument of the density function lives, so that:
  • The Lebesgue-densities involved in a change of variables are related by a factor dependent on the derivative(s) of the transformation (  in this example; and   for the above change of variables,  ).

Neither of these conventions apply to the   densities in this article:

  • For   the density,   is not defined w.r.t. Lebesgue measure in   where   lives, because that measure does not agree with the standard notion of hyperspherical area. Instead, the density is defined w.r.t. a measure that is pulled back (via the embedding function) to angular coordinate space, from Lebesgue measure in the  -dimensional tangent space of the hypersphere. This will be explained below.
  • With the embedding  , a density,   cannot be defined w.r.t. Lebesgue measure, because   has Lebesgue measure zero. Instead,   is defined w.r.t. scaled Hausdorff measure.

The pullback and Hausdorff measures agree, so that:

 

where there is no change-of-variables factor, because the densities use different measures.

To better understand what is meant by a density being defined w.r.t. a measure (a function that maps subsets in sample space to a non-negative real-valued 'volume'), consider a measureable subset,  , with embedded image   and let  , then the probability for finding the sample in the subset is:

 

where   are respectively the pullback and Hausdorff measures; and the integrals are Lebesgue integrals, which can be rewritten as Riemann integrals thus:

 

Pullback measure

edit

The tangent space at   is the  -dimensional linear subspace perpendicular to  , where Lebesgue measure can be used. At very small scale, the tangent space is indistinguishable from the sphere (e.g. Earth looks locally flat), so that Lebesgue measure in tangent space agrees with area on the hypersphere. The tangent space Lebesgue measure is pulled back via the embedding function, as follows, to define the measure in coordinate space. For   a measureable subset in coordinate space, the pullback measure, as a Riemann integral is:

 

where the Jacobian of the embedding function,  , is the   matrix   the columns of which span the  -dimensional tangent space where the Lebesgue measure is applied. It can be shown:   When plugging the pullback measure (2), into equation (1) and exchanging the order of integration:[6]

 

where the first integral is Lebesgue and the second Riemann. Finally, for better geometric understanding of the square-root factor, consider:

  • For  , when integrating over the unitcircle, w.r.t.  , with embedding  , the Jacobian is  , so that  . The angular differential,   directly gives the subtended arc length on the circle.
  • For  , when integrating over the unitsphere, w.r.t.  , we get  , which is the radius of the circle of latitude at   (compare equator to polar circle). The area of the surface patch subtended by the two angular differentials is:  .
  • More generally, for  , let   be a square or tall matrix and let   denote the parallelotope spanned by its colums (which represent the edges meeting at a common vertex). The parallelotope volume is   the square root of the absolute value of the Gram determinant. For square  , the volume simplifies to   Now let  , so that   is a rectangle with infinitessimally small volume,  . Since the smooth embedding function is linear at small scale, the embedded image is the paralleotope,  , with volume (area of the subtended hyperspherical surface patch): 

Circular distribution

edit

For  , parametrising the position on the unit circle in polar coordinates as  , the density function can be written with respect to the parameters   and   of the initial normal distribution as

 

where   and   are the density and cumulative distribution of a standard normal distribution,  , and   is the indicator function.[3]

In the circular case, if the mean vector   is parallel to the eigenvector associated to the largest eigenvalue of the covariance, the distribution is symmetric and has a mode at   and either a mode or an antimode at  , where   is the polar angle of  . If the mean is parallel to the eigenvector associated to the smallest eigenvalue instead, the distribution is also symmetric but has either a mode or an antimode at   and an antimode at  .[7]

Spherical distribution

edit

For  , parametrising the position on the unit sphere in spherical coordinates as   where   are the azimuth   and inclination   angles respectively, the density function becomes

 

where  ,  ,  , and   have the same meaning as the circular case.[8]

Angular Central Gaussian Distribution

edit

In the special case,  , the projected normal distribution, with   is known as the angular central Gaussian (ACG)[9] and in this case, the density function can be obtained in closed form as a function of Cartesian coordinates. Let   and project radially:   so that   (the unit hypersphere). We write  , which as explained above, at  , has density:

 

where the integral can be solved by a change of variables and then using the standard definition of the gamma function. Notice that:

  • For any   there is the parameter indeterminacy:
 .
  • If  , the uniform hypershpere distribution,   results, with constant density equal to the reciprocal of the surface area of  :
 

ACG via transformation of normal or uniform variates

edit

Let   be any  -by-  invertible matrix such that  . Let   (uniform) and   (chi distribution), so that:   (multivariate normal). Now consider:

 

which shows that the ACG distribution also results from applying, to uniform variates, the normalized linear transform:[9]

 

Some further explanation of these two ways to obtain   may be helpful:

  • If we start with  , sampled from a multivariate normal, we can project radially onto   to obtain ACG variates. To derive the ACG density, we first do a change of variables:  , which is still an  -dimensional representation, and this transformation induces the differential volume change factor,  , which is proportional to volume in the  -dimensional tangent space perpendicular to  . Then, to finally obtain the ACG density on the  -dimensional unitsphere, we need to marginalize over  .
  • If we start with  , sampled from the uniform distribution, we do not need to marginalize, because we are already in   dimensions. Instead, to obtain ACG variates (and the associated density), we can directly do the change of variables,  , for which further details are given in the next subsection.

Caveat: when   is nonzero, although  , a similar duality does not hold:

 

Although we can radially project affine-transformed normal variates to get   variates, this does not work for uniform variates.

Wider application of the normalized linear transform

edit

The normalized linear transform,  , is a bijection from the unitsphere to itself; the inverse is  . This transform is of independent interest, as it may be applied as a probabilistic flow on the hypersphere (similar to a normalizing flow) to generalize also other (non-uniform) distributions on hyperspheres, for example the Von Mises-Fisher distribution. The fact that we have a closed form for the ACG density allows us to recover also in closed form the differential volume change induced by this transform.

For the change of variables,   on the manifold,  , the uniform and ACG densities are related as:[6]

 

where the (constant) uniform density is   and where   is the differential volume change factor from the input to the output of the transformation; specifically, it is given by the absolute value of the determinant of an  -by-  matrix:

 

where   is the  -by-  Jacobian matrix of the transformation in Euclidean space,  , evaluated at  . In Euclidean space, the transformation and its Jacobian are non-invertible, but when the ___domain and co-___domain are restricted to  , then   is a bijection and the induced differential volume ratio,   is obtained by projecting   onto the  -dimensional tangent spaces at the transformation input and output:   are  -by-  matrices whose orthonormal columns span the tangent spaces. Although the above determinant formula is relatively easy to evaluate numerically on a software platform equipped with linear algebra and automatic differentiation, a simple closed form is hard to derive directly. However, since we already have  , we can recover:

 

where in the final RHS it is understood that   and  .

The normalized linear transform can now be used, for example, to give a closed-form density for a more flexible distribution on the hypersphere, that is generalized from the Von Mises-Fisher. Let   and  ; the resulting density is:

 

See also

edit

References

edit

Sources

edit
  • Pukkila, Tarmo M.; Rao, C. Radhakrishna (1988). "Pattern recognition based on scale invariant discriminant functions". Information Sciences. 45 (3): 379–389. doi:10.1016/0020-0255(88)90012-6.
  • Hernandez-Stumpfhauser, Daniel; Breidt, F. Jay; van der Woerd, Mark J. (2017). "The General Projected Normal Distribution of Arbitrary Dimension: Modeling and Bayesian Inference". Bayesian Analysis. 12 (1): 113–133. doi:10.1214/15-BA989.
  • Wang, Fangpo; Gelfand, Alan E (2013). "Directional data analysis under the general projected normal distribution". Statistical Methodology. 10 (1). Elsevier: 113–127. doi:10.1016/j.stamet.2012.07.005. PMC 3773532. PMID 24046539.
  • Tyler, David E (1987). "Statistical analysis for the angular central Gaussian distribution on the sphere". Biometrika. 74 (3): 579–589. doi:10.2307/2336697. JSTOR 2336697.
  • Sorrenson, Peter; Draxler, Felix; Rousselot, Armand; Hummerich, Sander; Köthe, Ullrich (2024). "Learning Distributions on Manifolds with Free-Form Flows". arXiv:2312.09852 [cs.LG].