From Topology to Retrieval: Decoding Embedding Spaces with Unified Signatures
cs.LG, cs.CL, cs.IR
Submitted: 2025-11-27
Updated: 2026-10-12
Code: https://github.com/Boehringer-Ingelheim/unified-topological-signatures
License: http://creativecommons.org/licenses/by/4.0/
The gist: Studying the shape of text embedding spaces enhances our understanding of model behavior and how geometry affects downstream task performance.
Terminology
Abstract
Studying the shape of text embedding spaces enhances our understanding of model behavior and how geometry affects downstream task performance. While a range of topological and geometric metrics have been explored, they are typically analyzed in isolation, and the connections between them remain largely unknown. In this paper, we present Unified Topological Signatures (UTS) for characterizing the structure of embedding spaces. We evaluate our method across a wide range of text embedding models and retrieval datasets, finding that it outperforms individual metrics in downstream performance prediction. Through several experiments, we demonstrate that the signatures characterize embedding spaces more accurately, improve the prediction of model-specific properties and biases, and reveal novel insights into representational similarity that challenge the global claims of the Platonic Representation Hypothesis.
Sources
- Representation Topology Divergence: A Method for Comparing Neural Network Representations
- Beyond Benchmarks: Evaluating Embedding Model Similarity for Retrieval Augmented Generation Systems
- SPECTER: Document-level Representation Learning using Citation-informed Transformers
- Reliability of CKA as a Similarity Measure in Deep Learning
- How Contextual are Contextualized Word Representations? Comparing the Geometry of BERT, ELMo, and GPT-2 Embeddings
- The Vendi Score: A Diversity Evaluation Metric for Machine Learning
- Persistent Topological Features in Large Language Models
- Large Language Models Are Zero-Shot Time Series Forecasters
- Geometric Interpretation of Layer Normalization and a Comparative Analysis with RMSNorm
- The Platonic Representation Hypothesis
- Manifold Approximation leads to Robust Kernel Alignment
- A Comparative Study of Learning Paradigms in Large Language Models via Intrinsic Dimension
- Measuring Intrinsic Dimension of Token Embeddings
- Shared Global and Local Geometry of Language Model Embeddings
- Gemini Embedding: Generalizable Embeddings from Gemini
- Geometric Signatures of Compositionality Across a Language Model's Lifetime
- Towards General Text Embeddings with Multi-stage Contrastive Learning
- Spectral Insights into Data-Oblivious Critical Layers in Large Language Models
- UMAP: Uniform Manifold Approximation and Projection for Dimension Reduction
- Arctic-Embed: Scalable, Efficient, and Accurate Text Embedding Models
Related papers
- Polynomial-Augmented Neural Networks (PANNs) with Weak Orthogonality Constraints for Enhanced Function and PDE Approximation
- AIRL-S: Unifying Reinforcement Learning and Search-Based Test-Time Scaling via Adversarial Inverse Reinforcement Learning
- Transformers as Bayesian In-Context Experimenters: Smoothness-Adaptive Efficient ATE Estimation
- Convergence issues in Relational Concept Analysis based on AOC-posets
- Beliefs Beyond Posteriors: The Predictive Cost of ELBO Binding
- Understanding Diffusion Models via Ratio-Based Function Approximation with SignReLU Networks