Concepts / Self-Supervision in Language Models

Self-Supervision in Language Models

Demand for AI applications has increased while the barrier to entry for building them has decreased. The expensive part, training the model, has already been done by someone else, so the up-front cost of getting started has dropped.

  • Programming

A Prediction Before Reading

Imagine trying to describe the state of artificial intelligence after 2020 in a single word. A strong candidate is scale. The AI models behind widely used products such as ChatGPT, Google's Gemini, and Midjourney are large enough that they consume a meaningful share of the world's electricity, and their developers face the possibility of running short of publicly available internet data for training.

Language models became the center of this scaling effort. Machine learning covers many kinds of models, including models that detect objects in images, forecast the weather, predict stock prices, and recommend products. The reason language models stood out is a property of how they can be trained: they can learn from self-supervision, while many other model types need supervision. This article explains that property, why it matters, and where people commonly misread it.

What do you think happens?

Is self-supervision the same thing as unsupervised learning, since both sound like they work without human-made answer keys?

  • Yes, they are two names for the same approach

  • No, self-supervision uses labels, but the labels are inferred from the text itself

Reveal answer
Answer:

No, self-supervision uses labels, but the labels are inferred from the text itself

A very common first reaction is to file self-supervision under unsupervised learning. The source states that self-supervised learning uses labels that come from the text, while unsupervised learning, as described there, does not need labels at all. These are different approaches. The rest of this article shows why.

Foundation Models and Model as a Service

Definition

A foundation model is a large-scale AI model developed by a few organizations and made available for others to use as a service.

The word foundation describes the role the model plays for others. It is a base that many different applications can be built on, rather than a finished product built for a single purpose. Examples named in the source include ChatGPT, Google's Gemini, and Midjourney. The source also makes clear that these models are not static: they are becoming more powerful and capable of more tasks over time.

Training large models requires data, compute resources, and specialized talent that only a few organizations can afford. Those organizations build the models and then offer access to them. This pattern is called model as a service. Anyone who wants to create an AI-powered application can use these models without first investing in building a model of their own.

Scale changes two things at once. Models become more capable and can handle a wider range of tasks, which opens up more applications. At the same time, the effort needed to train such models is concentrated in a small number of organizations. The rest of the field has to decide how to make use of what those organizations produce.

Key point

Demand for AI applications has increased while the barrier to entry for building them has decreased. The expensive part, training the model, has already been done by someone else, so the up-front cost of getting started has dropped.

Once models are available as services, the work of an AI engineer shifts toward assembling, adapting, and deploying applications around them. The source describes AI engineering as building applications on top of readily available models. It builds on application practices that predate large language models. Before large language models became prominent, AI already powered product recommendations, fraud detection, and churn prediction. What foundation models change is the scale and availability of the models, which brings new possibilities as well as new challenges.

From Language Models to Large Language Models

Foundation models did not appear suddenly. The source traces their lineage: foundation models emerged from large language models, which in turn originated as language models. Although applications such as ChatGPT and GitHub's Copilot can look like they arrived overnight, they are the result of decades of technical progress. The first language models emerged in the 1950s.

The progression is one of growth rather than replacement. A language model that has grown to a very large scale is a large language model, and a large-scale model offered for others to build on is a foundation model. The same underlying idea is present throughout, but each stage describes a different size of model and a different way it reaches its users.

Language modelsfirst emerged in the 1950sLarge language modelslanguage models grown tovery large scaleFoundation modelslarge-scale models offeredfor others to build onSelf-supervisiontechnique that enabledgrowth to today's scalegrewoffered as a baseenabled by
In what order did language models grow in scale, and where does self-supervision enter this progression?

Language models have existed for a long time, but they have only been able to grow to the scale they are today with self-supervision. The source introduces self-supervision briefly as part of an overview of what language models and self-supervision mean. The next section explains why it has this effect.

The Data Labeling Bottleneck

In supervised learning, you show a model examples of the behavior you want and attach a label to each one. Once trained, the model can be applied to new data. A fraud detection model, for instance, learns from transactions that are each tagged as fraud or not fraud, and then predicts the tag for transactions it has never seen. The success of many AI systems in the 2010s rested on this approach.

The weakness is cost. Obtaining labels is expensive and slow, and this is the data labeling bottleneck that limits how large a supervised dataset can grow. The bottleneck gets worse as tasks become harder. Labeling everyday objects is something most people can do without training, but labeling whether a medical scan shows signs of cancer demands expert time and is far more costly.

Self-supervision removes the need for an external answer key. Instead, the model infers the labels from the input data itself. Language modeling is a clean case of this. Every input sequence contains two things at once: the tokens the model should predict, and the surrounding tokens that serve as context for predicting them. Because the text supplies both, a large body of ordinary writing can be turned into training material without anyone annotating it.

Raw textordinary writing, noannotationWrap in markers<BOS> at start, <EOS> atendSplit into samplescontext plus next-tokentargetTrain modeltargets come from the textitselfwrappedeach positionfed to
How does the training pipeline get its targets from raw text without a human annotator in the loop?

Turning One Sentence into Training Samples

Take a new sentence to see the mechanism: Cats chase mice. The sentence is first wrapped in a beginning marker and an end marker. It has four ordinary tokens: Cats, chase, mice, and the period. Each position in the sequence becomes one training sample. The learner sees the context to the left of a position and must predict the token that comes next.

Sample 1context: <BOS> target: CatsSample 2context: <BOS> Cats target:chaseSample 3context: <BOS> Cats chasetarget: miceSample 4context: <BOS> Cats chasemice target: .Sample 5context: <BOS> Cats chasemice . target: <EOS>context growscontext growscontext growscontext grows
How does a single sentence become many (context, next-token target) pairs, and how does the context grow with each step?

Notice what happened. A single sentence produced five separate training samples, and no one wrote a label for any of them. Each target was simply the token that actually followed in the text. A longer passage of text yields many more samples in the same way, which is why this approach can scale to very large collections of writing.

Worked example

Counting samples for a new sentence

Sentence: Dogs bark. How many training samples does it produce, and what is the target of the last one?

  1. 1
    Wrap the sentence

    Add <BOS> at the start and <EOS> at the end: <BOS> Dogs bark . <EOS>.

  2. 2
    Count the ordinary tokens

    Dogs, bark, and the period make three ordinary tokens.

  3. 3
    Map positions to targets

    Each ordinary token and the end marker is a target, so the targets are Dogs, bark, the period, and <EOS>.

  4. 4
    Read the final sample

    The last sample has context <BOS> Dogs bark . and target <EOS>, which teaches the model where the sequence stops.

Result

Four training samples. The last one has the context <BOS> Dogs bark . and the next-token target <EOS>.

Marking Boundaries with BOS and EOS

The markers at the edges of each sequence are special tokens. The beginning marker <BOS> shows where a sequence starts, and the end marker <EOS> shows where it stops. Each marker is typically handled by the model as one token, not as a set of characters.

<BOS>marks where the sequencestartsCatsfirst ordinary tokenchasesecond ordinary tokenmicethird ordinary token.fourth ordinary token<EOS>marks where the sequencestopssequencesequencesequencesequencesequence
Where do <BOS> and <EOS> sit within a token sequence, and what part of the text does each one mark?

These markers matter because a language model usually works with many sequences, and it needs a clear way to tell where one ends and the next begins. The end marker also teaches the model when to stop. In the sample set above, the final target is <EOS>, so the model learns that the sentence is complete after the period.

Three Ways to Learn from Data

Now return to the belief from the start. The distinction is not whether a model is trained on raw data, since both approaches can use it. The distinction is whether the learning setup relies on labels, and where those labels come from. The table compares the three approaches on that question.

ApproachUses labels?Where the targets come fromExample
Supervised learningYesHuman-supplied tags attached to each exampleFraud or not fraud tags on transactions
Self-supervised learningYes, but inferredInferred from the input data itselfPredicting the next token in a sentence
Unsupervised learning (as described in the source)NoNo labels are neededLearning without an answer key

The key test is whether the setup relies on labels, and where those labels come from.

Key point

Self-supervision is not unsupervised learning. Its labels are real targets taken from the text, which is why it avoids the labeling bottleneck while still learning from explicit next-token answers.

Mistakes Beginners Make

  • Common mistake

    Filing self-supervision under unsupervised learning because neither needs a human answer key

    Why it fails

    Self-supervised learning uses labels that come from the text, while unsupervised learning as described in the source does not need labels at all.

    Do this instead

    Ask whether the setup relies on labels, and where they come from.

  • Common mistake

    Thinking a sentence produces only one training example

    What it looks like

    Cats chase mice yields five samples, one for each position including the end marker.

    Why it fails

    Each position in the sequence becomes its own sample with a growing context.

    Do this instead

    Count the positions, including <EOS>, to find the number of samples.

  • Common mistake

    Assuming <BOS> and <EOS> are ordinary words in the text

    Why it fails

    They are special tokens that mark where a sequence begins and ends, so the model can tell one sequence from the next.

    Do this instead

    Treat them as boundary markers that wrap each sequence.

  • Common mistake

    Believing that self-supervision requires someone to annotate the training text

  • Common mistake

    Treating the labeling bottleneck as only a problem for easy tasks, when it grows harder as tasks become more expert-dependent

Practice: Build the Samples Yourself

EASY

Take the sentence Birds sing. Write out the wrapped sequence with <BOS> and <EOS>, then list each training sample as a context and a next-token target. How many samples does it produce?

Hints
  • Start by adding <BOS> at the front and <EOS> at the end.
  • Each position in the sequence, including the position of <EOS>, becomes one target.
  • The context for each sample is everything to the left of its target.

Key Takeaways

  • A foundation model is a large-scale AI model developed by a few organizations and made available as a service, so others can build applications without training their own models.
  • Language models grew into large language models and then foundation models, and self-supervision is the technique that enabled growth to today's scale.
  • Self-supervision avoids the data labeling bottleneck by inferring labels from the input text itself, so each token in a sentence becomes a target for the tokens before it.
  • A single sentence yields several training samples, with the context growing one token at a time, and <BOS> and <EOS> mark where each sequence starts and stops.
  • The key test separating the three approaches is whether labels are used and where they come from: supervised learning uses human-supplied labels, self-supervised learning infers them from the text, and unsupervised learning, as described in the source, needs no labels.

Related concepts

More concepts starting with S