Wat is Prefill?

Eerste fase bij inferentie van grote taalmodellen waarin de tokens van de invoertekst in één keer worden verwerkt. Wordt gemakkelijk compute-bound. De daaropvolgende generatie per token heet decode.

Japanse benaming: プリフィル

“Prefill” komt voor in 2 oefenvragen uit 1 onderwerpen, bijv. AI-hardware en softwarestack.

Vakgebied: AI-hardware en softwarestack

Nu oefenen: vragen over Prefill

De inferentie van een groot taalmodel bestaat uit een fase waarin de invoertekst gezamenlijk wordt verwerkt (prefill) en een fase waarin per token de uitvoer wordt gemaakt (decode). Welke combinatie van wat de prestaties van elke fase hoofdzakelijk bepaalt is juist (bij een kleine batch)?

  1. APrefill: geheugenbandbreedte, decode: rekenprestaties
  2. BPrefill: rekenprestaties, decode: geheugenbandbreedte
  3. CBeide: de snelheid van de opslag
Antwoord en toelichting tonen

Antwoord: B. Prefill: rekenprestaties, decode: geheugenbandbreedte

Bij prefill worden honderden tot duizenden tokens van de invoer gezamenlijk berekend, zodat gewichten voor veel tokens worden hergebruikt en de bewerking een matrixproduct en rekenbegrensd wordt. Bij decode worden voor elk token alle gewichten gelezen terwijl slechts de berekening van één token wordt uitgevoerd, zodat de rekenintensiteit laag en de bewerking geheugenbegrensd is. De decodesnelheid wordt daarom ruwweg bepaald door geheugenbandbreedte ÷ grootte van de gewichten.

“Prefill” in de toelichting

Vragen met “Prefill” (2)

  1. De inferentie van een groot taalmodel bestaat uit een fase waarin de invoertekst gezamenlijk wordt verwerkt (prefill) en een fase waarin per token de uitvoer wordt gemaakt (decode). Welke combinatie van wat de prestaties van elke fase hoofdzakelijk bepaalt is juist (bij een kleine batch)?
  2. Waarom maakt W4A16-kwantisatie, waarbij alleen de gewichten 4 bits krijgen en de activaties FP16 blijven, de decode bij de inferentie van grote taalmodellen bijzonder snel?

Verwante termen

Inferentie (inference) Rekenintensiteit (arithmetic intensity) AI-compiler All-reduce Attention Backpropagatie Batchgrootte Dataparallellisme FlashAttention Gestructureerde sparsity

Naar de woordenlijst