Home › Tracks › Applied ML & Model Engineering › Transformers, Attention & Pretraining
Week 3 concept
Transformers, Attention & Pretraining
Why attention replaced recurrence, and what pretraining a language model on a corpus actually optimizes. Tokenization, the training objective, and scaling laws.
Bridges to Machine Learning — sequence modeling and representation learningBuilds on: Deep Learning Foundations
Study notes
Master this concept.
Transformers, Attention & Pretraining
What it is
The Transformer is a neural network architecture that processes entire sequences of data simultaneously rather than one element at a time. Its core mechanism is "Attention," which allows the model to dynamically weigh the importance of different words in a sentence, regardless of how far apart they are. Pretraining is the process of training this architecture on a massive, unlabeled text corpus to learn the statistical patterns of language.
Why it matters
Before Transformers, AI used recurrence (RNNs), which processed text linearly. This was slow and often "forgot" the beginning of a long sentence by the time it reached the end. Transformers enable massive parallelization across GPUs, allowing models to scale to trillions of tokens. This scalability is what makes modern Large Language Models (LLMs) capable of complex reasoning and general-purpose utility.
Core concepts to master
- Self-Attention: The process where each token in a sequence looks at every other token to determine context (e.g., realizing "it" refers to "the robot" in a previous sentence).
- Tokenization: The method of breaking raw text into smaller chunks (sub-words) that the model can process as numerical vectors.
- The Training Objective: Most pretrained models use "Next Token Prediction," where the model is optimized to minimize the difference between its guess and the actual next word in the dataset.
- Scaling Laws: The empirical observation that model performance improves predictably as you increase three variables: the number of parameters, the size of the dataset, and the amount of compute.
Common mistakes
- Confusing Attention with Intelligence: Attention is a mathematical weighting mechanism, not a conscious process of "focusing."
- Overlooking Tokenization: Assuming the model sees "words." In reality, it sees tokens, which can lead to errors in spelling or basic math because the model doesn't see individual letters.
- Ignoring Data Quality: Believing that more data always equals a better model; the quality and diversity of the pretraining corpus are often more impactful than raw volume.
Track connection
This concept serves as the foundation for the rest of the Applied ML track. Once you understand how a model is pretrained on a corpus, you can move into supervised fine-tuning (SFT) and Reinforcement Learning from Human Feedback (RLHF) to turn a raw base model into a helpful assistant.
Go to the source
Read, watch, and practice.
Free, world-class material chosen for this concept.
- Paper Attention Is All You Need arXiv (Vaswani et al.) Free (opens in a new tab)
- Course Stanford CS336: Language Modeling from Scratch Stanford University Free (opens in a new tab)
- Course LLM Course Hugging Face Free (opens in a new tab)
- Video But what is a GPT? Visual intro to transformers 3Blue1Brown Free (opens in a new tab)
- Video Attention in transformers, step by step 3Blue1Brown Free (opens in a new tab)