Wat is KV-cache?

Geheugen dat bij inferentie van grote taalmodellen de Key- en Value-waarden van de attention van al verwerkte tokens bewaart. Voorkomt herberekening, maar groeit evenredig met de lengte van de tekst en de batch.

Japanse benaming: KVキャッシュ

“KV-cache” komt voor in 4 oefenvragen uit 1 onderwerpen, bijv. AI-hardware en softwarestack.

Vakgebied: AI-hardware en softwarestack

Nu oefenen: vragen over KV-cache

Welke uitspraak over de rol van de KV-cache bij de inferentie van grote taalmodellen is juist?

  1. AVeelgestelde vragen en antwoorden worden opgeslagen om dezelfde vraag direct te beantwoorden
  2. BAlle gewichten van het model worden in het cachegeheugen van de CPU geplaatst
  3. CDe Key en Value van de attention van reeds verwerkte tokens worden opgeslagen, zodat ze niet bij elk nieuw token opnieuw hoeven te worden berekend
Antwoord en toelichting tonen

Antwoord: C. De Key en Value van de attention van reeds verwerkte tokens worden opgeslagen, zodat ze niet bij elk nieuw token opnieuw hoeven te worden berekend

Bij de attentionberekening verwijst een nieuw token naar de Key en Value van alle voorgaande tokens. Omdat de Key en Value van eerdere tokens niet veranderen, hoeft alleen het nieuwe token te worden berekend als ze worden opgeslagen in plaats van telkens opnieuw berekend. Dit is de KV-cache. De berekening neemt sterk af, maar de hoeveelheid opgeslagen Key en Value groeit evenredig met de lengte van de tekst en de grootte van de batch.

“KV-cache” in de toelichting

Vragen met “KV-cache” (4)

  1. Welke uitspraak over de rol van de KV-cache bij de inferentie van grote taalmodellen is juist?
  2. Hoeveel bedraagt het geheugen voor de gewichten ongeveer wanneer de gewichten van een model met 7 miljard (7B) parameters in FP16 (2 bytes) worden bijgehouden?
  3. Welke uitspraak over het nadeel van een te grote batchgrootte bij inferentie is juist?
  4. Welke uitspraak over wat bij de inferentie van grote taalmodellen gemakkelijk een probleem wordt bij lange teksten of grote batches is juist?

Verwante termen

Inferentie (inference) Batchgrootte GPU Doorvoer (throughput) AI-compiler All-reduce Attention Backpropagatie Dataparallellisme FlashAttention

Naar de woordenlijst