Iṣọkan-Ibeere Ifojusi
Ifarabalẹ-Ibeere akojọpọ (GQA) jẹ ọna lati dinku iranti ti o nilo lakoko iran ọrọ nipa jijẹ ki ọpọlọpọ awọn ori ibeere pin bọtini kanna ati awọn ori iye.
Akopọ
It makes large models much faster to serve with almost no quality loss.
Jin Dive
Ni ipele ifarabalẹ olona-ori boṣewa, gbogbo ori ni awọn ibeere tirẹ, awọn bọtini, ati awọn iye. Lakoko iran, awọn bọtini ati iye fun gbogbo awọn ami ami iṣaaju ti wa ni cache (‘kaṣe KV’) nitorinaa awoṣe ko ṣe tun ṣe iṣiro wọn. Pẹlu ọpọlọpọ awọn olori ati awọn aaye gigun, kaṣe yii di nla ati jẹ gaba lori bandiwidi iranti ni akoko itọkasi. GQA, ti a ṣe nipasẹ Google awọn oniwadi ni ọdun 2023, ṣe akojọpọ awọn ori ibeere ati fun ẹgbẹ kọọkan ni akojọpọ bọtini kan ṣoṣo ati awọn ori iye. Ti o ba ni awọn ori ibeere 32 ṣugbọn awọn ẹgbẹ KV 8 nikan, kaṣe KV dinku ni aijọju ilọpo mẹrin. Eyi joko laarin ifarabalẹ olona-ori kikun (gbogbo ori lọtọ) ati akiyesi ibeere pupọ (ọkan ti a pin KV fun gbogbo awọn ori), yiya pupọ julọ iyara ti MQA lakoko ti o tọju didara to sunmọ akiyesi kikun. Llama 2 70B ati ọpọlọpọ awọn awoṣe nigbamii gba o.
Imọ-imọ-ẹrọ
Didara akiyesi gbarale pupọ lori nini ọpọlọpọ awọn itọnisọna ibeere pato, ṣugbọn o fi aaye gba pinpin awọn bọtini ati iye. GQA nilo asymmetry yii: o tọju gbogbo awọn ori ibeere ṣugbọn ṣe atunṣe ori KV kọọkan ti o pin kọja awọn ibeere inu ẹgbẹ rẹ. Awọn ifowopamọ wa ni itọkasi, nibiti kaṣe KV jẹ olumulo akọkọ ti bandiwidi iranti; Awọn ori KV diẹ tumọ si data ti o dinku lati ka fun ami ti ipilẹṣẹ. Awọn awoṣe nigbagbogbo jẹ 'imudaniloju' ni ṣoki lati ṣe iyipada aaye ayẹwo ori-ọpọlọpọ ti o wa tẹlẹ sinu ọkan GQA kan.
Ipa Ilana
Iyara ati iwọn
Ṣiṣan iṣẹ ede le gbe ni iyara laisi irubọ aitasera.
Wiwọle ati arọwọto
O faagun iraye si kọja awọn ede ati awọn aza ibaraẹnisọrọ.
Awọn ipinnu diẹ sii
Awọn ẹgbẹ le lo akoko diẹ sii lori idajọ lakoko ti adaṣe n kapa atunwi.
Ojo iwaju ti Ifarabalẹ-Ibeere Ẹgbẹ
GQA jẹ aiyipada boṣewa ni bayi ni awọn awoṣe iwuwo-ìmọ nitori pe o ṣe iṣowo ni mimọ idiyele idiyele didara kekere fun awọn iṣẹgun iṣẹ nla. Reti o lati darapo pọ pẹlu awọn ẹtan ṣiṣe miiran bii FlashAttention, KV-cache quantization, ati awọn ero tuntun gẹgẹbi akiyesi wiwakọ ori-pupọ ti o fun kaṣe pọsi paapaa siwaju. Bi awọn window ti o tọ ti ndagba, iṣakoso iwọn KV-cache yoo duro ni iṣoro apẹrẹ aarin, ati pinpin ori ara-GQA yoo wa lefa bọtini.
Real-World imuse
Llama 2 70B ati Llama 3 ni lilo GQA lati ṣe iranṣẹ awọn aaye gigun pẹlu kaṣe KV kekere kan
Idinku iranti GPU nitori awoṣe iwiregbe nla kan baamu lori awọn iyara iyara tabi din owo
Iyara soke tokini-nipasẹ-àmi iran ni gbóògì APIs ibi ti KV-cache bandiwidi ni awọn bottleneck
Mu awọn iwọn ipele ti o tobi julọ ṣiṣẹ fun sìn ọpọlọpọ awọn olumulo nigbakanna laisi iranti arẹwẹsi
Awọn ewu & Awọn ọna iṣọ
Awọn otitọ ti a sọ di mimọ le tẹ awọn ijabọ sii ni idakẹjẹ, awọn ṣiṣan atilẹyin, tabi awọn abajade iwadii.
Ifamọ kiakia le ṣẹda awọn abajade aisedede kọja awọn ibeere ti o jọra.
Awọn data ọrọ ifarabalẹ le farahan ti awọn idari wiwọle ko lagbara.
Ilana Ilana imuse
Ṣetumo ọna kika iṣẹjade, ohun orin, ati awọn iṣedede didara ṣaaju ṣiṣejade.
Awọn idahun ilẹ pẹlu awọn orisun ti o gbẹkẹle nigbakugba ti deede ba ṣe pataki.
Jeki aaye ayẹwo atunyẹwo eniyan fun awọn abajade ti o ga julọ.
Tọpinpin awọn ilana ikuna ati tunṣe awọn itọsi tabi ṣiṣan iṣẹ nigbagbogbo.
Tesiwaju Ṣiṣawari
Free newsletter
Get the daily AI briefing
Three verified AI stories every weekday morning, written in plain English. Free forever, no ads.
One email each weekday. Unsubscribe in one click. We never sell or share your address.
Test yourself
Take the Grouped-Query Attention quiz
Instant feedback on every answer, and a shareable certificate with a verifiable ID once you pass a course.
Support free AI education. AI Understanding is a 501(c)(3) nonprofit — no ads, no paywall, ever. Make a donation
Itọsọna atẹle
Olona-Ibeere akiyesi
Awọn ibeere ti a beere nigbagbogbo
What is Grouped-Query Attention?
Ifarabalẹ-Ibeere akojọpọ (GQA) jẹ ọna lati dinku iranti ti o nilo lakoko iran ọrọ nipa jijẹ ki ọpọlọpọ awọn ori ibeere pin bọtini kanna ati awọn ori iye. O ṣe awọn awoṣe nla ni iyara pupọ lati sin pẹlu fere ko si pipadanu didara.
Ninu Ifarabalẹ-Ibeere ti Akojọpọ, kini a pin laarin ẹgbẹ kan ti awọn olori ibeere?
GQA tọju awọn ori ibeere lọtọ ṣugbọn jẹ ki ẹgbẹ kọọkan wọn pin ipin kan ti bọtini ati awọn ori iye, idinku kaṣe KV.
GQA ti wa ni ti o dara ju apejuwe bi a aarin laarin eyi ti meji extremes?
Olona-ori yoo fun gbogbo ori KV tirẹ; olona-ibeere pin ọkan KV fun gbogbo awọn olori; GQA joko laarin pẹlu awọn ẹgbẹ KV diẹ.
Kini apakan ti itọkasi GQA ni akọkọ jẹ din owo?
Awọn ifowopamọ ṣe afihan ni akoko iran, nibiti kika kaṣe KV jẹ idiyele iranti-bandwidth ti o ga julọ.
Ti awoṣe ba ni awọn ori ibeere 32 ati awọn ẹgbẹ KV 8, kaṣe KV dinku nipasẹ aijọju kini ifosiwewe?
Awọn ori ibeere 32 ti o pin nipasẹ awọn ẹgbẹ KV 8 ti o pin yoo funni ni aijọju idinku mẹrin ni awọn bọtini ipamọ ati awọn iye.
Kini idi ti GQA le ṣe itọju pupọ julọ didara awoṣe laibikita pinpin awọn ori KV?
Ifarabalẹ fi aaye gba awọn bọtini pinpin ati awọn iye dara julọ ju ti o fi aaye gba sisọnu awọn itọnisọna ibeere ọtọtọ, nitorinaa GQA jẹ ki didara ga ga.