Research map / Federated learning and privacy / Differential privacy
Federated t-SNE and UMAP for Distributed Data Visualization
Read the paperCheck your idea against 108,967 papers
AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.
Proposes Fed-tSNE/Fed-UMAP (and + variants with noise, plus federated spectral clustering) that learn a global set of landmark points via MMD-based federated distribution learning and Nyström-approximate the global distance matrix without sharing raw data.
Problem
Running t-SNE and UMAP (and spectral clustering) on high-dimensional data distributed across clients requires pairwise distances across all points, which requires sharing raw data and leaks privacy.
Approach
Federated distribution learning (FedDL) learns intermediate landmark points Y by minimizing the average MMD between each client's data and Y; the server then estimates the global distance/kernel matrix via Nyström approximation from client-side cross-distance matrices, enabling t-SNE, UMAP, and spectral clustering. Privacy-enhanced variants add Gaussian noise to data, variables, or gradients.
Key results
- On MNIST IID, Fed-tSNE 1-NN classification accuracy is 0.9400 vs 0.9618 for standard t-SNE, and Fed-UMAP 0.9066 vs 0.9322 for UMAP.
- Fed-SpeClust on COIL-20 yields NMI 0.8425 vs 0.8885 for standard spectral clustering (IID).
- Privacy-enhanced Fed-tSNE+ on non-IID MNIST shows lower NPA (e.g. 1-NN 0.1928 vs 0.2729 for Fed-tSNE), indicating a privacy-accuracy trade-off.
Measured effects reported in the paper
| What was changed | Where | Measured | Effect |
|---|---|---|---|
| Use Fed-tSNE instead of centralized t-SNE | t-SNE · MNIST IID | NMI: 0.7747$\pm$0.0243 -> 0.7534$\pm$0.0202 | Hurt |
| Use Fed-UMAP instead of centralized UMAP | UMAP · MNIST IID | NMI: 0.8285$\pm$0.0150 -> 0.7844$\pm$0.0208 | Hurt |
| Add noise for privacy protection (ablation) | Fed-tSNE · MNIST IID | NMI: 0.7534$\pm$0.0202 -> 0.7471$\pm$0.0073 | Hurt |
| Add noise for privacy protection (ablation) | Fed-tSNE · MNIST non-IID | NMI: 0.7399$\pm$0.0109 -> 0.7025$\pm$0.0149 | Hurt |
| Add noise for privacy protection (ablation) | Fed-UMAP · MNIST IID | NMI: 0.7844$\pm$0.0208 -> 0.7812$\pm$0.0153 | Hurt |
| Add noise for privacy protection (ablation) | Fed-UMAP · MNIST non-IID | NMI: 0.7919$\pm$0.0217 -> 0.7368$\pm$0.0194 | Hurt |
| Use Fed-tSNE instead of centralized t-SNE | t-SNE · Fashion-MNIST IID | NMI: 0.5787$\pm$0.0212 -> 0.5780$\pm$0.0154 | No clear effect |
| Increase noise by using the privacy-enhanced variant (ablation) | Fed-UMAP · Fashion-MNIST non-IID | NMI: 0.5877$\pm$0.0181 -> 0.5191$\pm$0.0132 | Hurt |
| Use Fed-SpeClust instead of centralized spectral clustering | Spectral clustering · Mice-Protein IID | NMI: 0.3241$\pm$0.0063 -> 0.3233$\pm$0.0121 | No clear effect |
| Add noise for privacy protection (ablation) | Fed-SpeClust · COIL-20 IID | NMI: 0.8425$\pm$0.0218 -> 0.8333$\pm$0.0173 | Hurt |
| Increase the number of landmarks n_y (ablation) | Fed-tSNE and Fed-UMAP · MNIST | NMI | Helped |
| Decrease noise level β (ablation) | Fed-tSNE+ and Fed-UMAP+ · MNIST | NMI | Helped |
Surprising findings
- Fed-tSNE performs slightly better on IID than non-IID splits, and the paper notes the non-IID setting with one class per client is hardest.
- Fed-tSNE+ noise reduces neighbor preservation considerably on non-IID data (e.g. Fashion-MNIST NPA 1-NN 0.0071 vs 0.0356 for standard t-SNE is not the comparison; within-method drops are large).
Limitations
- Accuracy drops relative to non-federated t-SNE/UMAP/spectral clustering, with privacy-enhanced variants losing more.
- Privacy guarantees depend on noise levels and assumptions (bounded data norms), and the trade-off between accuracy and privacy is governed by sigma.
- Experiments use only four datasets with a simple client-partition setup (IID vs single-class non-IID).
Experimental setup
Datasets: MNIST Fashion-MNIST COIL-20 Mice-Protein
Benchmarks: MNIST Fashion-MNIST COIL-20 Mice-Protein
Metrics: CA k-NN accuracy NPA k-NN NMI Silhouette coefficient ARI
Baselines: t-SNE UMAP dSNE SpeClust
Builds on: t-SNE UMAP Nyström method FedAvg
Closest papers by meaning
- OmniFC: Rethinking Federated Clustering via Lossless and Secure Distance Reconstruction (NeurIPS 2025 · similarity 0.86)
- Graph Consistency and Diversity Measurement for Federated Multi-View Clustering (AAAI 2025 · similarity 0.86)
- Differentially Private Federated $k$-Means Clustering with Server-Side Data (ICML 2025 · similarity 0.85)
- FedNE: Surrogate-Assisted Federated Neighbor Embedding for Dimensionality Reduction (NeurIPS 2024 · similarity 0.85)
- Scalable Federated One-Step Multi-View Clustering with Tensorized Regularization (AAAI 2025 · similarity 0.85)
- FedMSGL: A Self-Expressive Hypergraph Based Federated Multi-View Learning (AAAI 2025 · similarity 0.84)
Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.
