Tutorials › Statistical Methods for Single-Cell RNA Sequencing › UMAP Parameters and Their Geometric Effects

Dimension Reduction and Embeddings · Tutorial 219 of 700

UMAP Parameters and Their Geometric Effects

Understand which parts of UMAP’s graph and embedding each key parameter influences, and how to compare parameter settings without overinterpreting the resulting geometry.

Intermediate 10 min read

What You'll Learn

  • Distinguish n_neighbors, which changes the input-space neighborhood graph, from parameters that shape low-dimensional spacing.
  • Interpret how increasing n_neighbors can shift emphasis toward broader structure without guaranteeing preserved global geometry.
  • Explain min_dist as a control on the low-dimensional membership curve, not a hard minimum between points.
  • Relate spread to the distance scale of that curve and understand why its effect is not a guaranteed map expansion.
  • Compare parameter settings while holding preprocessing and other UMAP choices fixed.
  • Diagnose overinterpretation and recognize when an apparent parameter effect is confounded.

Parameters Act on Different Parts of UMAP

UMAP’s parameters do not all adjust the same geometric object. n_neighbors helps determine which cells are connected in the input-space fuzzy graph. min_dist and spread shape the low-dimensional membership curve used when fitting the embedding. Consequently, changing one parameter can alter the map through a different mechanism from changing another.

This distinction matters when visualizing single-cell RNA-sequencing data. A map may look more compact or show different apparent group structure after a parameter change, but that appearance alone does not identify what changed in the analysis. A useful comparison first asks whether the input graph changed, whether the low-dimensional spacing preference changed, or both.

The explanations below assume the same cells, input representation, distance metric, and other UMAP settings are used in each comparison. Examples use hypothetical data and embeddings to illustrate parameter effects; actual optimized coordinates depend on the full graph and optimization.

Definition: n_neighbors is the neighborhood-size setting used to construct UMAP’s input-space nearest-neighbor graph. min_dist sets the distance below which the low-dimensional membership curve has a high-membership plateau. spread controls the distance scale over which that curve declines beyond the plateau. These are distinct controls: the first acts on graph construction, while the latter two shape low-dimensional spacing preferences.

How the Parameters Enter the Geometry

For each cell, UMAP uses a neighborhood of nearby cells in the input representation to build a fuzzy graph. In the usual conceptual description, n_neighbors sets the approximate number of neighbors considered for each cell. Local distance calibration then determines how strongly those neighbor relations contribute. The directed local relations are combined into a fuzzy graph whose edge weights describe the strength of cell-cell connections.

Increasing n_neighbors generally allows each cell’s graph neighborhood to reach farther through the input data. The resulting graph can include more medium- or longer-range relationships, so the embedding objective has information beyond the very closest cells. Reducing n_neighbors emphasizes a more local graph. These are tendencies, not guarantees that the final map will preserve a particular global arrangement: local calibration, graph connectivity, optimization, and the rest of the data all matter.

In the low-dimensional space, write \(y_i\) for the coordinates of cell \(i\), and let \(d_{ij}=\lVert y_i-y_j\rVert\) be the Euclidean distance between two embedded cells. A common description of the target low-dimensional membership curve is

$$ f(d)= \begin{cases} 1, & d < \text{min\_dist},\\ \exp\!\left(-\dfrac{d-\text{min\_dist}}{\text{spread}}\right), & d\geq \text{min\_dist}. \end{cases} $$

This curve is a useful way to interpret the two spacing parameters. For distances below min_dist, the target membership remains at its high plateau; after that, it declines, with spread setting the scale of the decline. Implementations generally fit a smooth low-dimensional similarity curve to this target rather than imposing the displayed piecewise function as a literal pairwise constraint. Thus the equation explains the intended effect, not an exact formula for every optimized pair of coordinates.

The curve is used in the low-dimensional part of UMAP’s cross-entropy embedding objective. The input graph supplies the relationships the embedding is asked to represent; the low-dimensional curve maps coordinate distances to membership strengths. Since the objective balances matching graph relationships with the cost of mismatches, changing the curve changes the preferred spacing but does not dictate coordinates by itself.

Worked Example: A Larger Neighborhood Reaches Farther

Suppose a hypothetical cell A has the following nearest cells in the input representation, listed in order of increasing distance: B at 0.10, C at 0.14, D at 0.20, E at 0.35, F at 0.42, and G at 0.60. Compare conceptual neighborhoods of size 3 and size 6.

With n_neighbors set to 3, the nearest-neighbor list for A includes B, C, and D. With n_neighbors set to 6, it also includes E, F, and G. The additional relations reach as far as 0.60 in this example, compared with 0.20 for the first three neighbors. This is a change to the input graph, not a direct instruction to place E, F, and G at particular distances from A in the finished map.

For a parameter comparison, the appropriate conclusion is that the larger setting supplies a broader set of candidate graph relationships for A. Other cells’ neighborhoods and their calibrated weights also contribute, so the final embedding cannot be calculated from A’s list alone. If a map changes, the graph can be inspected to determine whether newly included or differently weighted relationships help explain the change.

Worked Example: Changing min_dist Changes the Plateau

Consider two hypothetical curve settings with spread fixed at 1.00. In setting A, min_dist is 0.10; in setting B, min_dist is 0.30. Evaluate the target membership at distances 0.05 and 0.30.

At distance 0.05, both distances are below both settings’ min_dist values. Therefore \(f(0.05)=1\) for setting A and \(f(0.05)=1\) for setting B. At distance 0.30, setting A is beyond its plateau, so \(f(0.30)=\exp[-(0.30-0.10)/1.00]=\exp(-0.20)\approx0.8187\). For setting B, 0.30 is still on the plateau, so \(f(0.30)=1\).

The larger min_dist therefore keeps the target membership high to a greater distance in this comparison. In an optimized embedding, this often produces less tightly packed very-nearby points than a smaller min_dist, but it does not enforce a fixed minimum distance between every pair of cells. Graph structure and optimization still determine the actual coordinates.

Worked Example: Changing spread Changes the Distance Scale

Hold min_dist at 0.10 and compare spread values of 1.00 and 2.00. At distance 1.10, the distance beyond the plateau is \(1.10-0.10=1.00\). With spread 1.00, the target membership is \(\exp(-1.00/1.00)=\exp(-1)\approx0.3679\). With spread 2.00, it is \(\exp(-1.00/2.00)=\exp(-0.50)\approx0.6065\).

For the same distance beyond the plateau, the larger spread gives a higher target membership: the curve declines more slowly across that distance range. For spread 1.00, a distance of 1.10 beyond the origin of the curve’s plateau would require evaluating at \(d=1.10\), as above. More directly, one spread unit beyond the plateau gives membership \(\exp(-1)\approx0.3679\); with spread 2.00, that same one-unit offset is half a spread unit and gives \(\exp(-0.50)\approx0.6065\).

This is a statement about the curve’s distance scale, not a promise that the entire embedding will have twice the diameter when spread doubles. The optimized layout is shaped by graph weights, repulsion, and other optimization details as well.

Comparing Settings Carefully

A paired embedding comparison is most informative when it isolates a change. Keep the input representation, metric, initialization, and other settings fixed, then vary one parameter. If several settings change at once, an observed visual difference cannot be attributed cleanly to n_neighbors, min_dist, or spread.

Parameter changedPrimary object affectedTypical tendencyWhat not to conclude
n_neighborsInput-space fuzzy neighborhood graphSmaller values emphasize nearer relationships; larger values incorporate more neighborsThat a larger value guarantees correct global distances
min_distLow-dimensional membership curveSmaller values allow a high-membership plateau only at shorter distances, often permitting tighter packingThat min_dist is a hard lower bound for all pairwise distances
spreadLow-dimensional membership curveLarger values make the curve’s decline operate over a broader distance scaleThat the map’s diameter must increase proportionally

The settings interact in the fitted map because the graph and the low-dimensional curve jointly contribute to the embedding objective. For example, a broader graph can ask the embedding to represent relationships across a larger input-space neighborhood, while min_dist and spread govern how low-dimensional distances are translated into memberships. It is therefore useful to inspect both the parameter values and the resulting neighborhood relationships, rather than reading a change in visual compactness as a direct measure of biological separation.

Diagnostics, Failure Modes, and Limits

Before interpreting a difference between paired maps, check that the analysis actually differs only in the intended parameter. Confirm that the same cells and input features were used, and that the input representation and distance metric were held constant. A changed feature set or metric can alter the neighbor graph and confound an interpretation of n_neighbors. If multiple UMAP parameters were altered together, the comparison does not isolate their individual effects.

Useful diagnostics follow the parameter’s mechanism. To investigate n_neighbors, compare the input-space neighbor graph or the neighbor lists for selected cells across settings. If many cells acquire different neighbors, the graph itself changed; that is expected, but the embedding’s appearance alone does not reveal which graph edges changed. For min_dist and spread, compare settings with the same graph and inspect whether low-dimensional close-neighbor spacing changes in the expected direction. These checks diagnose a parameter’s influence; they do not establish that one setting is universally correct.

A common failure is to treat a visual tendency as a guaranteed geometric rule. A very small min_dist can permit tight packing, but it does not require every group to collapse into a compact island. A large spread can broaden the target distance scale, but it does not force the map to expand by a fixed factor. If the expected visual effect is absent, the graph structure and the rest of the optimization may outweigh it; the parameter has not necessarily been ignored.

Another mistake is to infer that increasing n_neighbors automatically improves global geometry. It changes which input-space relationships contribute to the graph; it does not make the two-dimensional map a faithful ruler for all input-space distances. The appropriate claim is narrower: a larger neighborhood setting incorporates more neighbors into graph construction and can shift the information available to the embedding toward broader relationships.

No parameter setting removes the basic limitations of a two-dimensional representation of high-dimensional cell profiles. A parameter choice is a modeling choice about which relationships and spacing preferences to emphasize, not a test of whether biological groups exist. For a principled comparison, consider the scientific purpose, check graph changes when n_neighbors varies, and avoid selecting a setting solely because its picture looks most separated.

A Practical Comparison Sequence

1
Hold the analysis fixed.
Use the same cells, input representation, distance metric, and other UMAP settings when comparing a single parameter.
2
Identify the mechanism.
For n_neighbors, inspect the input graph. For min_dist or spread, interpret the low-dimensional membership curve and resulting spacing.
3
Describe what changed narrowly.
Report the parameter values and observed changes in neighborhoods or spacing; distinguish these from biological conclusions not established by the map.
Key takeaway: n_neighbors controls the scale of neighborhood information in UMAP’s input graph; min_dist and spread shape how low-dimensional distances translate into membership strengths. Their effects are tendencies mediated by the full embedding objective, not hard geometric guarantees.

Check Your Understanding

Answer each question, then compare with the solutions.

  1. A cell has input-space neighbors at distances 0.08, 0.12, 0.19, and 0.41. Which additional relation enters its conceptual neighborhood when n_neighbors increases from 3 to 4, and what part of UMAP changes directly?
  2. With spread 1.00, calculate the target membership at distance 0.50 for min_dist 0.20. Show the substitution.
  3. Does setting min_dist to 0.05 guarantee that no two embedded cells will be less than 0.05 units apart? Explain.
  4. What should be held fixed to attribute a paired-map difference specifically to spread?

Solutions. (1) The neighbor at distance 0.41 is added; the input-space neighborhood graph changes directly. (2) Since \(0.50\geq0.20\), \(f(0.50)=\exp[-(0.50-0.20)/1.00]=\exp(-0.30)\approx0.7408\). (3) No. min_dist shapes a target membership curve; it is not a hard lower bound on every pairwise embedding distance. (4) Hold fixed the cells, input representation, metric, initialization, and other UMAP settings, changing only spread.

Clinical Biostats