AI Blog

AI Blog

by Michele Laurelli

Top-K Sampling

Technique
Definition

Text generation technique sampling from only the K most likely next tokens.

Restricts sampling to top K tokens by probability. Reduces unlikely tokens while maintaining diversity. Common: K=40-50.

Examples

1

K=50 for balanced generation

2

Diverse text generation

3

Avoiding rare tokens

Michele Laurelli - AI Research & Engineering