Bird
Raised Fist0
NLPml~20 mins

Beam search decoding in NLP - Practice Problems & Coding Challenges

Choose your learning style10 modes available

Start learning this pattern below

Jump into concepts and practice - no test required

or
Recommended
Test this pattern10 questions across easy, medium, and hard to know if this pattern is strong
Challenge - 5 Problems
🎖️
Beam Search Mastery
Get all challenges correct to earn this badge!
Test your skills under time pressure!
🧠 Conceptual
intermediate
2:00remaining
What is the main purpose of beam search decoding in NLP?

Beam search is often used in sequence generation tasks like translation or text generation. What does beam search primarily help with?

AIt randomly selects sequences to generate diverse outputs without considering probabilities.
BIt finds the single most probable output sequence by exploring all possible sequences exhaustively.
CIt balances between exploring multiple candidate sequences and focusing on the most promising ones to find likely outputs efficiently.
DIt guarantees finding the globally optimal sequence by checking every possible output.
Attempts:
2 left
💡 Hint

Think about how beam search keeps track of multiple candidates but limits the number to avoid huge computation.

Predict Output
intermediate
2:00remaining
Output of beam search step with beam width 2

Given the following partial sequences and their log probabilities, what are the top 2 sequences after expanding one step?

partial_sequences = [("I am", -1.0), ("You are", -1.2)]
candidates = {
  "I am": [("happy", -0.5), ("sad", -1.5)],
  "You are": [("kind", -0.3), ("mean", -2.0)]
}
beam_width = 2

Calculate the new sequences with summed log probabilities and pick top 2.

A[('I am happy', -1.5), ('You are kind', -1.5)]
B[('I am happy', -1.5), ('You are mean', -3.2)]
C[('You are kind', -1.5), ('I am sad', -2.5)]
D[('I am sad', -2.5), ('You are mean', -3.2)]
Attempts:
2 left
💡 Hint

Add the log probabilities of partial sequences and their expansions, then pick the top 2 with highest (least negative) sums.

Model Choice
advanced
2:00remaining
Choosing beam width for a translation model

You have a neural machine translation model. You want to balance translation quality and decoding speed. Which beam width is most suitable?

ABeam width = 5 as a compromise between quality and speed.
BBeam width = 1 (greedy search) for fastest decoding but lower quality.
CBeam width = 100 for best quality but very slow decoding.
DBeam width = 0 to disable beam search and use random sampling.
Attempts:
2 left
💡 Hint

Think about typical beam widths used in practice for good quality without too much slowdown.

Metrics
advanced
2:00remaining
Effect of beam width on BLEU score and decoding time

In an experiment, increasing beam width from 1 to 10 affects BLEU score and decoding time. Which statement is true?

ABLEU score and decoding time are unaffected by beam width.
BBLEU score improves initially but plateaus or may degrade; decoding time increases roughly linearly with beam width.
CBLEU score decreases as beam width increases; decoding time decreases.
DBLEU score always increases linearly with beam width, decoding time stays constant.
Attempts:
2 left
💡 Hint

Consider how beam search explores more sequences with larger beam widths and the tradeoff involved.

🔧 Debug
expert
2:00remaining
Why does beam search sometimes produce repetitive outputs?

A sequence generation model using beam search often outputs repetitive phrases like 'the the the'. What is the most likely cause?

ABeam search always prevents repetition, so this must be a bug in the code.
BBeam search is not exploring enough sequences due to too small beam width.
CThe input data is corrupted, causing the model to repeat tokens.
DThe model's probability distribution is biased towards repeating tokens, and beam search amplifies this by focusing on high-probability sequences.
Attempts:
2 left
💡 Hint

Think about how beam search picks sequences with highest probabilities and how model biases affect output.

Practice

(1/5)
1. What is the main purpose of beam search decoding in natural language processing?
easy
A. To keep track of multiple best candidate sequences during prediction
B. To randomly select words for output generation
C. To generate only one possible output sequence
D. To speed up training by skipping steps

Solution

  1. Step 1: Understand beam search goal

    Beam search keeps multiple candidate sequences to explore more options than greedy search.
  2. Step 2: Compare options

    Only To keep track of multiple best candidate sequences during prediction describes keeping multiple best guesses; others describe random choice, single output, or unrelated speed-up.
  3. Final Answer:

    To keep track of multiple best candidate sequences during prediction -> Option A
  4. Quick Check:

    Beam search = multiple best sequences [OK]
Hint: Beam search tracks several top guesses, not just one [OK]
Common Mistakes:
  • Confusing beam search with random sampling
  • Thinking beam search outputs only one sequence
  • Assuming beam search speeds up training
2. Which of the following is the correct way to describe the beam width in beam search decoding?
easy
A. The size of the vocabulary used for prediction
B. The number of candidate sequences kept at each decoding step
C. The length of the output sequence generated
D. The number of layers in the neural network

Solution

  1. Step 1: Define beam width

    Beam width is how many top sequences the algorithm keeps at each step to explore.
  2. Step 2: Eliminate incorrect options

    Output length, vocabulary size, and network layers are unrelated to beam width.
  3. Final Answer:

    The number of candidate sequences kept at each decoding step -> Option B
  4. Quick Check:

    Beam width = candidate count per step [OK]
Hint: Beam width = how many sequences you keep each step [OK]
Common Mistakes:
  • Mixing beam width with output length
  • Confusing beam width with vocabulary size
  • Thinking beam width relates to model architecture
3. Consider a beam search with beam width 2 decoding a sequence. At step 1, the top 2 tokens have scores [0.6, 0.4]. At step 2, each token expands to two tokens with scores: from first token [0.5, 0.3], from second token [0.7, 0.2]. Which two sequences will beam search keep after step 2?
medium
A. [First token + second expansion (0.6*0.3), Second token + second expansion (0.4*0.2)]
B. [First token + first expansion (0.6*0.5), First token + second expansion (0.6*0.3)]
C. [Second token + first expansion (0.4*0.7), Second token + second expansion (0.4*0.2)]
D. [First token + first expansion (0.6*0.5), Second token + first expansion (0.4*0.7)]

Solution

  1. Step 1: Calculate scores for all expansions

    Calculate combined scores: 0.6*0.5=0.3, 0.6*0.3=0.18, 0.4*0.7=0.28, 0.4*0.2=0.08.
  2. Step 2: Select top 2 sequences by score

    Top two scores are 0.3 and 0.28, corresponding to first token + first expansion and second token + first expansion.
  3. Final Answer:

    [First token + first expansion (0.6*0.5), Second token + first expansion (0.4*0.7)] -> Option D
  4. Quick Check:

    Top scores = 0.3 and 0.28 [OK]
Hint: Multiply scores, pick top beam width sequences [OK]
Common Mistakes:
  • Choosing expansions only from one token
  • Not multiplying scores correctly
  • Picking lower scoring sequences
4. You implemented beam search decoding but notice it always returns the same output sequence regardless of input. What is the most likely bug?
medium
A. The vocabulary size is too large
B. The model is not trained
C. Beam width is set to 1, making it greedy search
D. The beam search is not normalizing scores

Solution

  1. Step 1: Analyze symptom of identical outputs

    Always same output suggests no exploration of multiple sequences.
  2. Step 2: Identify beam width effect

    If beam width = 1, beam search reduces to greedy search, always picking highest scoring token only.
  3. Final Answer:

    Beam width is set to 1, making it greedy search -> Option C
  4. Quick Check:

    Beam width 1 = greedy search [OK]
Hint: Check beam width; 1 means no beam search [OK]
Common Mistakes:
  • Blaming vocabulary size for output sameness
  • Ignoring beam width setting
  • Assuming model training causes identical outputs
5. In a machine translation task, you want to balance output quality and decoding speed. You have a beam search decoder with beam width 5. What happens if you increase the beam width to 20?
hard
A. Output quality may improve but decoding will be slower
B. Output quality will decrease and decoding will be faster
C. Output quality and decoding speed remain the same
D. Decoding speed improves but output quality is unpredictable

Solution

  1. Step 1: Understand beam width effect on quality

    Larger beam width explores more sequences, often improving output quality.
  2. Step 2: Understand beam width effect on speed

    More sequences to track means more computation, slowing decoding speed.
  3. Final Answer:

    Output quality may improve but decoding will be slower -> Option A
  4. Quick Check:

    Higher beam width = better quality, slower speed [OK]
Hint: Bigger beam = better results but slower decoding [OK]
Common Mistakes:
  • Assuming bigger beam always speeds decoding
  • Thinking quality decreases with bigger beam
  • Believing beam width doesn't affect speed