Zubnet AIAprenderWiki › Cross-Attention
Fundamentos

Cross-Attention

Encoder-Decoder Attention
Um mecanismo de atenção onde as queries vêm de uma sequência e as keys/values vêm de uma sequência diferente. Em modelos encoder-decoder, as queries do decoder atendem às keys e values do encoder, permitindo ao decoder “olhar” para a entrada enquanto gera a saída. Cross-attention também é como o texto condiciona a geração de imagens em modelos de difusão — o processo de geração de imagens atende ao prompt de texto.

Por que importa

Cross-attention é a ponte entre diferentes modalidades e diferentes partes de uma arquitetura. É como modelos de tradução conectam línguas fonte e destino, como geradores de imagens seguem prompts de texto, como modelos multimodais relacionam imagens a texto, e como sistemas Retrieval-Augmented incorporam documentos recuperados. Sempre que duas entradas diferentes precisam interagir, cross-attention geralmente está envolvida.

Deep Dive

In self-attention, Q, K, and V all come from the same sequence — each token attends to other tokens in the same input. In cross-attention, Q comes from one source (e.g., the decoder) and K, V come from another (e.g., the encoder). The decoder token asks "what in the input is relevant to what I'm generating right now?" and the attention mechanism provides a weighted summary of the input.

In Diffusion Models

Text-to-image models use cross-attention to condition image generation on text. The text prompt is encoded into embeddings (via CLIP or T5), and at each denoising step, the image features attend to the text embeddings through cross-attention layers. This is how the model knows to generate a "cat on a surfboard" — each spatial location in the image attends to the relevant words. Manipulating these cross-attention maps is how techniques like prompt weighting and attention editing work.

Attention Patterns

Self-attention and cross-attention have different computational profiles. Self-attention is quadratic in the sequence length (every token attends to every other token). Cross-attention is linear in the decoder length times the encoder length (each decoder token attends to all encoder tokens). In practice, the encoder output is often much shorter than the decoder sequence, making cross-attention cheaper than decoder self-attention.

Conceitos relacionados

← Todos os termos
← Cosine Similarity Curriculum Aprendering →