- New parameter:
rng_type
. This will be used in favor of the booleanpcg_rand
parameter, althoughpcg_rand
will still work for backwards compatibility. - New negative sampling option: set
rng_type = "deterministic"
to use a deterministic sampling of vertices during the optimization phase. This should give qualitatively similar results to using a real PRNG, but has the advantage of being faster and giving more reproducible output. This feature was inspired by a comment by Leland McInnes on Reddit.
- Setting
num_threads
directly inumap2
did not result in the number of SGD threads being updated to that value whenbatch = TRUE
, which it should have been. - Despite assertions to the contrary in version 0.2.1,
umap_transform
continued to return the fuzzy graph in transposed form. Thank you PedroMilanezAlmeida for reopening the issue (#118). - Relative paths could not be used to save a model. Thank you Wouter van der Bijl for the bug report (#131) and the suggested fix.
repulsion_strength
was silently ignored if used withtumap
orumap2
witha = 1, b = 1
. Ignoring the setting was on purpose, but it was not documented anywhere.repulsion_strength
is now compatible with these settings.
RSpectra
is now a required dependency (again). It was a required dependency up until version 0.1.12, when it became optional (irlba
was used in its place). However, problems with interactions of the current version ofirlba
with an ABI change in theMatrix
package means that it's hard for downstream packages and users to builduwot
without re-installingMatrix
andirlba
from source, which may not be an option for some people. Also it was causing a CRAN check error. I have changed some tests, examples and vignettes to useRSpectra
explicitly, and to only testirlba
code-paths where necessary. See #115 and links therein for more details.
- The HNSW approximate nearest neighbor
search algorithm is now supported via the
RcppHNSW package. Set
nn_method = "hnsw"
to use it. The behavior of the method can be controlled by the newnn_args
parameter, a list which may containM
,ef_construction
andef
. See the hnswlib library's ALGO_PARAMS documentation for details on these parameters. Although typically faster than Annoy (for a given accuracy), be aware that the only supportedmetric
values are"euclidean"
,"cosine"
and"correlation"
. Finally, RcppHNSW is only a suggested package, not a requirement, so you need to install it yourself (e.g. viainstall.packages("RcppHNSW")
). Also see the article on HNSW in uwot in the documentation. - The nearest neighbor descent approximate nearest neighbor search algorithm is
now supported via the
rnndescent package. Set
nn_method = "nndescent"
to use it. The behavior of the method can be controlled by the newnn_args
parameter. There are many supported metrics and possible parameters that can be set innn_args
, so please see the article on nearest neighbor descent in uwot in the documentation, and also the rnndescent package's documentation for details.rnndescent
is only a suggested package, not a requirement, so you need to install it yourself (e.g. viainstall.packages("rnndescent")
). - New function:
umap2
, which acts likeumap
but with modified defaults, reflecting my experience with UMAP and correcting some small mistakes. See the umap2 article for more details.
init_sdev = "range"
caused an error with a user-suppliedinit
matrix.- Transforming new data with the
correlation
metric was actually using thecosine
metric if you saved and reloaded the model. Thank you Holly Hall for the report and helpful detective work (#117). umap_transform
could fail if the new data to be transformed had thescaled:center
andscaled:scale
attributes set (e.g. from applying thescale
function).- If you asked
umap_transform
to return the fuzzy graph (ret_extra = c("fgraph")
), it was transposed whenbatch = TRUE, n_epochs = 0
. Thank you PedroMilanezAlmeida for reporting (#118). - Setting
n_sgd_threads = "auto"
withumap_transform
caused a crash. - A warning was being emitted due to not being specific enough about what
dist
class was meant that may have been particularly affecting Seurat users. Thank you AndiMunteanu for reporting (and suggesting a solution) (#121).
- A small change to a header file was required to fully support the next version of RcppAnnoy. Thank you Dirk Eddelbuettel for the PR (#112).
- New function:
optimize_graph_layout
. Use this to produce optimized output coordinates that reflect an input similarity graph (such as that produced by thesimilarity_graph
function.similarity_graph
followed byoptimize_graph_layout
is the same as runningumap
, so the purpose of these functions is to allow for more flexibility and decoupling between generating the nearest neighbor graph and optimizing the low-dimensional approximation to it. Based on a request by user Chengwei94 (#98). - New functions:
simplicial_set_union
andsimplicial_set_intersect
. These allow for the combination of different fuzzy graph representations of a dataset into a single fuzzy graph using the UMAP simplicial set operations. Based on a request in the Python UMAP issues tracker by user Dhar xion. - New parameter for
umap_transform
:ret_extra
. This works like the equivalent parameter forumap
, and should be a character vector specifying the extra information you would like returned in addition to the embedding, in which case a list will be returned with anembedding
member containing the optimized coordinates. Supported values are"fgraph"
,"nn"
,"sigma"
and"localr"
. Based on a request by user PedroMilanezAlmeida (#104). - New parameter from
umap
,tumap
andumap_transform
:seed
. This will do the equivalent of callingset.seed
internally, and hence will help with reproducibility. The chosen seed is exported ifret_model = TRUE
andumap_transform
will use that seed if present, so you only need to specify it inumap_transform
if you want to change the seed. The default behavior remains to not modify the random number state. Based on a request by SuhasSrinivasan (#110).
- A new setting for
init_sdev
: setinit_sdev = "range"
and initial coordinates will be range-scaled so each column takes values between 0-10. This pre-processing was added to the Python UMAP package at some point afteruwot
began development and so should probably always be used with the defaultinit = "spectral"
setting. However, it is not set by default to maintain backwards compatibility with older versions ofuwot
. ret_extra = c("sigma")
is now supported bylvish
. The Gaussian bandwidths are returned in asigma
vector. In addition, a vector of intrinsic dimensionalities estimated for each point using an analytical expression of the finite difference method given by Lee and co-workers is returned in thedint
vector.- The
min_dist
andspread
parameters are now returned in the model whenumap
is run withret_model = TRUE
. This is just for documentation purposes, these values are not used directly by the model inumap_transform
. If the parametersa
andb
are set directly when invokingumap
, then bothmin_dist
andspread
will be set toNULL
in the returned model. This feature was added in response to a question from kjiang18 (#95). - Some new checks for NA values in input data have been added. Also a warning
will be emitted if
n_components
seems to have been set too high. - If
n_components
was greater thann_neighbors
thenumap_transform
would crash the R session. Thank you to ChVav for reporting this (#102). - Using
umap_transform
with a model wheredens_scale
was set could cause a segmentation fault, destroying the session. Even if it didn't it could give an entirely artifactual "ring" structure. Thank you FemkeSmit for reporting this and providing assistance in diagnosing the underlying cause (#103). - If you set
binary_edge_weights = TRUE
, this setting was not exported whenret_model = TRUE
, and was therefore not respected byumap_transform
. This has now been fixed, but you will need to regenerate any models that used binary edge weights. - The rdoc for the
init
param said that if there were multiple disconnected components, a spectral initialization would attempt to merge multiple sub-graphs. Not true: actually, spectral initialization is abandoned in favor of PCA. The documentation has been updated to reflect the true state of affairs. No idea what I was thinking of there. load_model
andsave_model
didn't work on Windows 7 due to how the version oftar
there handles drive letters. Thank you mytarmail for the report (#109).- Warn if the initial coordinates have a very large scale (a standard deviation
10.0), because this can lead to small gradients and poor optimization. Thank you SuhasSrinivasan for the report (#110).
- A change to accommodate a forthcoming version of RcppAnnoy. Thank you Dirk Eddelbuettel for the PR (#111).
- New function:
similarity_graph
. If you are more interested in the high-dimensional graph/fuzzy simplicial set representation of your input data, and don't care about the low dimensional approximation, thesimilarity_graph
function offers a similar API toumap
, but neither the initialization nor optimization of low-dimensional coordinates will be performed. The return value is the same as that which would be returned in the results list as thefgraph
member if you had providedret_extra = c("fgraph")
. Compared to getting the same result via runningumap
, this function is a bit more convenient to use, makes your intention clearer if you would be discarding the embedding, and saves a small amount of time. A t-SNE/LargeVis similarity graph can be returned by settingmethod = "largevis"
.
- If a model was generated without using pre-generated nearest neighbors, you
couldn't use
umap_transform
with pre-generated nearest neighbors (also the error message was completely useless). Thank you to AustinHartman for reporting this (#97).
- This is a resubmission of 0.1.12 but with an internal function
(
fuzzy_simplicial_set
) refactored to behave more like that of previous versions. This change was breaking the behavior of the CRAN package bbknnR.
- New parameter:
dens_weight
. If set to a value between 0 and 1, an attempt is made to include the relative local densities of the input data in the output coordinates. This is an approximation to the densMAP method. A large value ofdens_weight
will use a larger range of output densities to reflect the input data. If the data is too spread out, reduce the value ofdens_weight
. For more information see the documentation at the uwot repo. - New parameter:
binary_edge_weights
. If set toTRUE
, instead of smoothed knn distances, non-zero edge weights all have a value of 1. This is how PaCMAP works and there is practical and theoretical reasons to believe this won't have a big effect on UMAP but you can try it yourself. - New options for
ret_extra
:"sigma"
: the return value will contain asigma
entry, a vector of the smooth knn distance scaling normalization factors, one for each observation in the input data. A small value indicates a high density of points in the local neighborhood of that observation. Forlvish
the equivalent bandwidths calculated for the input perplexity is returned.- also, a vector
rho
will be exported, which is the distance to the nearest neighbor after the number of neighbors specified by thelocal_connectivity
. Only applies forumap
andtumap
. "localr"
: exports a vector of the local radii, the sum ofsigma
andrho
and used to scale the output coordinates whendens_weight
is set. Even if not usingdens_weight
, visualizing the output coordinates using a color scale based on the value oflocalr
can reveal regions of the input data with different densities.
- For functions
umap
andtumap
only: new data type for precomputed nearest neighbor data passed as thenn_method
parameter: you may use a sparse distance matrix of formatdgCMatrix
with dimensionsN x N
whereN
is the number of observations in the input data. Distances should be arranged by column, i.e. a non-zero entry in rowj
of thei
th column indicates that thej
th observation in the input data is a nearest neighbor of thei
th observation with the distance given by the value of that element. Note that this is a different format to the sparse distance matrix that can be passed as input toX
: notably, the matrix is not assumed to be symmetric. Unlike other input formats, you may have a different number of neighbors for each observation (but there must be at least one neighbor defined per observation). umap_transform
can also take a sparse distance matrix as itsnn_method
parameter if precomputed nearest neighbor data is used to generate an initial model. The format is the same as for thenn_method
withumap
. Because distances are arranged by columns, the expected dimensions of the sparse matrix isN_model x N_new
whereN_model
is the number of observations in the original data andN_new
is the number of observations in the data to be transformed.
- Models couldn't be re-saved after loading. Thank you to ilyakorsunsky for reporting this (#88).
- RSpectra is now a 'Suggests',
rather than an 'Imports'. If you have RSpectra installed, it will be used
automatically where previous versions required it (for spectral initialization).
Otherwise, irlba will be used.
For two-dimensional output, you are unlikely to notice much difference in speed
or accuracy with real-world data. For highly-structured simulation datasets
(e.g. spectral initialization of a 1D line) then RSpectra will give much better,
faster initializations, but these are not the typical use cases envisaged for
this package. For embedding into higher dimensions (e.g.
n_components = 100
or higher), RSpectra is recommended and will likely out-perform irlba even if you have installed a good linear algebra library. init = "laplacian"
returned the wrong coordinates because of a slightly subtle issue around how to order the eigenvectors when using the random walk transition matrix rather than normalized graph laplacians.- The
init_sdev
parameter was ignored when theinit
parameter was a user-supplied matrix. Now the input will be scaled. - Matrix input was being converted to and from a data frame during pre-processing, causing R to allocate memory that it was disinclined to ever give up even after the function exited. This unnecessary manipulation is now avoided.
- The behavior of the
bandwidth
parameter has been changed to give results more like the current version (0.5.2) of the Python UMAP implementation. This is likely to be a breaking change for non-default settings ofbandwidth
, but this is not a parameter which is actually exposed by the Python UMAP public API any more, so is on the road to deprecation in uwot too and I don't recommend you change this. - Transforming data with multiple blocks would give an error if the number of rows of the new data did not equal the number of number of rows in the original data.
- New parameter:
batch
. IfTRUE
, then results are reproducible whenn_sgd_threads > 1
(as long as you useset.seed
). The price to be paid is that the optimization is slightly less efficient (because coordinates are not updated as quickly and hence gradients are staler for longer), so it is highly recommended to setn_epochs = 500
or higher. Thank you to Aaron Lun who not only came up with a way to implement this feature, but also wrote an entire C++ implementation of UMAP which does it (#83). - New parameter:
opt_args
. The default optimization method whenbatch = TRUE
is Adam. You can control its parameters by passing them in theopt_args
list. As Adam is a momentum-based method it requires extra storage of previous gradient data. To avoid the extra memory overhead you can also useopt_args = list(method = "sgd")
to use a stochastic gradient descent method like that used whenbatch = FALSE
. - New parameter:
epoch_callback
. You may now pass a function which will be invoked at the end of each epoch. Mainly useful for producing an image of the state of the embedding at different points during the optimization. This is another feature taken from umappp. - New parameter:
pca_method
, used when thepca
parameter is supplied to reduce the initial dimensionality of the data. This controls which method is used to carry out the PCA and can be set to one of:"irlba"
which usesirlba::irlba
to calculate a truncated SVD. If this routine deems that you are trying to extract 50% or more of the singular vectors, you will see a warning to that effect logged to the console."rsvd"
, which usesirlba::svdr
for truncated SVD. This method uses a small number of iterations which should give an accuracy/speed up trade-off similar to that of the scikit-learn TruncatedSVD method. This can be much faster than using"irlba"
but potentially at a cost in accuracy. However, for the purposes of dimensionality reduction as input to nearest neighbor search, this doesn't seem to matter much."bigstatsr"
, which uses the bigstatsr package will be used. Note: that this is not a dependency ofuwot
. If you want to usebigstatsr
, you must install it yourself. On platforms without easy access to fast linear algebra libraries (e.g. Windows), usingbigstatsr
may give a speed up to PCA calculations."svd"
, which usesbase::svd
. Warning: this is likely to be very slow for most datasets and exists as a fallback for small datasets where the"irlba"
method would print a warning."auto"
(the default) which uses"irlba"
to calculate a truncated SVD, unless you are attempting to extract 50% or more of the singular vectors, in which case"svd"
is used.
- If row names are provided in the input data (or nearest neighbor data, or
initialization data if it's a matrix), this will be used to name the rows of the
output embedding (#81), and also the
nearest neighbor data if you set
ret_nn = TRUE
. If the names exist in more than one of the input data parameters listed above, but are inconsistent, no guarantees are made about which names will be used. Thank you jwijffels for reporting this. - In
umap_transform
, the learning rate is now down-scaled by a factor of 4, consistent with the Python implementation of UMAP. If you need the old behavior back, use the (newly added)learning_rate
parameter inumap_transform
to set it explicitly. If you used the default value inumap
when creating the model, the correct setting inumap_transform
islearning_rate = 1.0
. - Setting
nn_method = "annoy"
andverbose = TRUE
would lead to an error with datasets with fewer than 50 items in them. - Using multiple pre-computed nearest neighbors blocks is now supported with
umap_transform
(this was incorrectly documented to work). - Documentation around pre-calculated nearest neighbor data for
umap_transform
was wrong in other ways: it has now been corrected to indicate that there should be neighbor data for each item in the test data, but the neighbors and distances should refer to items in training data (i.e. the data used to build the model). n_neighbors
parameter is now correctly ignored in model generation if pre-calculated nearest neighbor data is provided.- Documentation incorrectly said
grain_size
didn't do anything.
This release is mainly to allow for some internal changes to keep compatibility with RcppAnnoy, used for the nearest neighbor calculations.
- Passing in data with missing values will now raise an error early. Missing data in factor columns intended for supervised UMAP is still ok. Thank you David McGaughey for tweeting about this issue.
- The documentation for the return value of
umap
andtumap
now note that the contents of themodel
list are subject to change and not intended to be part of the uwot public API. I recommend not relying on the structure of themodel
, especially if your package is intended to appear on CRAN or Bioconductor, as any breakages will delay future releases of uwot to CRAN.
- New metric:
metric = "correlation"
a distance based on the Pearson correlation (#22). Supporting this required a change to the internals of how nearest neighbor data is stored. Backwards compatibility with models generated by previous versions usingret_model = TRUE
should have been preserved.
- New parameter,
nn_method
, forumap_transform
: pass a list containing pre-computed nearest neighbor data (identical to that used in theumap
function). You should not pass anything to theX
parameter in this case. This extends the functionality for transforming new points to the case where nearest neighbor data between the original data and new data can be calculated external touwot
. Thanks to Yuhan Hao for contributing the PR (#63 and #64). - New parameter,
init
, forumap_transform
: provides a variety of options for initializing the output coordinates, analogously to the same parameter in theumap
function (but without as many options currently). This is intended to replaceinit_weighted
, which should be considered deprecated, but won't be removed until uwot 1.0 (whenever that is). Instead ofinit_weighted = TRUE
, useinit = "weighted"
; replaceinit_weighted = FALSE
withinit = "average"
. Additionally, you can pass a matrix toinit
to act as the initial coordinates. - Also in
umap_transform
: previously, settingn_epochs = 0
was ignored: at least one iteration of optimization was applied. Now,n_epochs = 0
is respected, and will return the initialized coordinates without any further optimization. - Minor performance improvement for single-threaded nearest neighbor search when
verbose = TRUE
: the progress bar calculations were taking up a detectable amount of time and has now been fixed. With very small data sets (< 50 items) the progress bar will no longer appear when building the index. - Passing a sparse distance matrix as input now supports upper/lower triangular matrix storage rather than wasting storage using an explicitly symmetric sparse matrix.
- Minor license change: uwot used to be licensed under GPL-3 only; now it is GPL-3 or later.
- default for
n_threads
is nowNULL
to provide a bit more protection from changing dependencies. - parallel code now uses the standard C++11 implementation of threading rather than tinythread++.
- The
grain_size
parameter has been undeprecated. As the version that deprecated this never made it to CRAN, this is unlikely to have affected many people.
- uwot should no longer trigger undefined behavior in sanitizers, due to the temporary replacement of the RcppParallel package with code "borrowed" from that package and using tinythread++ rather than tbb (#52).
- Further sanitizer improvements in the nearest neighbor search code due to the upstream efforts of erikbern and eddelbuettel (#50).
- The
grain_size
parameter is now ignored and remains to avoid breaking backwards compatibility only.
- New parameter,
ret_extra
, a vector which can contain any combination of:"model"
(same asret_model = TRUE
),"nn"
(same asret_nn = TRUE
) andfgraph
(see below). - New return value data: If the
ret_extra
vector contains"fgraph"
, the returned list will contain anfgraph
item representing the fuzzy simplicial input graph as a sparse N x N matrix. Forlvish
, use"P"
instead of"fgraph
" (#47). Note that there is a further sparsifying step where edges with a very low membership are removed if there is no prospect of the edge being sampled during optimization. This is controlled byn_epochs
: the smaller the value, the more sparsifying will occur. If you are only interested in the fuzzy graph and not the embedded coordinates, setn_epochs = 0
. - New function:
unload_uwot
, to unload the Annoy nearest neighbor indices in a model. This prevents the model from being used inumap_transform
, but allows for the temporary working directory created by bothsave_uwot
andload_uwot
to be deleted. Previously, bothload_uwot
andsave_uwot
were attempting to delete the temporary working directories they used, but would always silently fail because Annoy is making use of files in those directories. - An attempt has been made to reduce the variability of results due to different
compiler and C++ library versions on different machines. Visually results are
unchanged in most cases, but this is a breaking change in terms of numerical
output. The best chance of obtaining floating point determinism across machines
is to use
init = "spca"
, fixed values ofa
andb
(rather than allowing them to be calculated through settingmin_dist
andspread
) andapprox_pow = TRUE
. Using thetumap
method withinit = "spca"
is probably the most robust approach.
- New behavior when
n_epochs = 0
. This used to behave like (n_epochs = NULL
) and gave a default number of epochs (dependent on the number of vertices in the dataset). Now it more usefully carries out all calculations except optimization, so the returned coordinates are those specified by theinit
parameter, so this is an easy way to access e.g. the spectral or PCA initialization coordinates. If you want the input fuzzy graph (ret_extra
vector contains"fgraph"
), this will also prevent the graph having edges with very low membership being removed. You still get the old default epochs behavior by settingn_epochs = NULL
or to a negative value. save_uwot
andload_uwot
have been updated with averbose
parameter so it's easier to see what temporary files are being created.save_uwot
has a new parameter,unload
, which if set toTRUE
will delete the working directory for you, at the cost of unloading the model, i.e. it can't be used withumap_transform
until you reload it withload_uwot
.save_uwot
now returns the saved model with an extra field,mod_dir
, which points to the location of the temporary working directory, so you should now assign the result of callingsave_uwot
to the model you saved, e.g.model <- save_uwot(model, "my_model_file")
. This field is intended for use withunload_uwot
.load_uwot
also returns the model with amod_dir
item for use withunload_uwot
.save_uwot
andload_uwot
were not correctly handling relative paths.- A previous bug fix to
load_uwot
in uwot 0.1.4 to work with newer versions of RcppAnnoy (#31) failed in the typical case of a single metric for the nearest neighbor search using all available columns, giving an error message along the lines of:Error: index size <size> is not a multiple of vector size <size>
. This has now been fixed, but required changes to bothsave_uwot
andload_uwot
, so existing saved models must be regenerated. Thank you to reporter OuNao.
- The R API was being accessed from inside multi-threaded code to seed the (non-R) random number generators. Probably this was causing users in downstream projects (seurat and monocle) to experience strange RcppParallel-related crashes. Thanks to aldojongejan for reporting this (#39).
- Passing a floating point value smaller than one to
n_threads
caused a crash. This was particularly insidious if running with a system with only one default thread available as the defaultn_threads
becomes0.5
. Nown_threads
(andn_sgd_threads
) are rounded to the nearest integer. - Initialization of supervised UMAP should now be faster (#34). Contributed by Aaron Lun.
- Fixed incorrect loading of Annoy indexes to be compatible with newer versions of RcppAnnoy (#31). My thanks to Dirk Eddelbuettel and Erik Bernhardsson for aid in identifying the problem.
- Fix for
ERROR: there is already an InterruptableProgressMonitor instance defined
. - If
verbose = TRUE
, thea
,b
curve parameters are now logged.
- Fixed an issue where the session would crash if the Annoy nearest neighbor search was unable to find k neighbors for an item.
Even with a fix for the bug mentioned above, if the nearest neighbor index file
is larger than 2GB in size, Annoy may not be able to read the data back in. This
should only occur with very large or high-dimensional datasets. The nearest
neighbor search will fail under these conditions. A work-around is to set
n_threads = 0
, because the index will not be written to disk and re-loaded
under these circumstances, at the cost of a longer search time. Alternatively,
set the pca
parameter to reduce the dimensionality or lower n_trees
, both of
which will reduce the size of the index on disk. However, either may lower the
accuracy of the nearest neighbor results.
Initial CRAN release.
- New parameter,
tmpdir
, which allows the user to specify the temporary directory where nearest neighbor indexes will be written during Annoy nearest neighbor search. The default isbase::tempdir()
. Only used ifn_threads > 1
andnn_method = "annoy"
.
-
Fixed an issue with
lvish
where there was an off-by-one error when calculating input probabilities. -
Added a safe-guard to
lvish
to prevent the gaussian precision, beta, becoming overly large when the binary search fails during perplexity calibration. -
The
lvish
perplexity calibration uses the log-sum-exp trick to avoid numeric underflow if beta becomes large.
- New parameter:
pcg_rand
. IfTRUE
(the default), then a random number generator from the PCG family is used during the stochastic optimization phase. The old PRNG, a direct translation of an implementation of the Tausworthe "taus88" PRNG used in the Python version of UMAP, can be obtained by settingpcg_rand = FALSE
. The new PRNG is slower, but is likely superior in its statistical randomness. This change in behavior will be break backwards compatibility: you will now get slightly different results even with the same seed. - New parameter:
fast_sgd
. IfTRUE
, then the following combination of parameters are set:n_sgd_threads = "auto"
,pcg_rand = FALSE
andapprox_pow = TRUE
. These will result in a substantially faster optimization phase, at the cost of being slightly less accurate and results not being exactly repeatable.fast_sgd = FALSE
by default but if you are only interested in visualization, thenfast_sgd
gives perfectly good results. For more generic dimensionality reduction and reproducibility, keepfast_sgd = FALSE
. - New parameter:
init_sdev
which specifies how large the standard deviation of each column of the initial coordinates should be. This will scale any input coordinates (including user-provided matrix coordinates).init = "spca"
can now be thought of as an alias ofinit = "pca", init_sdev = 1e-4
. This may be too aggressive scaling for some datasets. The typical UMAP spectral initializations tend to result in standard deviations of around2
to5
, so this might be more appropriate in some cases. If spectral initialization detects multiple components in the affinity graph and falls back to scaled PCA, it usesinit_sdev = 1
. - As a result of adding
init_sdev
, theinit
optionssspectral
,slaplacian
andsnormlaplacian
have been removed (they weren't around for very long anyway). You can get the same behavior by e.g.init = "spectral", init_sdev = 1e-4
.init = "spca"
is sticking around because I use it a lot.
- Spectral initialization (the default) was sometimes generating coordinates that had too large a range, due to an erroneous scale factor that failed to account for negative coordinate values. This could give rise to embeddings with very noticeable outliers distant from the main clusters.
- Also during spectral initialization, the amount of noise being added had a standard deviation an order of magnitude too large compared to the Python implementation (this probably didn't make any difference though).
- If requesting a spectral initialization, but multiple disconnected components
are present, fall back to
init = "spca"
. - Removed dependency on C++
<random>
header. This breaks backwards compatibility even if you setpcg_rand = FALSE
. metric = "cosine"
results were incorrectly using the unmodified Annoy angular distance.- Numeric matrix columns can be specified as the target for the
categorical
metric (fixes #20).
- Data is now stored column-wise during optimization, which should result in
an increase in performance for larger values of
n_components
(e.g. approximately 50% faster optimization time with MNIST andn_components = 50
). - New parameter:
pca_center
, which controls whether to center the data before applying PCA. It would be typical to set this toFALSE
if you are applying PCA to binary data (although note you can't use this with setting withmetric = "hamming"
) - PCA will now be used when the
metric
is"manhattan"
and"cosine"
. It's still not applied when using"hamming"
(data still needs to be in binary format, not real-valued). - If using mixed datatypes, you may override the
pca
andpca_center
parameter values for a given data block by using a list for the value of the metric, with the column ids/names as an unnamed item and the overriding values as named items, e.g. instead ofmanhattan = 1:100
, usemanhattan = list(1:100, pca_center = FALSE)
to turn off PCA centering for just that block. This functionality exists mainly for the case where you have mixed binary and real-valued data and want to apply PCA to both data types. It's normal to apply centering to real-valued data but not to binary data.
- Fixed bug that affected
umap_transform
, where negative sampling was over the size of the test data (should be the training data). - Some other performance improvements (around 10% faster for the optimization stage with MNIST).
- When
verbose = TRUE
, log the Annoy recall accuracy, which may help tune values ofn_trees
andsearch_k
.
- New parameter:
n_sgd_threads
, which controls the number of threads used in the stochastic gradient descent. By default this is now single-threaded and should result in reproducible results when usingset.seed
. To get back the old, less consistent, but faster settings, setn_sgd_threads = "auto"
. - API change for consistency with Python UMAP:
alpha
is nowlearning_rate
.gamma
is nowrepulsion_strength
.
- Default spectral initialization now looks for disconnected components and
initializes them separately (also applies to
laplacian
andnormlaplacian
). - New
init
options:sspectral
,snormlaplacian
andslaplacian
. These are likespectral
,normlaplacian
,laplacian
respectively, but scaled so that each dimension has a standard deviation of 1e-4. This is like the difference between thepca
andspca
options.
- Hamming distance support (was actually using Euclidean distance).
- Smooth knn/perplexity calibration results had a small dependency on the number of threads used.
- Anomalously long spectral initialization times should now be reduced.
- Internal changes and fixes thanks to a code review by Aaron Lun.
- New parameter
pca
: set this to a positive integer to reduce matrix of data frames to that number of columns using PCA. Only works ifmetric = "euclidean"
. If you have > 100 columns, this can substantially improve the speed of the nearest neighbor search. t-SNE implementations often set this value to 50.
- Laplacian Eigenmap initialization convergence failure is now correctly detected.
- C++ code was over-writing data passed from R as a function argument.
- Highly experimental mixed data type support for
metric
: instead of specifying a single metric name (e.g.metric = "euclidean"
), you can pass a list, where the name of each item is the metric to use and the value is a vector of the names of the columns to use with that metric, e.g.metric = list("euclidean" = c("A1", "A2"), "cosine" = c("B1", "B2", "B3"))
treats columnsA1
andA2
as one block, using the Euclidean distance to find nearest neighbors, whereasB1
,B2
andB3
are treated as a second block, using the cosine distance. - Factor columns can also be used in the metric, using the metric name
categorical
. y
may now be a data frame or matrix if multiple target data is available.- New parameter
target_metric
, to specify the distance metric to use with numericaly
. This has the same capabilities asmetric
. - Multiple external nearest neighbor data sources are now supported. Instead of passing a list of two matrices, pass a list of lists, one for each external metric.
- More details on mixed data types can be found at https://github.com/jlmelville/uwot#mixed-data-types.
- Compatibility with older versions of RcppParallel (contributed by sirusb).
scale = "Z"
To Z-scale each column of input (synonym forscale = TRUE
orscale = "scale"
).- New scaling option,
scale = "colrange"
to scale columns in the range (0, 1).
- Hamming distance is now supported, due to upgrade to RcppAnnoy 0.0.11.
- For supervised UMAP with numeric
y
, you may pass nearest neighbor data directly, in the same format as that supported byX
-related nearest neighbor data. This may be useful if you don't want to use Euclidean distances for they
data, or if you have missing data (and have a way to assign nearest neighbors for those cases, obviously). See the Nearest Neighbor Data Format section for details.
- New parameter
ret_nn
: whenTRUE
returns nearest neighbor matrices as ann
list: indices in itemidx
and distances in itemdist
. Embedded coordinates are inembedding
. Bothret_nn
andret_model
can beTRUE
, and should not cause any compatibility issues with supervised embeddings. nn_method
can now take precomputed nearest neighbor data. Must be a list of two matrices:idx
, containing integer indexes, anddist
containing distances. By no coincidence, this is the format return byret_nn
.
- Embedding to
n_components = 1
was broken (#6) - User-supplied matrices to
init
parameter were being modified, in defiance of basic R pass-by-copy semantics.
metric = "cosine"
is working again forn_threads
greater than0
(#5)
-
August 5 2018. You can now use an existing embedding to add new points via
umap_transform
. See the example section below. -
August 1 2018. Numerical vectors are now supported for supervised dimension reduction.
-
July 31 2018. (Very) initial support for supervised dimension reduction: categorical data only at the moment. Pass in a factor vector (use
NA
for unknown labels) as they
parameter and edges with bad (or unknown) labels are down-weighted, hopefully leading to better separation of classes. This works remarkably well for the Fashion MNIST dataset. -
July 22 2018. You can now use the cosine and Manhattan distances with the Annoy nearest neighbor search, via
metric = "cosine"
andmetric = "manhattan"
, respectively. Hamming distance is not supported because RcppAnnoy doesn't yet support it.