Smoothing Affinity Matrix Impacts t-SNE Neighborhood Preservation

Shirin Mohebi, Guillaume Bied, Jefrey Lijffijt· August 19, 2026 View original

Key takeaways

  • Modifying t-SNE's affinity matrix via a power transform impacts neighborhood preservation.
  • Sharpening the affinity matrix improves preservation of immediate nearest neighbors.
  • Smoothing the affinity matrix enhances broader local neighborhood preservation.
  • This technique can outperform multiscale methods in the mid-local range.

Who benefits

Data ScienceBioinformaticsHealthcareMarketingFinance

Summary

This research investigates how smoothing the affinity matrix in t-SNE, controlled by a row-wise power transform, affects neighborhood preservation at different scales. It finds that sharpening improves preservation of nearest neighbors, while smoothing enhances broader local neighborhoods, outperforming multiscale methods in the mid-local range.

Researchers have explored the impact of modifying the affinity matrix within t-SNE, a widely used dimensionality reduction technique, on its ability to preserve neighborhood structures. The affinity matrix, which represents pairwise similarities as symmetrized probabilities, is central to t-SNE's optimization. The study introduces a row-wise power transform, governed by a parameter 'gamma', to either smooth or sharpen each row of this matrix while maintaining sparsity and rank order. This transform is shown to be equivalent to adjusting the Gaussian bandwidth, and thus perplexity, but with a key distinction: a fixed 'gamma' results in point-dependent effective perplexities, making it different from a global perplexity change. The empirical findings reveal that sharpening the probability distribution improves the preservation of the very nearest neighbors. Conversely, smoothing the distribution enhances the preservation of broader local neighborhoods. Notably, this smoothing technique was found to outperform alternative affinity constructions, including existing multiscale methods, specifically within the mid-local range of neighborhood preservation.

Why it matters

For data scientists and analysts, t-SNE is a crucial tool for visualizing high-dimensional data. Understanding how to manipulate the affinity matrix allows for more precise control over which aspects of neighborhood structure are preserved, leading to more insightful and accurate data visualizations tailored to specific analytical goals.

How to implement this in your domain

  1. 1Experiment with the row-wise power transform (gamma parameter) in t-SNE implementations to fine-tune neighborhood preservation.
  2. 2Apply sharpening (lower gamma) when visualizing very tight clusters or emphasizing immediate nearest neighbors in your data.
  3. 3Utilize smoothing (higher gamma) to better represent broader local structures and relationships in your t-SNE plots.
  4. 4Compare the results of this affinity matrix modification with standard perplexity tuning and other multiscale methods for your specific datasets.
  5. 5Integrate this technique into your data visualization workflow to generate more interpretable and context-specific t-SNE embeddings.

Original post by Shirin Mohebi, Guillaume Bied, Jefrey Lijffijt

"arXiv:2608.17190v1 Announce Type: new Abstract: Dimensionality reduction methods are instrumental to visualize high-dimensional data, and t-SNE stands as one of the most widely used methods due to its emphasis on local neighborhood preservation. A central component of t-SNE is th…"

View on X

Originally posted by Shirin Mohebi, Guillaume Bied, Jefrey Lijffijt on X · view source

Want to go deeper?

Turn these trends into skills with Learnijoy's hands-on AI & tech courses.

Explore courses

More in AI Research