Overview
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.
Flash Attention inzvimbo yekuvaka yehunyanzvi inobata mhando yemhando, mutengo wezvivakwa, latency, uye kuvimbika pachiyero.
Deep Dive
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.
Mastering 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. Flash Attention inzvimbo yekuvaka yehunyanzvi inobata mhando yemhando, mutengo wezvivakwa, latency, uye kuvimbika pachiyero. Kuvaka kunzwisisa kwakadzama, bata Flash Attention semuenzaniso wekushandisa, kwete chinhu chimwe chete: tsanangura zvinodikanwa, kujekesa fungidziro, uye patsanura izvo zvingaitwe nehurongwa hwakavimbika kubva kune zvichiri kuda kutonga kwenyanzvi.
Mukuita, zvikwata zvakasimba zvinoshandisa Flash Attention inokwenenzvera zvivakwa, data, uye sarudzo dzezvivakwa zvinopesana nekuvimbika uye mutengo. Ivo vanonyora zvakajeka maitiro ebudiriro, bvunzo vachipokana ne data rechokwadi uye mafambiro ebasa, uye iterate zvichibva pane zvakacherechedzwa maitiro ekutadza kwete kuhwina-nguva imwe chete yebhenji. Apa ndipo apo kunzwisisa kwe theoretical kunoshanduka kuve kugona kwakasimba pane chigadzirwa, mutemo, uye mashandiro.
Zvisarudzo zvezvivakwa zvinotyaira kuita uye mutengo wekushandisa kwemakore. Panguva imwecheteyo, Kukwirisa imwe bhenji kunogona kuvanza yakafara system kushaya simba. Nzira yakatsiga ndeyekubatanidza kukurumidza kuyedza nekutonga: mhanyisa vatyairi vendege, tora humbowo, buritsa matanda esarudzo, uye urambe uchivandudza chengetedzo semaitiro emuenzaniso, zvinotarisirwa nemushandisi, uye zvinodikanwa zvekutonga.
Strategic Impact
Zvisarudzo zvezvivakwa zvinotyaira kuita uye mutengo wekushandisa kwemakore.
Zvisarudzo zvezvivakwa zvinotyaira kuita uye mutengo wekushandisa kwemakore. Mukutumirwa kwemhando yepamusoro, izvi zvinoshandurirwa kuita mitemo inoyerwa yekushanda, miganhu yevaridzi, uye tsika dzekudzokorora dzinodzokororwa kuitira kuti zvikwata zvikwire kuvimba pane kukwidza kusajeka.
Dzidzo yehunyanzvi inobatsira zvikwata kusarudza murwi wakakodzera, kwete iwo mutsva chete.
Dzidzo yehunyanzvi inobatsira zvikwata kusarudza murwi wakakodzera, kwete iwo mutsva chete. Mukutumirwa kwemhando yepamusoro, izvi zvinoshandurirwa kuita mitemo inoyerwa yekushanda, miganhu yevaridzi, uye tsika dzekudzokorora dzinodzokororwa kuitira kuti zvikwata zvikwire kuvimba pane kukwidza kusajeka.
Sarudzo dzeinjiniya dziri nani dzinoderedza zviitiko zvekuvimbika mukugadzira.
Sarudzo dzeinjiniya dziri nani dzinoderedza zviitiko zvekuvimbika mukugadzira. Mukutumirwa kwemhando yepamusoro, izvi zvinoshandurirwa kuita mitemo inoyerwa yekushanda, miganhu yevaridzi, uye tsika dzekudzokorora dzinodzokororwa kuitira kuti zvikwata zvikwire kuvimba pane kukwidza kusajeka.
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.
Maitiro Ekuita
Flash Attention mukuita
Kudzidzisa mhando dzemitauro mikuru seLlama uye GPT-kirasi masisitimu ane marefu mamiriro windows pamutengo wakaderera wekuyeuka.
Kudzidzira mamodheru emitauro mikuru seLlama neGPT-kirasi masisitimu ane akareba mahwindo pamutengo wakaderera wekuyeuka Matimu anowanzo kuwana mhedzisiro iri nani kana achinge atsanangura mhando yepamusoro kumberi, chengetedza nzira yekukwira kwevanhu yemakesi ekumucheto, uye kuteedzera zvese zvakawanikwa zvechigadzirwa uye mutengo wekukanganisa nekufamba kwenguva.
Flash Attention mukuita
Kushandira vabatsiri vekutaura nekukurumidza nekumhanyisa prefill nhanho uko kukurumidza kukurumidza kunotanga kuverengwa.
Kushandira vabatsiri vekutaura nekukurumidza nekumhanyisa nhanho yekuzadza apo kukurumidza kunotanga kuverengwa Matimu anowanzo kuwana mhedzisiro iri nani kana achinge atsanangura emhando yepamusoro kumberi, chengetedza nzira yekukwira kwevanhu yemakesi ekumucheto, uye kuteedzera zvese zvakawanikwa zvechigadzirwa nemitengo yekukanganisa nekufamba kwenguva.
Flash Attention mukuita
Kugonesa maturusi ekuongorora-gwaro anomedza mabhuku ese kana macodebase nekuita kutevedzana-kurebesa kutarisisa kugoneke paGPU imwechete.
Kugonesa maturusi ekuongorora-gwaro anopinza mabhuku akazara kana macodebase nekuita kuteedzana-kurebesa kutarisisa kunogoneka pane imwechete GPU Matimu anowanzo kuwana mhedzisiro iri nani kana vachinge vatsanangura hunhu hwepamberi, chengetedza nzira yekukwira kwevanhu yemakesi emupendero, uye kuteedzera zvese zvakawanikwa zvechigadzirwa uye kukanganisa mutengo nekufamba kwenguva.
Flash Attention mukuita
Simba rekuona uye odhiyo Transformers uko yakakwirira-resolution yekuisa inogadzira yakareba kwazvo tokeni kutevedzana.
Simba rekuona uye odhiyo Transformers uko yakakwirira-resolution yekuisa inogadzira yakareba kwazvo tokeni kutevedzana Matimu anowanzo kuwana mhedzisiro iri nani kana achinge atsanangura emhando yepamusoro kumberi, chengetedza nzira yekukwira kwevanhu yemakesi ekumucheto, uye kuteedzera zvese zvakawanikwa zvechigadzirwa uye mutengo wekukanganisa nekufamba kwenguva.
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.
Tsanangura latency, mhando, uye mutengo zvinangwa usati waitwa. Bata nhanho yega yega segedhi rehumbowo: kana maitiro asina kusangana, imbomira kuburitsa, vhara gaka, uye wobva wawedzera kushandiswa.
Benchmark pasi pechokwadi mutoro uye data mamiriro.
Benchmark pasi pechokwadi mutoro uye data mamiriro. Bata nhanho yega yega segedhi rehumbowo: kana maitiro asina kusangana, imbomira kuburitsa, vhara gaka, uye wobva wawedzera kushandiswa.
Chishandiso chekutarisa zvikanganiso, kudonha, uye mushandisi maitiro.
Chishandiso chekutarisa zvikanganiso, kudonha, uye mushandisi maitiro. Bata nhanho yega yega segedhi rehumbowo: kana maitiro asina kusangana, imbomira kuburitsa, vhara gaka, uye wobva wawedzera kushandiswa.
Gadzirira nzira dzekudzosera kumashure uye dzezviitiko usati wawedzera.
Gadzirira nzira dzekudzosera kumashure uye dzezviitiko usati wawedzera. Bata nhanho yega yega segedhi rehumbowo: kana maitiro asina kusangana, imbomira kuburitsa, vhara gaka, uye wobva wawedzera kushandiswa.