AI-hardware en softwarestack

Hoe neemt de rekenlast van self-attention in een Transformer toe met de lengte n van de invoerreeks?

  1. AEvenredig met n
  2. BEvenredig met het kwadraat van n
  3. CConstant, onafhankelijk van n
Antwoord en toelichting tonen

Antwoord: B. Evenredig met het kwadraat van n

Bij self-attention wordt het inwendig product van de Query van elk token met de Key van alle tokens berekend, wat een n×n-tabel met gewichten oplevert. De rekenlast en het geheugen voor de tussentabel bij een eenvoudige implementatie zijn daarom evenredig met het kwadraat van de lengte van de reeks. Bij verdubbeling van de lengte wordt het attentiongedeelte viermaal zo groot, zodat bij lange teksten de efficiëntie van de attention de prestaties bepaalt.

Dit onderwerp oefenen Naar de onderwerppagina

Meer vragen over AI-hardware en softwarestack

  1. Welke uitspraak over het basisprincipe van snelle attention-implementaties zoals FlashAttention is juist?
  2. Wat is het belangrijkste doel van "operator fusion" (samenvoegen van bewerkingen) in AI-berekeningen?
  3. Het optellen van een bias en het toepassen van ReLU op het resultaat van een matrixproduct, eerst drie aparte kernels, zijn samengevoegd tot één kernel. Wat neemt hierdoor hoofdzakelijk af?
  4. Welke uitspraak over de belangrijkste taak van AI-compilers zoals XLA, TVM en de compilerfunctie van PyTorch is juist?
  5. Welke uitspraak over de "rekengraaf" (computational graph) in AI-frameworks en -compilers is juist?

Vergelijkingen bij dit onderwerp