Mai departeUrmătorul ghid
Modele spațiale de stat și Mamba
Fundamentele
GHID AI limbaj
Mamba este un model de secvență construit pe modele de spațiu de stat (SSM) care procesează textul în timp liniar, oferind o alternativă rapidă la atenția pătratică a Transformerului.
Ascultă acest ghidNarativ AI · ~2 min
Trucul său cheie este ca modelul să decidă în mod selectiv ce să-și amintească și să uite, pe baza intrării în sine.
Mamba, introdus de Albert Gu și Tri Dao la sfârșitul anului 2023, este construit pe modele structurate de spațiu de stat. Un SSM clasic comprimă întreaga istorie a unei secvențe într-o stare ascunsă de dimensiuni fixe și o actualizează pas cu pas, ca o rețea recurentă sofisticată. Descoperirea este selectivitatea: Mamba face ca parametrii SSM-ului (cât să păstrați, cât să lăsați să intre) să depindă de simbolul curent, astfel încât modelul să se poată concentra pe cuvintele relevante și să ignore umplerea. Aceasta permite unei stări de dimensiune fixă să acționeze ca o memorie conștientă de conținut. Deoarece evită compararea fiecărui jeton cu orice alt jeton, Mamba se scalează liniar cu lungimea secvenței și rămâne rapid la intrări foarte lungi, cum ar fi genomul, audio sau textul cu lungimea unei cărți.
Modelarea secvențelor de ADN extrem de lungi în care Transformerele de milioane de jetoane sunt prea scumpe
Alimentarea asistenților lingvistici cu context lung care rezumă cărți întregi fără trunchiere
Generarea audio în timp real și modelarea vorbirii care procesează eficient formele de undă brute
Implementări pe dispozitiv sau edge în care o stare recurentă de dimensiuni mici economisește memorie față de un cache de atenție în creștere
Mamba și succesorul său Mamba-2 forțează arhitecturi hibride care intercalează câteva straturi de atenție cu multe straturi SSM, captând punctele forte ale ambelor. Așteptați-vă SSM-uri în asistenți cu context lung, modele pe dispozitiv în care memoria este constrânsă și domenii non-text precum ADN și audio. Cercetările verifică dacă SSM-urile pure pot potrivi Transformers în sarcinile care necesită o retragere precisă și dacă se pot adapta la cele mai mari dimensiuni de model.
Buletin gratuit
Trei povești AI verificate în fiecare dimineață de zi lucrătoare, scrise în engleză simplă. Gratuit pentru totdeauna, fără reclame.
Un e-mail în fiecare zi lucrătoare. Dezabonare cu un singur clic. Nu vindem niciodată și nu împărtășim adresa ta.
Testează-te
Feedback instantaneu la fiecare răspuns și un certificat partajabil cu un ID verificabil după ce parcurgi un curs.
Susține educația AI gratuită. AI Understanding este o organizație nonprofit 501(c)(3) — fără reclame, fără barieră de plată, pentru totdeauna. Fă o donație
Mamba este un model de secvență construit pe modele de spațiu de stat (SSM) care procesează textul în timp liniar, oferind o alternativă rapidă la atenția pătratică a Transformerului. Trucul său cheie este ca modelul să decidă în mod selectiv ce să-și amintească și să uite, pe baza intrării în sine.
Mamba evită compararea tuturor perechilor de jetoane, astfel încât costul său crește liniar cu lungimea secvenței, mai degrabă decât pătratic ca atenția.
Selectivitatea înseamnă că parametri precum B, C și delta devin funcții ale intrării curente, oferind memorie conștientă de conținut.
SSM-urile sunt modele în stil recurent care pliază trecutul într-o stare de dimensiune fixă, similară cu un RNN.
Vederea de convoluție necesită parametri fix (invarianți în timp); parametrii dependenți de intrare o întrerup, așa că Mamba folosește o scanare paralelă.
Mamba utilizează o scanare selectivă conștientă de hardware care păstrează stările intermediare în SRAM rapid pentru a recupera debitul.
Continuați să învățați
Mai multe ghiduri alese pentru acest subiect
Mai departeUrmătorul ghid
Modele spațiale de stat și Mamba
Fundamentele