Paper Overview
Research area: NLP Authors: Songhao Wu, Zhongxin Chen, Yuxuan Liu Published: 2025-06-11 arXiv: 2506.08638
Original Abstract
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 potential cause underlying this deficiency. Our motivation stems from an unexpected observation: text embeddings tend to align with frequent but uninformative tokens when projected onto the vocabulary space. We argue that this excessive expression of high-frequency tokens suppresses the model's ability to capture nuanced semantics. To address this, we introduce EmbedFilter, a simple linear transformation designed to refine text embeddings derived from LLMs directly. Specifically, we uncover that the unembedding matrix ...
Key Points (from the forum summary)
- LLMs perform poorly as plug-and-play embedding models on massive text embedding benchmarks, despite strong zero-shot capabilities elsewhere.
- Cause identified: text embeddings, when projected onto vocabulary space, align with frequent but uninformative (high-frequency) tokens, which suppresses fine-grained semantic capture.
- EmbedFilter: a simple linear transformation applied directly to LLM-derived text embeddings.
- The unembedding matrix in an LLM encodes a latent subspace that actively writes high-frequency token directions into the embedding space; filtering this subspace suppresses their influence and enhances semantic representation.
- Side benefit: inherent dimensionality reduction — lower index storage cost and faster retrieval, with refined embedding quality fully preserved.
- Multi-backbone experiments show LLMs equipped with EmbedFilter achieve superior zero-shot downstream performance even with significantly reduced embedding dimensions.