Your UnEmbedding Matrix is Secretly a Feature Lens for Text Embeddings

29d ago · Global · primary source: export.arxiv.org

A team of researchers has identified a mechanism that degrades the performance of large language models when used as text embedding tools and proposed a linear transformation called EmbedFilter to correct it, according to a paper submitted June 5, 2026. [1] Large language models, or LLMs, show strong zero-shot abilities across many tasks but deliver suboptimal results when deployed directly as embedding models on large-scale text embedding benchmarks. [1] The authors of the paper trace this shortfall to an unexpected behavior: text embeddings, when projected onto the vocabulary space, align with frequent but semantically weak tokens. [1] That alignment suppresses the model's capacity to capture nuanced meaning. [1] The investigation focused on the unembedding matrix inside LLMs. The researchers found that this matrix encodes a latent subspace they call the "edge spectrum" space, spanned by the right singular vectors with the smallest and largest singular values. [3] This subspace actively writes high-frequency tokens into the embedding space. [1] When the projection of a reverse-engineered "average" token onto this subspace is truncated, the logits of those frequent tokens are significantly disrupted. [3] To counteract the effect, the team built EmbedFilter, a simple linear transformation that filters out the edge spectrum subspace. [1] By removing that component, the method suppresses the influence of high-frequency tokens and strengthens semantic representations. [1] The approach also produces an inherent dimensionality reduction, which lowers index storage requirements and speeds up retrieval while preserving the refined embedding quality. [1] Experiments across multiple LLM backbones showed that models equipped with EmbedFilter achieve superior zero-shot downstream performance even with significantly reduced embedding dimensions. [1] The paper's authors describe the unembedding matrix as secretly functioning as a feature lens for text embeddings and hope the findings will encourage more principled designs for training text embeddings. [9] The code for EmbedFilter has been released on GitHub. [1]

research-papermodel-releaseproduct-launchtool-release

Background sources we checked (10)
  • arxiv.org ↗ Large language models exhibit impressive zero-shot capabilities across a wide range of downstream tasks. However, they struggle to function as off-the-shelf embedding models, leading to suboptimal performance on massive text embedding benchmarks. In this paper, we identify a pote…
  • arxiv.org ↗ Large language models exhibit impressive zero-shot capabilities across a wide range of downstream tasks. However, they struggle to function as off-the-shelf embedding models, leading to suboptimal performance on massive text embedding benchmarks. In this paper, we identify a pote…
  • arxiv.org ↗ Large language models exhibit impressive zero-shot capabilities across a wide range of downstream tasks. However, they struggle to function as off-the-shelf embedding models, leading to suboptimal performance on massive text embedding benchmarks. In this paper, we identify a pote…
  • arxiv.org ↗ Large language models exhibit impressive zero-shot capabilities across a wide range of downstream tasks. However, they struggle to function as off-the-shelf embedding models, leading to suboptimal performance on massive text embedding benchmarks. In this paper, we identify a pote…
  • en.wikipedia.org ↗ Google Maps is a web mapping platform and consumer application developed by Google. It offers satellite imagery, aerial photography, street maps, 360° interactive panoramic views of streets (Street View), real-time traffic conditions, and route planning for traveling by foot, car…
  • en.wikipedia.org ↗ Privacy concerns with Google include a wide range of issues and controversies related to collection, use, and sharing of user data across Google products, services, and platforms, including Google Search. Since the mid-2000s, civil liberties organizations and government regulator…
  • en.wikipedia.org ↗ Criticism of Google includes concern for tax avoidance, misuse and manipulation of search results, its use of others' intellectual property, concerns that its compilation of data may violate people's privacy and collaboration with the U.S. military on Google Earth to spy on users…
  • huggingface.co ↗ Title: Your UnEmbedding Matrix is Secretly a Feature Lens for Text Embeddings [...] Large language models exhibit impressive zero-shot capabilities across a wide range of downstream tasks. However, they struggle to function as off-the-shelf embedding models, leading to suboptimal…
  • en.wikipedia.org ↗ Algorithmic bias describes systematic and repeatable harmful tendency in a computerized sociotechnical system to create "unfair" outcomes, such as "privileging" one category over another in ways that may or may not be different from the intended function of the algorithm. Bias ca…
  • info.arxiv.org ↗ arXiv Labs - arXiv info | arXiv e-print repository Skip to content # arXiv Labs Attention arXiv Users: arXiv Labs is pausing new proposals ## What are arXiv Labs? arXiv Labs are a way for the community to contribute new, useful features to arXiv. These integrations are avail…

Sources

Spot something wrong? Report an issue