Rdimtools (version 0.3.2)

do.tsne: t-distributed Stochastic Neighbor Embedding

Description

t-distributed Stochastic Neighbor Embedding (t-SNE) is a variant of Stochastic Neighbor Embedding (SNE) that mimicks patterns of probability distributinos over pairs of high-dimensional objects on low-dimesional target embedding space by minimizing Kullback-Leibler divergence. While conventional SNE uses gaussian distributions to measure similarity, t-SNE, as its name suggests, exploits a heavy-tailed Student t-distribution. For do.tsne, we implemented a naive version of t-SNE as well as an interface with Barnes-Hut algorithm, which takes advantage of speed in sacrifice of accuracy.

Usage

do.tsne(X, ndim = 2, perplexity = 30, eta = 0.05, maxiter = 2000,
  jitter = 0.3, jitterdecay = 0.99, momentum = 0.5,
  preprocess = c("null", "center", "scale", "cscale", "decorrelate",
  "whiten"), pca = TRUE, pcaratio = 0.9, pcascale = FALSE,
  symmetric = FALSE, BarnesHut = FALSE, BHtheta = 0.5)

Arguments

X

an \((n\times p)\) matrix or data frame whose rows are observations and columns represent independent variables.

ndim

an integer-valued target dimension.

perplexity

desired level of perplexity; ranging [5,50].

eta

learning parameter.

maxiter

maximum number of iterations.

jitter

level of white noise added at the beginning.

jitterdecay

decay parameter in (0,1). The closer to 0, the faster artificial noise decays.

momentum

level of acceleration in learning.

preprocess

an additional option for preprocessing the data. Default is "null". See also aux.preprocess for more details.

pca

whether to use PCA as preliminary step; TRUE for using it, FALSE otherwise.

pcaratio

proportion of variances explained in finding PCA preconditioning. See also do.pca for more details.

pcascale

a logical; FALSE for using Covariance, TRUE for using Correlation matrix. See also do.pca for more details.

symmetric

a logical; FALSE to solve it naively, and TRUE to adopt symmetrization scheme.

BarnesHut

a logical; FALSE not to use Barnes-Hut heuristic for faster computation, TRUE otherwise.

BHtheta

a positive real number for speed/accuracy trade-off in Barnes-Hut computation. See also theta parameter from Rtsne.

Value

a named list containing

Y

an \((n\times ndim)\) matrix whose rows are embedded observations.

trfinfo

a list containing information for out-of-sample prediction.

References

van_der_maaten_visualizing_2008Rdimtools

See Also

do.sne, Rtsne.

Examples

# NOT RUN {
## generate swiss roll data
X = aux.gensamples(n=100)

## compare different perplexity
out1 <- do.tsne(X, ndim=2, perplexity=5)
out2 <- do.tsne(X, ndim=2, perplexity=10)
out3 <- do.tsne(X, ndim=2, perplexity=50)

## Visualize three different projections
par(mfrow=c(1,3))
plot(out1$Y[,1], out1$Y[,2], main="tSNE::perplexity=5")
plot(out1$Y[,1], out1$Y[,2], main="tSNE::perplexity=10")
plot(out1$Y[,1], out1$Y[,2], main="tSNE::perplexity=50")
# }
# NOT RUN {
# }