Yashotantra Academy logo
level 1
Curious · 0 XP
streak
0d
perfect
0
badges
0

100 XP to Apprentice

Architecture · easy

3. The transformer and self-attention

Attention lets every token weigh every other token.

1 min read3 questionsup to 50 XPtopic 3/12

Read in another language

AI translates the explanation and quiz text while preserving code and answer scoring.

The transformer replaced recurrence with self-attention. Each token is projected into a query, key, and value vector; the attention score between two tokens is the scaled dot product of query and key, softmaxed into weights over the values.

text
Attention(Q, K, V) = softmax(Q Kᵀ / √d) V

Key building blocks of a decoder block: - Multi-head attention — several attention heads learn different relations (syntax, coreference, position). - Causal mask — a token may only attend to earlier positions, which is what makes generation possible. - Feed-forward network (MLP) — per-token nonlinear transformation, usually 4x the hidden width. - Residual connections + normalisation (RMSNorm) — keep very deep stacks trainable.

Attention is O(n²) in sequence length, which is why long contexts are expensive and why FlashAttention, sliding-window attention, and KV-cache tricks exist.

Gotcha: "bigger context" is not free — doubling the prompt roughly quadruples attention work in the prefill phase.

How to study this topic

  1. 1. Read the explanation once end to end, skipping the code.
  2. 2. Re-read the code snippets and say out loud what each line does.
  3. 3. Answer the quiz without scrolling back — that is the real check.
  4. 4. Read every explanation, even for the answers you got right.

Scoring

  • · Up to 30 XP from quiz accuracy.
  • · +15 XP for a perfect first attempt.
  • · +5 XP while a 3-day streak is running.
  • · 75% or higher marks the topic complete.

Check your understanding

  1. 1. What does the causal mask in a decoder-only transformer enforce?

  2. 2. How does self-attention cost scale with sequence length n?

  3. 3. Why use multiple attention heads?

Badges

First light
Pass your first topic
Sharp shooter
3 perfect first-try quizzes
On a roll
Study 3 days in a row
Week warrior
Study 7 days in a row
Grinder
Earn 500 XP
Deep diver
Earn 1500 XP