Review

Review the frozen token-to-logit chain, block and readout shapes, tied and untied parameter ledgers, target alignment, stable loss, causal intervention, and architecture audit order.

The frozen decoder maps IDs to vocabulary logits through one explicit chain:

IE[I]+P[:T]X(0)X(1)X(2)Nf(X(2))Z.I\rightarrow E[I]+P[:T]\rightarrow X^{(0)} \rightarrow X^{(1)}\rightarrow X^{(2)} \rightarrow N_f(X^{(2)})\rightarrow Z.

Every decoder block preserves (B,T,4)(B,T,4) and applies its own causal mask. Final LayerNorm acts across the feature axis. Tied unembedding uses WU=EW_U=E^\top to produce Z:(B,T,8)Z:(B,T,8).

Frozen Configuration

VVCCdmodeld_{model}LLhhdhd_hdffd_{ff}
8442228

The deterministic audit uses learned absolute positions, pre-LayerNorm blocks, ReLU, final LayerNorm, zero dropout, attention matrices without bias, biased MLPs, no output bias, and tied input-output weights.

Reconstruct the Parameter Count

  • embedding: 32;
  • position table: 16;
  • each block: 64+76+16=15664+76+16=156;
  • two blocks: 312;
  • final LayerNorm: 8;
  • tied unembedding: 0 additional entries.

The tied total is 368. An independent 4×84\times8 unembedding raises it to 400.

Reconstruct the Target Contract

For input <bos> A B C, the target is A B C <eos>. Each logits row predicts the token one step to its right. The causal mask prevents later input positions from affecting earlier rows even though all known training positions are calculated in parallel.

For target yy and logit row zz:

=zy+logsumexp(z).\ell=-z_y+\operatorname{logsumexp}(z).

Average only included target positions. Stable log-sum-exp subtracts the row maximum before exponentiation.

Audit Order

  1. ID and context ranges;
  2. lookup and position rows;
  3. block shapes and causal interventions;
  4. final-normalization axis;
  5. tied parameter identity and vocabulary logits;
  6. input-target shift;
  7. vocabulary-axis cross-entropy and valid denominator;
  8. complete parameter ledger.

The next chapter keeps these contracts fixed while introducing trainable parameters, batches, gradients, optimization, validation, and checkpoints.

Pause and reflect

What can you now explain without looking back, and what should you revisit? The note stays with this review.

Review

Not marked done.