NastępnyNastępny poradnik
Uwaga dotycząca zapytania grupowego
Językowa sztuczna inteligencja
PRZEWODNIK Językowy AI
Uwaga wielozapytaniowa (MQA) to oszczędzająca pamięć odmiana uwagi transformatora, która udostępnia jeden zestaw kluczy i wartości wszystkim głowicom uwagi.
Posłuchaj tego przewodnikaOpowiadane przez sztuczną inteligencję · ~2 min
Znacząco przyspiesza generowanie tekstu poprzez zmniejszenie pamięci, którą musi przetasować model.
Standardowa uwaga wielogłowa zapewnia każdej głowie własne prognozy zapytań, kluczy i wartości. Podczas generowania klucze i wartości wszystkich poprzednich tokenów muszą być buforowane i ponownie ładowane na każdym kroku — pamięć podręczna KV staje się głównym wąskim gardłem, ponieważ odczytanie jej z pamięci jest wolniejsze niż sama matematyka. Funkcja Multi-Query Attention, zaproponowana przez Noama Shazeera w 2019 r., utrzymuje oddzielne projekcje zapytań na głowę, ale zwija klucze i wartości do jednej wspólnej głowy. Zmniejsza to pamięć podręczną KV o współczynnik równy liczbie głowic, czasami od 8 do 64 razy mniejszy. Rezultatem jest znacznie szybsze dekodowanie autoregresyjne i mniejsze zużycie pamięci, przy jedynie niewielkim spadku jakości. Rozwiązanie pośrednie, uwaga na zapytania grupowe, równoważy kompromis.
Przyspieszenie generowania tokenów tokenów w asystentach czatu, gdzie pamięć podręczna KV, a nie surowe obliczenia, ogranicza przepustowość.
PaLM firmy Google, który wykorzystuje funkcję Multi-Query Attention w celu umożliwienia wydajnego wnioskowania na dużą skalę.
Obsługa wielu jednoczesnych użytkowników na jednym procesorze graficznym poprzez zmniejszenie pamięci podręcznej KV na żądanie.
Grouped-Query Attention w Llama 2 70B i Llama 3, bezpośredni następca równoważący prędkość MQA z jakością pełnej uwagi.
MQA ustaliło, że można przyciąć zbędne głowy klucz/wartość przy niewielkich szkodach, a ta wiedza kształtuje obecnie prawie każdy LLM z szybkim wnioskowaniem. Dziedzina ta w dużej mierze skupia się na funkcji Grouped-Query Attention (GQA), używanej w Lamie 2/3 i wielu innych, która wykorzystuje kilka grup KV zamiast jednej w celu odzyskania jakości przy zachowaniu większości przyspieszenia. Przyszłe prace łączą te pomysły z kompresją pamięci podręcznej KV, kwantyzacją i wieloutajoną uwagą, aby wydłużyć konteksty i obniżyć koszty obsługi.
Darmowy newsletter
Trzy zweryfikowane historie AI każdego poranka w dni robocze, napisane prostym angielskim. Bezpłatne na zawsze, bez reklam.
Jedno e‑mail każdego dnia roboczego. Wypisz się jednym kliknięciem. Nigdy nie sprzedajemy ani nie udostępniamy Twojego adresu.
Sprawdź się
Natychmiastowa informacja zwrotna na każde pytanie oraz udostępnialny certyfikat z weryfikowalnym ID po zaliczeniu kursu.
Wspieraj darmową edukację AI. AI Understanding jest organizacją non-profit 501(c)(3) — bez reklam, bez płatnej ściany, zawsze. Przekaż darowiznę
Uwaga wielozapytaniowa (MQA) to oszczędzająca pamięć odmiana uwagi transformatora, która udostępnia jeden zestaw kluczy i wartości wszystkim głowicom uwagi. Znacząco przyspiesza generowanie tekstu poprzez zmniejszenie pamięci, którą musi przetasować model.
MQA utrzymuje oddzielne zapytania na głowę, ale dzieli jedną projekcję klucza i jedną projekcję wartości dla wszystkich głów.
Każdy krok ponownego ładowania dużej pamięci podręcznej KV jest ograniczony przepustowością; MQA zmniejsza pamięć podręczną, aby przyspieszyć dekodowanie.
Ponieważ klucze i wartości zbiegają się z głów H do jednej, pamięć podręczna zmniejsza się w przybliżeniu o liczbę głów.
Udostępnianie kluczy i wartości nieznacznie zmniejsza możliwości reprezentacji, powodując niewielki spadek jakości w zamian za duży wzrost szybkości.
Funkcja Grouped-Query Attention (GQA) wykorzystuje kilka grup KV zamiast jednej, równoważąc jakość i szybkość.
Ucz się dalej
Wybrano więcej przewodników na ten temat
NastępnyNastępny poradnik
Uwaga dotycząca zapytania grupowego
Językowa sztuczna inteligencja