Mai departeUrmătorul ghid
Interpolarea poziției pentru extensia contextului
AI limbaj
GHID tehnic
YaRN (Yet another ROPE extension) este o tehnică eficientă pentru a extinde fereastra de context utilizabilă a unui model cu mult dincolo de ceea ce a fost antrenat.
Ascultă acest ghidNarativ AI · ~2 min
Redimensionează în mod inteligent înglobările de poziție rotativă, astfel încât un model antrenat pe, să zicem, jetoane 4K să poată gestiona 32K sau mai mult cu o reglare fină minimă.
Majoritatea LLM-urilor moderne codifică pozițiile jetonelor cu RoPE (Rotary Position Embeddings), care rotesc interogarea și vectorii cheie în funcție de unghiuri legate de poziție. Când alimentați secvențe mai lungi decât durata antrenamentului, aceste rotații intră în intervale nevăzute și modelul se defectează. YaRN, introdus în 2023 de Bowen Peng și colaboratori, rezolvă acest lucru cu interpolarea conștientă de NTK aplicată pe frecvență: lasă dimensiunile de înaltă frecvență (care captează relațiile locale, pe distanță scurtă) în mare parte neatinse, în timp ce interpolează dimensiunile de joasă frecvență (care urmăresc poziția pe distanță lungă). YaRN adaugă, de asemenea, o ajustare a temperaturii atenției pentru a contracara modificările entropiei care provin din contexte mai lungi. Rezultatul este o performanță puternică în context lung, după reglarea fină pe doar o mică parte a datelor și a pașilor pe care îi necesită abordările naive.
Extinderea unui model de context 4K deschis la 32K sau 128K pentru răspunsuri lungi la întrebări la documente cu o scurtă reglare fină
Permiterea sistemelor optimizate de recuperare pentru a ingera multe pasaje concatenate fără trunchiere
Alimentarea asistenților de cod care au nevoie de un întreg fișier de depozit mare sau mai multe fișiere într-un singur prompt
Adaptarea unui model de bază pentru conversații lungi cu mai multe rânduri care acumulează istorice mari de chat
Extensia contextului este acum o practică standard: modelele deschise livrează în mod obișnuit variante extinse YaRN care ating 128.000 de jetoane sau mai mult. Cercetările se îndreaptă către metode care extind contextul cu reglaj fin zero sau aproape zero, combină redimensionarea RoPE cu trucuri de model de atenție și mențin calitatea pe toată fereastra, mai degrabă decât doar la capete. Așteptați-vă la o integrare mai strânsă a acestor tehnici în preformare, astfel încât contextul de lungă durată este mai degrabă nativ decât modernizat.
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
YaRN (Yet another ROPE extension) este o tehnică eficientă pentru a extinde fereastra de context utilizabilă a unui model cu mult dincolo de ceea ce a fost antrenat. Redimensionează în mod inteligent înglobările de poziție rotativă, astfel încât un model antrenat pe, să zicem, jetoane 4K să poată gestiona 32K sau mai mult cu o reglare fină minimă.
YaRN, al cărui nume înseamnă încă o altă extensie RoPE, redimensionează încorporațiile de poziție rotativă utilizate în majoritatea LLM-urilor moderne.
Pozițiile dincolo de antrenament produc unghiuri de rotație pe care modelul nu le-a văzut niciodată, așa că comportamentul atenției se degradează brusc.
YaRN folosește o rampă NTK-by-parts care interpolează dim-urile de joasă frecvență cu lungime de undă lungă și lasă intacte dim-urile de înaltă frecvență cu rază scurtă.
YaRN adaugă o scalare a temperaturii la logit-urile de atenție pentru a contracara schimbările de entropie care apar pe măsură ce contextul crește.
YaRN atinge o calitate puternică a contextului lung după reglajul fin pe o mică parte din datele necesare metodelor naive.
Continuați să învățați
Mai multe ghiduri alese pentru acest subiect
Mai departeUrmătorul ghid
Interpolarea poziției pentru extensia contextului
AI limbaj