Skip to content
Optimization

Draft Model

Small, fast auxiliary model used in speculative decoding to propose candidate tokens for the larger target model to verify.

Definition

A draft model is a smaller and cheaper version of the target model used in speculative decoding. It generates K speculative tokens autoregressively, which are then verified in one batch by the large target model. The draft model is often a pruned or distilled version of the target, or a model in the same family (e.g., Llama 3 8B drafting for Llama 3 70B). For the technique to be efficient, the draft must be fast enough that K draft steps plus one target step is faster than K target steps alone.

More Optimization terms