首页 > AI前沿 > Lend Me Your Eyes: Instruction-Aware Text Embeddings via Attention Relay

Lend Me Your Eyes: Instruction-Aware Text Embeddings via Attention Relay

arXiv自然语言 2026-10-05 05:52 5 阅读 查看原文

Text embedding models trained with contrastive learning learn to follow task instructions from instruction-paired data, while instruction-tuned LLMs already know how to follow them.

We show that this instruction-following ability can carry over from an LLM to a Transformer-based embedder without any training.

We propose Attention Relay, which passes the attention weights an LLM produces to the embedder's own attention.

Across six instruction-tuned LLMs from the Qwen3, Llama 3.1 and OLMo 3 families and ten widely used embedding models that differ in tokenizer, size and pooling type, Attention Relay makes nearly every combination instruction-aware.

Experiments that break the method down into its parts show that the LLM's attention weights track the instruction in its later layers and come largely from instruction tuning.