AI-hardware en softwarestack
Welke uitspraak over de rol van de KV-cache bij de inferentie van grote taalmodellen is juist?
- AVeelgestelde vragen en antwoorden worden opgeslagen om dezelfde vraag direct te beantwoorden
- BAlle gewichten van het model worden in het cachegeheugen van de CPU geplaatst
- 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.