FlashAttention
FlashAttention jẹ alugoridimu-daradara-iranti ti o ṣe iṣiro akiyesi deede kanna bi awọn Ayirapada boṣewa ṣugbọn laisi kikọ matrix akiyesi omiran nigbagbogbo lati fa fifalẹ iranti GPU.
Akopọ
It made long-context training and inference dramatically faster and cheaper.
Jin Dive
Ifarabalẹ deede ṣe iṣiro Dimegilio kan fun gbogbo bata ti awọn ami, ti n ṣe agbejade matrix N-by-N. Fun ọkọọkan 4,000-àmi ti o jẹ awọn ikun miliọnu 16, ati pe matrix gbọdọ wa ni kikọ si ati ka pada lati iranti bandiwidi giga ti GPU (HBM). Ijabọ iranti yẹn, kii ṣe iṣiro, jẹ igo gidi. FlashAttention, ti a ṣe nipasẹ Tri Dao ati awọn ẹlẹgbẹ ni 2022, tun ṣe iṣiro naa ki matrix naa ko ni ohun elo ni kikun. O ṣe ilana ọkọọkan ni awọn alẹmọ ti o baamu ni kekere GPU, SRAM iyara-yara lori chip, ṣiṣe iṣiro softmax ni afikun bi o ti n lọ. Abajade jẹ aami mathematiki si akiyesi boṣewa ṣugbọn o nlo iranti ti o kere pupọ ati ṣiṣe ni iyara pupọ, ti n mu awọn window ọrọ to gun to gun.
Imọ-imọ-ẹrọ
Awọn omoluabi ni 'online softmax' ni idapo pelu tiling. FlashAttention ṣe ẹru awọn bulọọki kekere ti awọn ibeere, awọn bọtini, ati awọn iye sinu SRAM, ṣe iṣiro awọn abajade ifarabalẹ apakan, ati ṣe atunṣe awọn akopọ ṣiṣe bi awọn bulọọki tuntun ti de nitoribẹẹ imudara softmax deede duro laisi ri gbogbo awọn ikun ni ẹẹkan. Nitoripe ko tọju matrix N-by-N ni kikun ni HBM, awọn iwọn iranti laini kuku ju quadratically, ati ekuro ti dapọ sinu iṣẹ GPU kan lati dinku iranti kika ati kikọ.
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 FlashAttention
FlashAttention ti di bulọọki ile aiyipada. FlashAttention-2 ṣe ilọsiwaju pipin iṣẹ GPU, ati FlashAttention-3 lo awọn ẹya ohun elo ohun elo Hopper tuntun bii asynchrony ati FP8 konge kekere. Reti ti o tẹsiwaju-apẹrẹ pẹlu awọn eerun igi, isọpọ jinle sinu awọn olupin ifokansi fun awọn iwe aṣẹ gigun, ati awọn iyatọ aifwy fun fọnka tabi akiyesi window sisun. Gẹgẹbi awọn window ti o tọ titari si awọn miliọnu awọn ami-ami, awọn kernel ti o mọ IO bii eyi jẹ pataki si titọju ikẹkọ ati ṣiṣe awọn idiyele iṣakoso.
Real-World imuse
Ikẹkọ awọn awoṣe ede nla bii Llama ati awọn ọna ṣiṣe GPT ni iyara ati ni idiyele GPU kekere
Ṣiṣẹ awọn oluranlọwọ iwiregbe ọrọ-gigun ti o njẹ gbogbo awọn iwe tabi awọn koodu koodu laisi ṣiṣiṣẹ ti iranti
Iyara awọn opo gigun ti akopọ iwe-ipamọ ti o ṣe ilana awọn mewa ti ẹgbẹẹgbẹrun awọn ami ni ẹẹkan
Iran agbara ati awọn oluyipada multimodal nibiti awọn ilana gigun ti awọn abulẹ aworan jẹ ki akiyesi gbowolori
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 FlashAttention 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
Rotari Ipo ifibọ
Awọn ibeere ti a beere nigbagbogbo
What is FlashAttention?
FlashAttention jẹ alugoridimu-daradara-iranti ti o ṣe iṣiro akiyesi deede kanna bi awọn Ayirapada boṣewa ṣugbọn laisi kikọ matrix akiyesi omiran nigbagbogbo lati fa fifalẹ iranti GPU. O ṣe ikẹkọ ọrọ-ọrọ gigun ati itọkasi ni iyara yiyara ati din owo.
Kini awọn ibi-afẹde FlashAttention akọkọ igo ni akiyesi boṣewa?
Ifarabalẹ boṣewa jẹ asopọ-iranti: tiipa matrix N-by-N nla si ati lati iranti bandiwidi giga jẹ gaba lori akoko, kii ṣe iṣiro funrararẹ.
Bawo ni iṣelọpọ FlashAttention ṣe afiwe si akiyesi boṣewa?
FlashAttention ṣe iṣiro awọn iye akiyesi gangan kanna; o kan tun ṣe iṣiro lati yago fun ohun elo matrix kikun.
Iranti GPU wo ni FlashAttention nilokulo nipasẹ sisẹ data ni awọn alẹmọ?
FlashAttention ṣe ẹru awọn alẹmọ kekere sinu GPU ti o yara lori-chip SRAM, ti o jẹ ki iṣẹ eru naa sunmọ awọn ẹya iṣiro.
Ilana wo ni o jẹ ki FlashAttention ṣe iṣiro softmax laisi ri gbogbo awọn ikun ni ẹẹkan?
Ohun elo softmax lori ayelujara n ṣetọju awọn iwọn ti o pọju ati awọn akopọ, n ṣe atunṣe awọn abajade apa kan bi awọn alẹmọ tuntun ti de nitorina deede isọdọtun ti o tọ.
Kini FlashAttention-3 ṣe pataki ni anfani?
FlashAttention-3 jẹ apẹrẹ papọ pẹlu awọn GPU Hopper ode oni, ni lilo ipaniyan asynchronous ati FP8 konge-kekere fun awọn iyara siwaju sii.