Diffusion maps is a dimensionality reduction or feature extraction algorithm introduced by R. R. Coifman and S. Lafon. It computes a family of embeddings of a data set into Euclidean space (often low-dimensional) whose coordinates can be computed from the eigenvectors and eigenvalues of a diffusion operator on the data. The Euclidean distance between points in the embedded space is equal to the "diffusion distance" between probability distributions centered at those points. Different from linear dimensionality reduction methods such as principal component analysis (PCA) and multi-dimensional scaling (MDS), diffusion maps is part of the family of nonlinear dimensionality reduction methods which focus on discovering the underlying manifold that the data has been sampled from. By integrating local similarities at different scales, diffusion maps gives a global description of the data-set. Compared with other methods, the diffusion maps algorithm is robust to noise perturbation and is computationally inexpensive.
Definition of diffusion maps
Diffusion maps exploit the relationship between heat diffusion and random walk Markov chain. The basic observation is that if we take a random walk on the data, walking to a nearby data-point is more likely than walking to another that is far away. Let be a measure space, where is the data set and represents the distribution on the points on .
Based on this, the connectivity between two data points, and , can be defined as the probability of walking from to in one step of the random walk. Usually, this probability is specified in terms of kernel function on the two points: . For example, the popular Gaussian kernel:
More generally, the kernel function has the following properties
( is symmetric)
( is positivity preserving).
The kernel constitutes the prior definition of the local geometry of data-set. Since a given kernel will capture a specific feature of the data set, its choice should be guided by the application that one has in mind. This is a major difference with methods such as principal component analysis, where correlations between all data points are taken into account at once.
Given , we can then construct a reversible Markov chain on (a process known as the normalized graph Laplacian construction):
Although the new normalized kernel does not inherit the symmetric property, it does inherit the positivity-preserving property and gains a conservation property:
From we can construct a transition matrix of a Markov chain () on . In other words, represents the one-step transition probability from to , and gives the t-step transition matrix.
We define the diffusion matrix (it is also a version of graph Laplacian matrix)
We then define the new kernel
where D is a diagonal matrix and
We apply the graph Laplacian normalization to this new kernel:
where is a diagonal matrix and
One of the main ideas of diffusion framework is that running the chain forward in time (taking larger and larger powers of ) reveals the geometric structure of at larger and larger scales (the diffusion process). Specifically, the notion of a cluster in the data set is quantified as a region in which the probability of escaping this region is low (within a certain time t). Therefore, t not only serves as a time parameter, but also has the dual role of scale parameter.
The eigendecomposition of the matrix yields
where is the sequence of eigenvalues of and and are the biorthogonal right and left eigenvectors respectively. Due to the spectrum decay of the eigenvalues, only a few terms are necessary to achieve a given relative accuracy in this sum.
Parameter and the Diffusion Operator
The reason to introduce the normalization step involving is to tune the influence of the data point density on the infinitesimal transition of the diffusion. In some applications, the sampling of the data is generally not related to the geometry of the manifold we are interested in describing. In this case, we can set and the diffusion operator approximates the Laplace–Beltrami operator. We then recover the Riemannian geometry of the data set regardless of the distribution of the points. To describe the long-term behavior of the point distribution of a system of stochastic differential equations, we can use and the resulting Markov chain approximates the Fokker-Planck diffusion. With , it reduces to the classical graph Laplacian normalization.
The diffusion distance at time between two points can be measured as the similarity of two points in the observation space with the connectivity between them. It is given by
where is the stationary distribution of the Markov chain, given by the first left eigenvector of . Explicitly:
Intuitively, is small if there is a large number of short paths connecting and . There are several interesting features associated with the diffusion distance, based on our previous discussion that also serves as a scale parameter:
- Points are closer at a given scale (as specified by ) if they are highly connected in the graph, therefore emphasizing the concept of a cluster.
- This distance is robust to noise, since the distance between two points depends on all possible paths of length between the points.
- From a machine learning point of view, the distance takes into account all evidences linking to , allowing us to conclude that this distance is appropriate for the design of inference algorithms based on the majority of preponderance.
Diffusion process and low-dimensional embedding
The diffusion distance can be calculated using the eigenvectors by
So the eigenvectors can be used as a new set of coordinates for the data. The diffusion map is defined as:
Because of the spectrum decay, it is sufficient to use only the first k eigenvectors and eigenvalues. Thus we get the diffusion map from the original data to a k-dimensional space which is embedded in the original space.
In, it is proved that
so the Euclidean distance in the diffusion coordinates approximates the diffusion distance.
The basic algorithm framework of diffusion map is as:
Step 1. Given the similarity matrix L
Step 2. Normalize the matrix according to parameter :
Step 3. Form the normalized matrix
Step 4. Compute the k largest eigenvalues of and the corresponding eigenvectors
Step 5. Use diffusion map to get the embedding
In the paper, they showed how to design a kernel that reproduces the diffusion induced by a Fokker-Planck equation. Also, they explained that when the data approximate a manifold, then one can recover the geometry of this manifold by computing an approximation of the Laplace-Beltrami operator. This computation is completely insensitive to the distribution of the points and therefore provides a separation of the statistics and the geometry of the data. Since Diffusion map gives a global description of the data-set, it can measure the distances between pair of sample points in the manifold the data is embedded. Based on diffusion map, there are many applications, such as spectral clustering, low dimensional representation of images, image segmentation, 3D model segmentation, speaker identification, sampling on manifolds,anomaly detection, image inpainting, and so on.
- Coifman, R.R.; Lafon, S; Lee, A B; Maggioni, M; Nadler, B; Warner, F; Zucker, S W (2005). "Geometric diffusions as a tool for harmonic analysis and structure definition of data: Diffusion maps". PNAS.
- Coifman, R.R.; Lafon, S; Lee, A B; Maggioni, M; Nadler, B; Warner, F; Zucker, S W (2005). "Geometric diffusions as a tool for harmonic analysis and structure definition of data: Multiscale methods". PNAS.
- Coifman, R.R.; S. Lafon. (2006). "Diffusion maps". Applied and Computational Harmonic Analysis 21: 5–30. doi:10.1016/j.acha.2006.04.006.
- Lafon, S.S. (2004). "Diffusion Maps and Geometric Harmonics". PhD thesis, Yale University.
- De la Porte, J.; Herbst, B M; Hereman, W; Van der Walt, S J (2008). "An Introduction to Diffusion Maps". Proceedings of the nineteenth annual symposium of the Pattern Recognition Association of South Africa (PRASA).
- Nadler, Boaz; Stéphane Lafon; Ronald R. Coifman; Ioannis G. Kevrekidis (2005). "Diffusion Maps, Spectral Clustering and Eigenfunctions of Fokker–Planck Operators" (PDF). in Advances in Neural Information Processing Systems 18.
- Zeev, Farbman; Fattal Raanan; Lischinski Dani (2010). "Diffusion maps for edge-aware image editing". ACM Trans. Graph 29: 145:1–145:10. doi:10.1145/1882261.1866171.
- Oana, Sidi; Oliver van Kaick ; Yanir Kleiman ; Hao Zhang ; Daniel Cohen-Or (2011). Unsupervised Co-Segmentation of a Set of Shapes via Descriptor-Space Spectral Clustering. ACM Transactions on Graphics.
- Michalevsky, Yan; Ronen Talmon; Israel Cohen (2011). "Speaker Identification Using Diffusion Maps" (PDF).
- Mishne, Gal; Israel Cohen (2013). "Multiscale Anomaly Detection Using Diffusion Maps". IEEE Selected Topics in Signal Processing 7: 111–123. doi:10.1109/jstsp.2012.2232279.
- Gepshtein, Shai; Yosi Keller (2013). "Image Completion by Diffusion Maps and Spectral Relaxation". IEEE Transactions on Image Processing 22: 2983–2994. doi:10.1109/tip.2013.2237916.