mlsanityData to 14 Oct 2026

KV-Embedding: Training-free Text Embedding via Internal KV Re-routing in Decoder-only LLMs

arXiv 2601.01046 ↗First posted 3 Jan 2026cs.CLcited by 0

Not on the map. Why?

Abstract (arXiv)

While LLMs are powerful embedding backbones, their application in training-free settings faces two structural challenges: causal attention restricts early tokens from accessing subsequent context, and the next-token prediction objective biases representations toward generation rather than semantic compression. To address these limitations, we propose KV-Embedding, a framework that activates the latent representation power of frozen LLMs. Our method leverages the observation that the key-value (KV) states of the final token at each layer encode a compressed view of the sequence. By re-routing these states as a prepended prefix, we enable all tokens to access sequence-level context within a single forward pass. To ensure model-agnostic applicability, we introduce an automated layer selection strategy based on intrinsic dimensionality. Evaluations on MTEB across Qwen, Mistral, and Llama backbones show that KV-Embedding outperforms existing training-free baselines by up to 10%, while maintaining robust performance on sequences up to 4,096 tokens. These results demonstrate that internal state manipulation offers an efficient alternative to input modification, and we hope this work encourages further exploration of LLM internals for representation learning.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 19 of its 25 references that are in the map. First posted on arXiv 3 Jan 2026.

Builds on

19 of 25 references.

References · 19

  1. 2017Attention Is All You Needcited by 60,361
  2. 2020Retrieval-Augmented Generation for Knowledge-Intensive NLP Taskscited by 9,297
  3. 2021Prefix-Tuning: Optimizing Continuous Prompts for Generationcited by 3,915
  4. 2023Lost in the Middle: How Language Models Use Long Contextscited by 3,455
  5. 2023Efficient Streaming Language Models with Attention Sinkscited by 2,186
  6. 2020Transformers are RNNs: Fast Autoregressive Transformers with Linear Attentioncited by 2,022
  7. 2020Understanding Contrastive Representation Learning through Alignment and Uniformity on the Hyperspherecited by 1,405
  8. 2020Transformer Feed-Forward Layers Are Key-Value Memoriescited by 1,336
  9. 2023Towards General Text Embeddings with Multi-stage Contrastive Learningcited by 723
  10. 2024NV-Embed: Improved Techniques for Training LLMs as Generalist Embedding Modelscited by 472
  11. 2024LLM2Vec: Large Language Models Are Secretly Powerful Text Encoderscited by 404
  12. 2025Layer by Layer: Uncovering Hidden Representations in Language Modelscited by 311
7 more
  1. 2018Estimating the intrinsic dimension of datasets by a minimal neighborhood informationcited by 250
  2. 2023Label Words are Anchors: An Information Flow Perspective for Understanding In-Context Learningcited by 236
  3. 2024Generative Representational Instruction Tuningcited by 233
  4. 2022SGPT: GPT Sentence Embeddings for Semantic Searchcited by 204
  5. 2024Benchmarking and Building Long-Context Retrieval Models with LoCo and M2-BERTcited by 21
  6. 2024Token Prepending: A Training-Free Approach for Eliciting Better Sentence Embeddings from LLMscited by 18
  7. 2025IFIR: A Comprehensive Benchmark for Evaluating Instruction-Following in Expert-Domain Information Retrievalcited by 16

Built on by

Not cited yet.