Flash Attention
Flash Attention inzira yakangwara yekuverengera nhanho yekutarisisa mukati meTransformers pasina kumbonyora hofori yekutarisisa matrix kuti inonoke ndangariro.
Pfupiso
It makes long-context models far faster and more memory-efficient without changing their math.
Kudzika Kwakadzika
Kutarisisa kwakajairwa kunofananidza tokeni yega yega kune imwe tokeni, ichigadzira N-by-N mamakisi matrix inokura quadratically nehurefu hwekutevedzana. Naively, iyo matrix inonyorerwa uye kuverenga kumashure kubva kuGPU yakakwira-bandwidth memory (HBM), uye iyo shuttling - kwete kuwanda - ndiyo chaiyo bhodhoro. Flash Attention, yakaunzwa naTri Dao uye vaanoshanda navo muna 2022, inorongazve komputa kuitira kuti matrix isambochengetwa zvizere. Inoita mibvunzo, makiyi, uye kukosha mumataira madiki anokodzera nekukurumidza pa-chip SRAM, inokokorodza mhedzisiro mhedzisiro, uye inoabatanidza pamwe chete uchishandisa online inomhanya-softmax trick. Kubuda kwakafanana nemasvomhu kutarisisa zvakajairwa asi inoshandisa mutsara ndangariro uye inomhanya kakawanda nekukurumidza, kunyanya pakutevedzana kwakareba.
Technical Insight
Iyo kiyi yehunyengeri ndeyekuteedzera pamwe neyepamhepo softmax. Softmax inowanzoda mutsara wese wezvibodzwa kuti iverenge denominator yayo, asi Flash Attention inochengeta huwandu hwepamusoro uye inomhanya sezvo ichiyerera tirairi yega yega, ichidzoreredza zvakambobuda zvishoma kuti mhedzisiro yacho ive chaiyo. Nekuti zvibodzwa zvepakati zvinogara muSRAM (maodha ehukuru nekukurumidza kupfuura HBM), iyo algorithm ndeyeIO-inoziva: inoderedza ndangariro kuverenga nekunyora kwete kuita mbishi arithmetic.
Strategic Impact
Mutengo uye bhajeti
Zvisarudzo zvezvivakwa zvinotyaira kuita uye mutengo wekushandisa kwemakore.
Sarudzo dzakajeka
Dzidzo yehunyanzvi inobatsira zvikwata kusarudza murwi wakakodzera, kwete iwo mutsva chete.
Kudzora kwemhando yepamusoro
Sarudzo dzeinjiniya dziri nani dzinoderedza zviitiko zvekuvimbika mukugadzira.
Ramangwana reFlash Attention
Flash Attention yave chivakwa chekuvaka, neFlashAttention-2 neFlashAttention-3 ichidzvanya yakawedzera kubva kumaGPU matsva seH100 nekuvandudza kupatsanura kwebasa uye kushandisa nzira dzakaderera dzeFP8. Tarisira kuenderera mberi dhizaini nehardware, kusanganisa kwakasimba mukudzidziswa uye masisitimu ekufungidzira, uye akasiyana akarongedzerwa sparse, inotsvedza-hwindo, uye-yakareba-yakareba-mamiriro ekutarisa. Sezvo mahwindo emamiriro akatambanudzwa akananga kumamirioni ezviratidzo, IO-inoziva kernels seizvi zvinoramba zvakakosha kuchengetedza ndangariro nekumhanya kunoshanda.
Real-World Implementation
Kudzidzisa mhando dzemitauro mikuru seLlama uye GPT-kirasi masisitimu ane marefu mamiriro windows pamutengo wakaderera wekuyeuka.
Kushandira vabatsiri vekutaura nekukurumidza nekumhanyisa prefill nhanho uko kukurumidza kukurumidza kunotanga kuverengwa.
Kugonesa maturusi ekuongorora-gwaro anomedza mabhuku ese kana macodebase nekuita kutevedzana-kurebesa kutarisisa kugoneke paGPU imwechete.
Simba rekuona uye odhiyo Transformers uko yakakwirira-resolution yekuisa inogadzira yakareba kwazvo tokeni kutevedzana.
Njodzi & Guardrails
Kugadzirisa imwe bhenji kunogona kuvanza yakafara system kushaya simba.
Infrastructure uye mari yekugadzirisa inowanzotarisirwa pasi.
Chengetedzo uye kucherechedzwa mapundu anogona kukura sezvo masisitimu anowedzera kuoma.
Implementation Roadmap
Tsanangura latency, mhando, uye mutengo zvinangwa usati waitwa.
Benchmark pasi pechokwadi mutoro uye data mamiriro.
Chishandiso chekutarisa zvikanganiso, kudonha, uye mushandisi maitiro.
Gadzirira nzira dzekudzosera kumashure uye dzezviitiko usati wawedzera.
Ramba Uchiongorora
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 Flash 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
Gaidhi rinotevera
Attention Rollout uye Head Pruning
Mibvunzo inowanzo bvunzwa
What is Flash Attention?
Flash Attention inzira yakangwara yekuverengera nhanho yekutarisisa mukati meTransformers pasina kumbonyora hofori yekutarisisa matrix kuti inonoke ndangariro. Inoita marefu-emamiriro emhando nekukurumidza uye zvakanyanya ndangariro-inoshanda pasina kuchinja masvomhu avo.
Ndeipi bhodhoro guru rinotariswa neFlash Attention?
Flash Attention iIO-inoziva: inoderedza iyo data yakavharika pakati pekukurumidza pa-chip SRAM uye inononoka yakakwira-bandwidth ndangariro, inova ndiyo chaiyo bhodhoro pane arithmetic pachayo.
Flash Attention inodzivirira sei kuchengetedza iyo yakazara N-by-N yekutarisisa matrix?
Iyo inoisa iyo computation kuitira kuti bhuroka rega rega rikwane muSRAM inokurumidza, komputa uye kuunganidza zvikamu zvakabuda pasina kumbogadzira iyo yese matrix muHBM.
Ndeipi nzira inobvumira Flash Attention kuverengera softmax nenzira kwayo pasina kuona mutsara wese kamwechete?
Iyo yepamhepo softmax inochengeta yakakwira uye inomhanya dhinominator uku uchitenderera mataira, ichidzoreredza zvakamboitika zvishoma zvakabuda kuitira kuti kujairana kwekupedzisira kuve chaiko.
Nei Flash Attention ichibatsira zvakanyanya nemateedzana akareba?
Iyo naive yekutarisisa matrix zviyero seN-yakapetwa mundangariro, saka kudzivirira kwayo kuzere kuchengetedza kunopa iyo yakakura kuchengetedza chaizvo kana kutevedzana kwakareba.
Chii chinonzi 'IO-aware' dhizaini yeFlash Attention inotungamira kudzikisira?
IO-kuziva zvinoreva kuti algorithm yakagadzirirwa kutenderedza mutengo wekufambisa data mundangariro, ichideredza HBM traffic pane kuita arithmetic.