Google

Causal language modeling, also known as standard language modeling, is an auto-regressive pre-training approach where tokens are sequentially predicted following their natural, fixed order in the text (typically left-to-right). For instance, a sequence of $${}5$$ tokens $$x_0 x_1 x_2 x_3 x_4$$ is generated in the order $$x_0 \to x_1 \to x_2 \to x_3 \to x_4$$. The overall sequence probability $$\Pr(\mathbf{x})$$ is the product of individual token probabilities conditioned on preceding tokens: $$\Pr(x_0) \cdot \Pr(x_1|x_0) \cdot \Pr(x_2|x_0,x_1) \cdot \Pr(x_3|x_0,x_1,x_2) \cdot \Pr(x_4|x_0,x_1,x_2,x_3)$$. By substituting $$\mathbf{e}_i$$ as the embedding for token $$x_i$$ (a combination of its token and positional embeddings), the generation process is modeled as: $$\Pr(x_0) \cdot \Pr(x_1|\mathbf{e}_0) \cdot \Pr(x_2|\mathbf{e}_0,\mathbf{e}_1) \cdot \Pr(x_3|\mathbf{e}_0,\mathbf{e}_1, \mathbf{e}_2) \cdot \Pr(x_4|\mathbf{e}_0, \mathbf{e}_1, \mathbf{e}_2, \mathbf{e}_3)$$. This demonstrates that each prediction depends solely on past context.

Causal Language Modeling

This schematic illustrates the sequential probability calculation in Causal Language Modeling, a type of auto-regressive model. For a sequence $$x_0, x_1, \dots, x_4$$, the model predicts each token based on the embeddings of the tokens that came before it. The process begins by setting the probability of the first token, $$\Pr(x_0)$$, to 1. Each subsequent token's probability is then conditioned on the embeddings of all prior tokens, as shown in the diagram below. This unidirectional, step-by-step dependency is a core feature of causal language models.

```text
Token:       x0       x1        x2            x3                x4
             ↓        ↓         ↓             ↓                 ↓
Probability: Pr(x0)=1 Pr(x1|e0) Pr(x2|e0, e1) Pr(x3|e0, e1, e2) Pr(x4|e0, e1, e2, e3)
```

Schematic of Probability Calculation in Causal Language Modeling

A researcher designs a language model with a specific objective: to fill in a blank word in a sentence. For example, given the input 'The quick brown ___ jumps over the lazy dog', the model must predict 'fox'. To do this, the model's architecture allows it to consider the context from both the left ('The quick brown') and the right ('jumps over the lazy dog') simultaneously when making its prediction for the blank word. Which statement accurately classifies this model?

An engineer is designing a language model for a real-time chatbot that must generate responses one word at a time as a user is typing. When the model is predicting the next word in its response, what is the fundamental limitation on the contextual information it can use, and why is this limitation critical for this specific application?

Information Flow in Language Models

Your team is building an internal model that must ...

Your team is pre-training a text model for an inte...

Your team is pre-training an internal LLM for a co...

Your team is pre-training an internal LLM to suppo...

You lead an internal LLM initiative for a company with two high-value use cases: (1) a customer-support assistant that must generate policy-compliant, fluent responses (long-form text generation), and (2) an enterprise search/reranking system that must accurately judge whether two pieces of text belong together (e.g., a ticket description and a proposed resolution, or a question and a candidate answer). You have a fixed pre-training budget and must choose ONE primary pre-training objective and optionally ONE auxiliary objective from the following families: masked token reconstruction (masked language modeling), sentence-pair coherence classification (next sentence prediction), left-to-right next-token prediction (causal language modeling), reconstructing clean text from a corrupted input using an encoder-decoder denoising objective, and predicting tokens under a random permutation order (permuted language modeling).

Write a recommendation memo that (a) selects your primary objective and optional auxiliary objective, (b) explains how the information flow/conditioning structure of your chosen objectives will shape what the model learns for BOTH use cases, and (c) explicitly discusses at least two tradeoffs you are accepting (e.g., bidirectional understanding vs generation behavior, exposure to [MASK]/corruption artifacts vs realistic inference-time inputs, learning inter-sentence relations vs relying on superficial cues, or permutation-based context vs strict left-to-right factorization). Your answer should make clear why your chosen combination is better than at least one plausible alternative combination for this company scenario.

Selecting a Pre-training Objective Mix for a Corporate LLM

You lead an internal team building an LLM-based assistant for a regulated enterprise. After pre-training and light fine-tuning, you observe three recurring failures in pilot use:

1) In long-form drafting, the model often contradicts itself across paragraphs and sometimes “jumps” to content that would only make sense if it had seen later text.
2) In document review, the model is strong at filling in missing words inside a sentence, but it is unreliable at deciding whether a proposed follow-up sentence is a coherent continuation of the previous one (e.g., it misses topic shifts and non sequiturs).
3) In reconstruction-style tasks (e.g., restoring a partially corrupted policy excerpt), the model sometimes produces fluent text that does not faithfully match the original wording, even when the corruption is minor.

Assume you can choose among these pre-training objectives (alone or in combination): masked token prediction, next-sentence relationship classification, left-to-right next-token prediction, denoising reconstruction from corrupted inputs using an encoder-decoder, and permuted-order token prediction.

Write a recommendation memo that (a) identifies the most likely objective-level causes of the three failures, and (b) proposes a revised objective mix (and why) that would reduce all three issues. Your answer must explicitly connect each proposed change to how information is (or is not) allowed to flow during training (bidirectional context vs strictly past context vs permuted context) and to the difference between predicting a few tokens vs reconstructing an entire sequence vs classifying sentence-pair coherence.

Diagnosing Pre-training Objective Mismatch from Product Failures

You lead an internal team pre-training a language model for two production uses: (1) a contract-review assistant that must accurately fill in missing clauses inside long documents (insertion and span infilling), and (2) a customer-support agent that must generate responses token-by-token with low latency. Your training data is a mix of clean policy documents and noisy OCR’d PDFs where words are frequently dropped, duplicated, or out of order. Compute budget allows only ONE primary pre-training objective (you may add at most one lightweight auxiliary loss).

Write a recommendation memo that:
- Selects the primary objective from: masked language modeling, causal language modeling, denoising autoencoder reconstruction, or permuted language modeling.
- Justifies the choice by explicitly analyzing how the objective’s information flow (bidirectional vs left-to-right vs permuted) interacts with (a) the OCR noise pattern and (b) the two deployment requirements (infilling vs streaming generation).
- Proposes one auxiliary objective (either next sentence prediction or none) and argues for/against it, including at least one concrete risk of relying on superficial cues.

Your answer should make clear tradeoffs (what you gain and what you give up) and explain why your chosen combination is the best fit for this scenario.

Choosing a Pre-training Objective Under Data Constraints and Deployment Needs

You lead an internal team building an enterprise writing assistant that must (1) generate long, coherent policy drafts from a short prompt, (2) accurately fill in missing clauses inside existing documents during redlining, and (3) decide whether a proposed paragraph logically follows the previous paragraph in a policy (to flag non-sequiturs). You have budget to pre-train ONE base model from scratch and can choose ONE primary training objective; you may optionally add ONE auxiliary objective if you can justify why it complements the primary objective without undermining it. The candidate objectives are: causal language modeling, masked language modeling, denoising autoencoder reconstruction (corrupt input then reconstruct), permuted language modeling, and next sentence prediction.

Case constraints: Your training corpus is mostly internal policies and emails; at inference time the assistant must support both free-form generation and in-place editing. Early prototypes show two failure modes you must address: (A) generated drafts are locally fluent but drift off-topic after ~2 pages, and (B) in-place clause completion is often grammatically correct but contradicts a constraint stated later in the same paragraph.

Which objective would you choose as the PRIMARY objective, and which (if any) would you add as an AUXILIARY objective? In your answer, explicitly connect your choice to how information is (or is not) available to the model during training (bidirectional vs. left-to-right vs. permuted/denoising), and explain how your choice mitigates BOTH failure modes (A) and (B) while still supporting requirement (3) about paragraph-to-paragraph coherence.

Pre-training Objective Choice for a Multi-Modal Enterprise Writing Assistant

You lead an internal team pre-training an LLM for a corporate knowledge assistant that must (1) generate long, coherent incident summaries and (2) support sentence-pair tasks such as “Does this policy clause entail that control requirement?” During a red-team exercise, the model shows two issues: (a) when asked to fill a missing word in the middle of a sentence, it often produces plausible but inconsistent completions, and (b) when generating multi-paragraph summaries, it sometimes contradicts earlier sentences. Your training pipeline used an encoder-only Transformer with a masked-token objective plus a binary classifier trained to predict whether Sentence B follows Sentence A; you did not train any left-to-right next-token objective. A colleague proposes switching to an encoder-decoder denoising objective (reconstruct clean text from corrupted text) and adding a permuted prediction objective; another colleague proposes replacing everything with a pure causal (left-to-right) language modeling objective.

As the decision-maker, which proposal (or combination) would you choose to address BOTH issues with the least mismatch to the two product requirements, and why? In your answer, explicitly connect how the information available at training time (bidirectional masking vs sentence-pair classification vs left-to-right generation vs reconstruction-from-noise vs permuted prediction order) would causally affect (a) mid-sentence infilling quality and (b) long-form generation consistency.

Root-Cause Analysis of Pre-training Objective Leakage and Coherence Failures

You lead model strategy for an internal enterprise assistant used by legal and compliance teams. The assistant must (1) generate long, coherent policy drafts and email responses, (2) answer questions that require understanding relationships across two adjacent paragraphs (e.g., “Does the exception in paragraph 2 apply to the rule in paragraph 1?”), and (3) be robust to noisy inputs from OCR and copy/paste (missing words, duplicated phrases, and occasional sentence reordering). You have budget to pre-train ONE base model from scratch and can choose ONE primary pre-training objective from the following families: causal language modeling (left-to-right next-token prediction), masked language modeling (predict masked tokens using both left and right context), denoising autoencoder reconstruction (reconstruct clean text from a corrupted version), permuted language modeling (predict tokens in a random permutation order), and optionally add Next Sentence Prediction (NSP) as an auxiliary loss.

Case study task: Choose the single primary objective you would use and decide whether you would include NSP as an auxiliary loss. Justify your choices by explicitly explaining the tradeoffs among (a) generation quality for long outputs, (b) bidirectional understanding within a span, (c) modeling cross-sentence/paragraph relationships, and (d) robustness to the specified noise patterns. Your answer should make clear why at least two of the non-chosen objectives are less suitable given these constraints.

Selecting a Pre-training Objective for a Regulated Enterprise Assistant

In causal language modeling, tokens are sequentially generated following their natural text order, without any initial source-side context. For example, to generate a sequence autonomously, the model produces the output tokens one by one: `The`$$^1$$ `kitten`$$^2$$ `is`$$^3$$ `chasing`$$^4$$ `the`$$^5$$ `ball`$$^6$$ `.`$$^7$$. The superscripts indicate the strict left-to-right autoregressive generation order on the target side.

Example of Causal Language Modeling

An auto-regressive language model is designed to calculate the probability of a sequence of tokens. A key characteristic of this model is that the probability of any given token is conditioned *only* on the tokens that appeared before it. Given the sequence `token_A, token_B, token_C, token_D`, which expression correctly represents the calculation for the probability of `token_C`?

Learn Before

Related