User:TokekarSandbox/HKS

Heat Kernel Signature (HKS) is a feature descriptor for use in deformable shape analysis. For each point in the shape, HKS defines its feature vector representing the point's local and global geometric properties. Applications include segmentation, classification, structure discovery, shape matching and shape retrieval.

HKS was introduced in 2009 by Jian Sun, Maks Ovsjanikov and Leonidas Guibas. It is based on heat kernel, which is a fundamental solution to the heat equation. HKS is one of the many recently introduced shape descriptors which are based on the Laplace-Beltrami operator associated with the shape.

Overview
Shape analysis is the field of automatic digital analysis of shapes, e.g., 3D objects. For many shape analysis tasks (such as shape matching/retrieval), feature vectors for certain key points are used instead of using the complete 3D model of the shape. An important requirement of such feature descriptors is for them to be invariant under certain transformations. For rigid transformations, commonly used feature descriptors include shape context, spin images, integral volume descriptors and multiscale local features, among others. HKS allows isometric transformations which generalizes rigid transformations.



HKS is based on the concept of heat diffusion over a surface. Given an initial heat distribution $$ u_0(x)$$ over the surface, the heat kernel $$ h_t(x,y) $$ relates the amount of heat transferred from $$x$$ to $$y$$ after time $$t$$. The heat kernel is invariant under isometric transformations and stable under small perturbations to the isometry. In addition, the heat kernel fully characterizes shapes up to an isometry and represents increasingly global properties of the shape with increasing time. Since $$h_t(x,y)$$ is defined for a pair of points over a temporal domain, using heat kernels directly as features would lead to a high complexity. HKS instead restricts itself to just the temporal domain by considering only $$h_t(x,x)$$. HKS inherits most of the properties of heat kernels under certain conditions.

Technical Details
The heat diffusion equation over a compact Riemannian manifold $$M$$ (possibly with a boundary) is given by,

\left(\Delta + \frac{\partial}{\partial t} \right) u(x,t) = 0 $$ where $$\Delta$$ is the Laplace-Beltrami operator and $$u(x,t)$$ is the heat distribution at a point $$x$$ at time $$t$$. The solution to this equation can be expressed as ,

u(x,t) = \int h_t(x,y)u_0(y)dy. $$ The eigen decomposition of the heat kernel is expressed as,

h_t(x,y) = \sum_{i=0}^{\infty} \exp(-\lambda_i t) \phi_i(x) \phi_i(y) $$ where $$\lambda_i$$ and $$\phi_i$$ are the $$i^{th}$$ eigenvalue and eigenfunction of $$\Delta$$. The heat kernel fully characterizes a surface up to an isometry: For any surjective map $$T:M\rightarrow N$$ between two Riemannian manifolds $$M$$ and $$N$$, if $$h_t(x,y) = h_t(T(x),T(y))$$ then $$T$$ is an isometry, and vice-versa. For a concise feature descriptor, HKS restricts the heat kernel only to the temporal domain,

h_t(x,x) = \sum_{i=0}^{\infty} \exp(-\lambda_i t) \phi_i^2(x). $$ HKS, similar to the heat kernel, characterizes surfaces under the condition that the eigenvalues of $$\Delta$$ for $$M$$ and $$N$$ are non-repeating. The terms $$\exp(-\lambda_i t)$$ can be intuited as a bank of low-pass filters, with $$\lambda_i$$ determining the cutoff frequencies.

Practical Considerations
Since $$h_t(x,x)$$ is, in general, a non-parametric continuous function, HKS is in practice represented as a discrete sequence of $$\{h_{t_1}(x,x),\ldots,h_{t_n}(x,x)\}$$ values sampled at times $$t_1,\ldots,t_n$$.

In most applications, the underlying manifold for an object is not known. The HKS can be computed if a mesh representation of the manifold is available, by using a discrete approximation to $$\Delta$$ and using the discrete analogue of the heat equation. In the discrete case, the Laplace-Beltrami operator is a sparse matrix and can be written as ,

L = A^{-1}W $$ where $$A$$ is a positive diagonal matrix with entries $$A(i,i)$$ corresponding to the area of the triangles in the mesh sharing the vertex $$i$$, and $$W$$ is a symmetric semi-definite weighting matrix. $$L$$ is symmetric and can be decomposed into $$L = \Phi \Lambda \Phi^T A$$, where $$\Lambda$$ is a diagonal matrix of the eigenvalues of $$L$$ arranged in the ascending order, and $$\Phi$$ is the matrix with the corresponding orthonormal eigenvectors. The discrete heat kernel is the matrix given by,

K_t = \Phi \exp(-t\Lambda) \Phi^T. $$ The elements $$k_t(i,j)$$ represents the heat diffusion between vertices $$i$$ and $$j$$ after time $$t$$. The HKS is then given by the diagonal entries of this matrix, sampled at discrete time intervals. Similar to the continuous case, the discrete HKS is robust to noise.

Non-repeating Eigenvalues
The main property that characterizes surfaces using HKS up to an isometry holds only when the eigenvalues of the surfaces are non-repeating. There are certain surfaces (especially those with symmetry) where this condition is violated. A sphere is a simple example of such a surface.

Poor feature localization
The HKS can be intuited as a collection of low pass filters suppressing high frequency components. Hence, the HKS suffers from poor feature localization as points close on the manifold have similar feature descriptors. The feature descriptors carry more global information which is useful for shape discrimination, but poor for feature localization.

Time parameter selection
The time parameter in the HKS is closely related to the scale of global information. However, there is no direct way to choose the time discretization. The existing method chooses time samples logarithmically which is a heuristic with no guarantees

Time complexity
The discrete heat kernel requires eigendecomposition of a matrix of size $$n \times n$$, where $$n$$ is the number of vertices in the mesh representation of the manifold. Computing the eigendecomposition is an expensive operation, especially as $$n$$ increases.

Non-isometric transformations
The performance guarantees for HKS only hold for truly isometric transformations. However, deformations for real shapes are often not isometric. A simple example of such transformation is closing of the fist by a person, where the geodesic distances between two fingers changes.

Curvature
The (continuous) HKS at a point $$x$$, $$h_t(x,x)$$ on the Riemannian manifold is related to the scalar curvature $$s(x)$$ by,

h_t(x,x) = \frac{1}{4\pi t} + \frac{s(x)}{12\pi} + O(t). $$ Hence, HKS can as be interpreted as the curvature of $$x$$ at scale $$t$$.

Wave Kernel Signature (WKS)
The WKS follows a similar idea to the WKS, replacing the heat equation with the Schrödinger wave equation,

\left( i\Delta + \frac{\partial}{\partial t} \right) \psi(x,t) = 0 $$ where $$\psi(x,t)$$ is the complex wave function. The average probability of measuring the particle at a point $$x$$ is given by,

p(x) = \sum_{i=0}^{\infty} f^2(\lambda_i) \phi_i^2(x) $$ where $$f$$ is the initial energy distribution. By fixing a family of this energy distributions $$f_i(x)$$, the WKS can be obtained as a discrete sequence $$\{ p_{f_1}(x),\ldots,p_{f_n}(x)\}$$. Unlike HKS, the WKS can be intuited as a set of band-pass filters leading to better feature localization. However, the WKS does not represent large-scale features well (as they are ``filtered'' out) yielding poor performance at shape matching applications.

Global Point Signature (GPS)
Similar to the HKS, the GPS is based on the Laplace-Beltrami operator. GPS at a point $$x$$ is a vector of scaled eigenfunctions of the Laplace-Beltrami operator compute at $$x$$. The GPS is a global feature whereas the scale of the HKS can be varied by varying the time parameter for heat diffusion. Hence, the HKS can be used in partial shape matching applications whereas the GPS cannot.

Scale Invariance
Even though the HKS represents the shape at multiple scales, it is not inherently scale invariant. For example, the HKS for a shape and its scaled version are not the same without pre-normalization. A scale-invariant version of the HKS can be constructed by generating a scale-space representation. In the scale-space, the HKS of a scaled shape corresponds to a translation up to a multiplicative factor. The fourier transform of this HKS changes the time-translation into the complex plane, and the dependency on translation can be eliminated by considering the modulus of the transform.

Volumetric HKS
The HKS is defined for a boundary surface of a 3D shape, represented as a 2D Riemannian manifold. Instead of considering only the boundary, the entire volume of the 3D shape can be considered to define the volumetric version of the HKS. The Volumetric HKS is defined analogous to the normal HKS by considering the heat equation over the entire volume (as a 3-submanifold) and defining a Neumann boundary condition over the 2-manifold boundary of the shape. Volumetric HKS characterizes transformations up to a volume isometry, which represent the transformation for real 3D objects more faithfully than boundary isometry.

Shape Google
The scale-invariant HKS features can be used in the Bag-of-features model for shape retrieval applications. The features are used to construct ``geometric words'' by taking into account their spatial relations, from which shapes can be constructed (analogous to using features as words and shapes as sentences). Shapes themselves are represented using compact binary codes to form an indexed collection. Given a query shape, similar shapes in the index with possibly isometric transformations can be retrieved (hence the term Shape Google) by using the Hamming distance of the code as the nearness-measure.