Parametric UMAP embeddings for representation and semi-supervised learning
Not on the map. Why?
Abstract (arXiv)
UMAP is a non-parametric graph-based dimensionality reduction algorithm using applied Riemannian geometry and algebraic topology to find low-dimensional embeddings of structured data. The UMAP algorithm consists of two steps: (1) Compute a graphical representation of a dataset (fuzzy simplicial complex), and (2) Through stochastic gradient descent, optimize a low-dimensional embedding of the graph. Here, we extend the second step of UMAP to a parametric optimization over neural network weights, learning a parametric relationship between data and embedding. We first demonstrate that Parametric UMAP performs comparably to its non-parametric counterpart while conferring the benefit of a learned parametric mapping (e.g. fast online embeddings for new data). We then explore UMAP as a regularization, constraining the latent distribution of autoencoders, parametrically varying global structure preservation, and improving classifier accuracy for semi-supervised learning by capturing structure in unlabeled data. Google Colab walkthrough: https://colab.research.google.com/drive/1WkXVZ5pnMrm17m0YgmtoNjM_XHdnE5Vp?usp=sharing
Key facts. Cited by 103 papers in the map (data through 7 Oct 2026). First posted on arXiv 27 Sep 2020.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 6; 2022: 14; 2023: 14; 2024: 21; 2025: 32; 2026: 16.
103 papers.
Sort:
By area:
- 2022Mind the Gap: Understanding the Modality Gap in Multi-modal Contrastive Representation Learningcited by 553
- 2021MatSciBERT: A Materials Domain Language Model for Text Mining and Information Extractioncited by 82
- 2023V2A-Mapper: A Lightweight Solution for Vision-to-Audio Generation by Connecting Foundation Modelscited by 65
- 2023Reverse Engineering Self-Supervised Learningcited by 33
- 2023Learning Topology-Preserving Data Representationscited by 30
- 2022From t-SNE to UMAP with contrastive learningcited by 28
- 2023Geometric Autoencoders -- What You See is What You Decodecited by 23
- 2024WildVis: Open Source Visualizer for Million-Scale Chat Logs in the Wildcited by 20
- 2022DLME: Deep Local-flatness Manifold Embeddingcited by 16
- 2022Unsupervised visualization of image datasets using contrastive learningcited by 16
- 2022Hierarchical Nearest Neighbor Graph Embedding for Efficient Dimensionality Reductioncited by 14
- 2024Bridging Vision and Language Spaces with Assignment Predictioncited by 11
38 more
- 2022EVNet: An Explainable Deep Network for Dimension Reductioncited by 8
- 2022Enhancing Mechanical Metamodels with a Generative Model-Based Augmented Training Datasetcited by 7
- 2023DiffAug: Enhance Unsupervised Contrastive Learning with Domain-Knowledge-Free Diffusion-based Data Augmentationcited by 7
- 2021GenURL: A General Framework for Unsupervised Representation Learningcited by 6
- 2025Stop Misusing t-SNE and UMAP for Visual Analyticscited by 6
- 2022UDRN: Unified Dimensional Reduction Neural Network for Feature Selection and Feature Projectioncited by 5
- 2022How people talk about each other: Modeling Generalized Intergroup Bias and Emotioncited by 5
- 2023ActUp: Analyzing and Consolidating tSNE and UMAPcited by 5
- 2024A General Approach for Determining Applicability Domain of Machine Learning Modelscited by 5
- 2025The Shape of Attraction in UMAP: Exploring the Embedding Forces in Dimensionality Reductioncited by 5
- 2022Exploring Adversarial Attacks and Defenses in Vision Transformers trained with DINOcited by 4
- 2022ParaDime: A Framework for Parametric Dimensionality Reductioncited by 4
- 2023Field-level simulation-based inference with galaxy catalogs: the impact of systematic effectscited by 4
- 2024Image-guided topic modeling for interpretable privacy classificationcited by 4
- 2025Random Forest Autoencoders for Guided Representation Learningcited by 4
- 2025Evaluating Autoencoders for Parametric and Invertible Multidimensional Projectionscited by 4
- 2021Dimension Reduction with Prior Information for Knowledge Discoverycited by 3
- 2023Augment on Manifold: Mixup Regularization with UMAPcited by 3
- 2024Sampling-enabled scalable manifold learning unveils the discriminative cluster structure of high-dimensional datacited by 3
- 2024An embedding-based distance for temporal graphscited by 3
- 2024Inductive Global and Local Manifold Approximation and Projectioncited by 3
- 2024MoE-Enhanced Explainable Deep Manifold Transformation for Complex Data Embedding and Visualizationcited by 3
- 2024Navigating the Effect of Parametrization for Dimensionality Reductioncited by 3
- 2021Leveraging Domain Adaptation for Low-Resource Geospatial Machine Learningcited by 2
- 2024Adversarial Attacks on Machine Learning-Aided Visualizationscited by 2
- 2024Online Posterior Sampling with a Diffusion Priorcited by 2
- 2025Unsupervised Machine Learning for Scientific Discovery: Workflow and Best Practicescited by 2
- 2025Efficient Manifold-Constrained Neural ODE for High-Dimensional Datasetscited by 2
- 2025Generalization of Long-Range Machine Learning Potentials in Complex Chemical Spacescited by 2
- 2021Focusing on Potential Named Entities During Active Label Acquisitioncited by 1
- 2022Local Distance Preserving Auto-encoders using Continuous k-Nearest Neighbours Graphscited by 1
- 2023Domain Adaptation via Rebalanced Sub-domain Alignmentcited by 1
- 2023Topological Data Analysis in smart manufacturing: State of the art and futuredirectionscited by 1
- 2024Exploring the Influence of Dimensionality Reduction on Anomaly Detection Performance in Multivariate Time Seriescited by 1
- 2024HistoKernel: Whole Slide Image Level Maximum Mean Discrepancy Kernels for Pan-Cancer Predictive Modellingcited by 1
- 2024A Hybrid Framework for Spatial Interpolation: Merging Data-driven with Domain Knowledgecited by 1
- 2024Explainable Moral Values: a neuro-symbolic approach to value classificationcited by 1
- 2024Federated t-SNE and UMAP for Distributed Data Visualizationcited by 1
50 most cited of 103 All 103 (JSON)
- 2026A shape-similarity latent space for fluid interfaces: invertible reduced-order modelling of droplet morphology29 Sep 2026
- 2026Partially Linear Autoencoders for Manifold Learning and Dimensionality Reduction30 Aug 2026
- 2026FADEx: Feature Attribution and Distortion-based Explanation of Dimensionality Reduction29 Jul 2026
- 2026FloDR: An invertible dimensionality reduction method based on a normalising flow28 Jul 2026
- 2026Context-Aware Deep Learning for Defect Classification in Atomic-Resolution STEM8 Jun 2026
- 2026Sparsely gated tiny linear experts5 Jun 2026
- 2026On Out-of-sample Embedding in UMAP3 Jun 2026
- 2026Controlled Paraphrase Geometry in Sentence Embedding Space: Local Manifold Modeling and Latent Probing1 May 2026
- 2026Class Angular Distortion Index for Dimensionality Reduction1 May 2026
- 2026Local Neighborhood Instability in Parametric Projections: Quantitative and Visual Analysis23 Apr 2026
- 2026RuleEdit: Failure-Guided Human-AI Model Editing with Prospective Impact Preview12 Apr 2026
- 2026Bi-Lipschitz Autoencoder With Injectivity Guarantee8 Apr 2026
38 more
- 2026Interpreting Manifolds and Graph Neural Embeddings from Internet of Things Traffic Flows5 Feb 2026
- 2026MAPLE: Self-Supervised Learning-Enhanced Nonlinear Dimensionality Reduction for Visual Analysis28 Jan 2026
- 2026The GDN-CC Dataset: Automatic Corpus Clarification for AI-enhanced Democratic Citizen Consultations21 Jan 2026
- 2026Does medical specialization of VLMs enhance discriminative power?: A comprehensive investigation through feature distribution analysis21 Jan 2026
- 2025Cross-modal Counterfactual Explanations: Uncovering Decision Factors and Dataset Biases in Subjective Classification21 Dec 2025
- 2025From Minutes to Days: Scaling Intracranial Speech Decoding with Supervised Pretraining17 Dec 2025
- 2025Generalization of Long-Range Machine Learning Potentials in Complex Chemical Spaces8 Dec 2025
- 2025Probabilistic Foundations of Fuzzy Simplicial Sets for Nonlinear Dimensionality Reduction3 Dec 2025
- 2025Associative Syntax and Maximal Repetitions reveal context-dependent complexity in fruit bat communication30 Nov 2025
- 2025TopoReformer: Mitigating Adversarial Attacks Using Topological Purification in OCR Models19 Nov 2025
- 2025MMCM: Multimodality-aware Metric using Clustering-based Modes for Probabilistic Human Motion Prediction19 Nov 2025
- 2025Chemistry-Enhanced Diffusion-Based Framework for Small-to-Large Molecular Conformation Generation15 Nov 2025
- 2025Efficient Manifold-Constrained Neural ODE for High-Dimensional Datasets5 Oct 2025
- 2025Guided Manifold Alignment with Geometry-Regularized Twin Autoencoders26 Sep 2025
- 2025Conditional Multidimensional Scaling with Incomplete Conditioning Data20 Sep 2025
- 2025Variational Gaussian Mixture Manifold Models for Client-Specific Federated Personalization4 Sep 2025
- 2025TopoMap: A Feature-based Semantic Discriminator of the Topographical Regions in the Test Input Space3 Sep 2025
- 2025Chronotome: Real-Time Topic Modeling for Streaming Embedding Spaces1 Sep 2025
- 2025UMATO: Bridging Local and Global Structures for Reliable Visual Analytics with Dimensionality Reduction22 Aug 2025
- 2025DE-VAE: Revealing Uncertainty in Parametric and Inverse Projections with Variational Autoencoders using Differential Entropy16 Aug 2025
- 2025Leveraging Convolutional and Graph Networks for an Unsupervised Remote Sensing Labelling Tool1 Aug 2025
- 2025Manifold Learning for Personalized and Label-Free Detection of Cardiac Arrhythmias19 Jun 2025
- 2025Stop Misusing t-SNE and UMAP for Visual Analytics10 Jun 2025
- 2025Unsupervised Machine Learning for Scientific Discovery: Workflow and Best Practices5 Jun 2025
- 2025AKRMap: Adaptive Kernel Regression for Trustworthy Visualization of Cross-Modal Embeddings20 May 2025
- 2025Mixed Signals: Understanding Model Disagreement in Multimodal Empathy Detection20 May 2025
- 2025Latent Manifold Reconstruction and Representation with Topological and Geometrical Regularization7 May 2025
- 2025Explanations Go Linear: Post-hoc Explainability for Tabular Data with Interpretable Meta-Encoding29 Apr 2025
- 2025Evaluating Autoencoders for Parametric and Invertible Multidimensional Projections23 Apr 2025
- 2025Manifold Learning for Hyperspectral Images19 Mar 2025
- 2025Enhanced High-Dimensional Data Visualization through Adaptive Multi-Scale Manifold Embedding18 Mar 2025
- 2025The Shape of Attraction in UMAP: Exploring the Embedding Forces in Dimensionality Reduction12 Mar 2025
- 2025Finer Disentanglement of Aleatoric Uncertainty Can Accelerate Chemical Histopathology Imaging27 Feb 2025
- 2025Random Forest Autoencoders for Guided Representation Learning18 Feb 2025
- 2025Supervised Quadratic Feature Analysis: Information Geometry Approach for Dimensionality Reduction31 Jan 2025
- 2025Generalizable Spectral Embedding with an Application to UMAP20 Jan 2025
- 2024Federated t-SNE and UMAP for Distributed Data Visualization18 Dec 2024
- 2024Navigating the Effect of Parametrization for Dimensionality Reduction24 Nov 2024
50 newest of 103 All 103 (JSON)
- 2025Chemistry-Enhanced Diffusion-Based Framework for Small-to-Large Molecular Conformation Generation15 Nov 2025
- 2024Online Posterior Sampling with a Diffusion Prior4 Oct 2024
- 2023DiffAug: Enhance Unsupervised Contrastive Learning with Domain-Knowledge-Free Diffusion-based Data Augmentation10 Sep 2023
- 2022Mind the Gap: Understanding the Modality Gap in Multi-modal Contrastive Representation Learningcited by 553
- 2021MatSciBERT: A Materials Domain Language Model for Text Mining and Information Extractioncited by 82
- 2023V2A-Mapper: A Lightweight Solution for Vision-to-Audio Generation by Connecting Foundation Modelscited by 65
- 2023Reverse Engineering Self-Supervised Learningcited by 33
- 2023Learning Topology-Preserving Data Representationscited by 30
- 2022From t-SNE to UMAP with contrastive learningcited by 28
- 2023Geometric Autoencoders -- What You See is What You Decodecited by 23
- 2024WildVis: Open Source Visualizer for Million-Scale Chat Logs in the Wildcited by 20
- 2022DLME: Deep Local-flatness Manifold Embeddingcited by 16
- 2022Unsupervised visualization of image datasets using contrastive learningcited by 16
- 2022Hierarchical Nearest Neighbor Graph Embedding for Efficient Dimensionality Reductioncited by 14
- 2024Bridging Vision and Language Spaces with Assignment Predictioncited by 11
12 most cited of 97 All 97 (JSON)
- 2026A shape-similarity latent space for fluid interfaces: invertible reduced-order modelling of droplet morphology29 Sep 2026
- 2026Partially Linear Autoencoders for Manifold Learning and Dimensionality Reduction30 Aug 2026
- 2026FADEx: Feature Attribution and Distortion-based Explanation of Dimensionality Reduction29 Jul 2026
- 2026FloDR: An invertible dimensionality reduction method based on a normalising flow28 Jul 2026
- 2026Context-Aware Deep Learning for Defect Classification in Atomic-Resolution STEM8 Jun 2026
- 2026On Out-of-sample Embedding in UMAP3 Jun 2026
- 2026Controlled Paraphrase Geometry in Sentence Embedding Space: Local Manifold Modeling and Latent Probing1 May 2026
- 2026Class Angular Distortion Index for Dimensionality Reduction1 May 2026
- 2026Local Neighborhood Instability in Parametric Projections: Quantitative and Visual Analysis23 Apr 2026
- 2026RuleEdit: Failure-Guided Human-AI Model Editing with Prospective Impact Preview12 Apr 2026
- 2026Bi-Lipschitz Autoencoder With Injectivity Guarantee8 Apr 2026
- 2026Interpreting Manifolds and Graph Neural Embeddings from Internet of Things Traffic Flows5 Feb 2026
12 newest of 97 All 97 (JSON)