Zubnet AIAprenderWiki › Embedding Layer
Fundamentos

Embedding Layer

Token Embedding, Embedding Table, Lookup Table
Uma tabela de lookup que mapeia cada token no vocabulário a um vetor denso (o embedding do token). Quando o modelo recebe o token ID 42, a embedding layer retorna a linha 42 de uma matriz aprendida. Esse vetor é a representação inicial do modelo desse token — o ponto de partida para todo o processamento subsequente através das camadas de atenção e feedforward.

Por que importa

A embedding layer é onde texto se torna matemática. Todo LLM começa convertendo tokens discretos (palavras, subpalavras) em vetores contínuos que a rede neural pode processar. A tabela de embedding também é um dos maiores componentes de modelos pequenos — um vocabulário de 128K com embeddings de 4096 dimensões são 512 milhões de parâmetros. Entender isso te ajuda a raciocinar sobre tamanhos de modelo e design de vocabulário.

Deep Dive

The embedding layer is just a matrix E of shape (vocab_size, model_dim). For token ID i, the embedding is E[i] — a simple row lookup, no computation. But these embeddings are learned during training: tokens that appear in similar contexts get similar embeddings. The classic example: the embeddings for "king" − "man" + "woman" ≈ "queen," showing that the embedding space captures semantic relationships.

Tied Embeddings

Many models share (tie) the embedding matrix with the output layer (the "unembedding" or "language model head"). The output layer converts hidden states back into vocabulary probabilities by computing a dot product with each token's embedding. Tying these layers means the same embedding both represents a token on input and predicts it on output, saving parameters and often improving quality. Most modern LLMs use tied embeddings.

Positional + Token Embeddings

The full input representation is typically: token_embedding + positional_encoding. The token embedding captures what the token means. The positional encoding captures where it appears in the sequence. In models with learned position embeddings (BERT), this is a second embedding table indexed by position. In models with RoPE (LLaMA), positional information is injected differently (by rotating Q and K vectors), and the embedding layer only handles token identity.

Conceitos relacionados

← Todos os termos
← Embedding Emergence →