A Text is Worth Several Tokens: Text Embedding from LLMs Secretly Aligns Well with The Key Tokens
Not on the map. Why?
Abstract (arXiv)
Text embeddings from large language models (LLMs) have achieved excellent results in tasks such as information retrieval, semantic textual similarity, etc. In this work, we show an interesting finding: when feeding a text into the LLM-based embedder, the obtained text embedding will be able to be aligned with the key tokens in the input text. We first fully analyze this phenomenon on eight LLM-based embedders and show that this phenomenon is universal and is not affected by model architecture, training strategy, and embedding method. With a deeper analysis, we find that the main change in embedding space between these embedders and their LLM backbones is in the first principal component. By adjusting the first principal component, we can align text embedding with the key tokens. Finally, we give several examples to demonstrate the vast application potential of this finding: (1) we propose a simple and practical sparse retrieval method based on the aligned tokens, which can achieve 80% of the dense retrieval effect of the same model while reducing the computation significantly; (2) we show that our findings provide a novel perspective to help understand novel technologies (e.g., instruction-following embedding) and fuzzy concepts (e.g., semantic relatedness vs. similarity) in this field.
Key facts. Cited by 8 papers in the map (data through 7 Oct 2026). First posted on arXiv 25 Jun 2024.
Builds on
0 of 0 references.
No references here yet.
Built on by
Citing papers per year: 2024: 1; 2025: 3; 2026: 4.
8 papers.
- 2024LLMs are Also Effective Embedding Models: An In-depth Overviewcited by 32
- 2025Two Heads are Better Than One: Test-time Scaling of Multi-agent Collaborative Reasoningcited by 29
- 2025Attention Illuminates LLM Reasoning: The Preplan-and-Anchor Rhythm Enables Fine-Grained Policy Optimizationcited by 23
- 2026How Does Reasoning Flow? Tracing Attention-Induced Information Flow for Targeted RL in LLMscited by 3
- 2026SemPA: Improving Sentence Embeddings of Large Language Models through Semantic Preference Alignmentcited by 1
- 2026Your UnEmbedding Matrix is Secretly a Feature Lens for Text Embeddingscited by 1
- 2025Text2Token: Unsupervised Text Representation Learning with Token Target Predictioncited by 0
- 2026RGAlign-Rec: Ranking-Guided Alignment for Latent Query Reasoning in Recommendation Systemscited by 0
- 2026How Does Reasoning Flow? Tracing Attention-Induced Information Flow for Targeted RL in LLMs9 Jun 2026
- 2026Your UnEmbedding Matrix is Secretly a Feature Lens for Text Embeddings5 Jun 2026
- 2026RGAlign-Rec: Ranking-Guided Alignment for Latent Query Reasoning in Recommendation Systems13 Feb 2026
- 2026SemPA: Improving Sentence Embeddings of Large Language Models through Semantic Preference Alignment8 Jan 2026
- 2025Attention Illuminates LLM Reasoning: The Preplan-and-Anchor Rhythm Enables Fine-Grained Policy Optimization15 Oct 2025
- 2025Text2Token: Unsupervised Text Representation Learning with Token Target Prediction11 Oct 2025
- 2025Two Heads are Better Than One: Test-time Scaling of Multi-agent Collaborative Reasoning14 Apr 2025
- 2024LLMs are Also Effective Embedding Models: An In-depth Overview17 Dec 2024
Similar, not cited
Close in wording, no citation either way
- 2024Recent advances in text embedding: A Comprehensive Review of Top-Performing Methods on the MTEB Benchmark
- 2026The Embedder's Dilemma: LLMs Are Better, but at What Cost?
- 2025Following the Autoregressive Nature of LLM Embeddings via Compression and Alignment
- 2025Dewey Long Context Embedding Model: A Technical Report
- 2025TokAlign: Efficient Vocabulary Adaptation via Token Alignment
- 2025Align-then-Unlearn: Embedding Alignment for LLM Unlearning
- 2025GEM: Empowering LLM for both Embedding Generation and Language Understanding
- 2026Giga-Embeddings: Mixture-of-Experts Encoders for High-Throughput Text Embeddings