Mai departeUrmătorul ghid
Optimizatoare ZeRO și Sharded
Tehnic
GHID tehnic
Adam este calul de lucru pentru optimizarea celor mai multe rețele neuronale moderne, regland automat o rată de învățare separată pentru fiecare parametru.
Ascultă acest ghidNarativ AI · ~2 min
Contează pentru că face antrenamentul modelelor adânci mai rapid și mult mai puțin captivant decât coborârea simplă în gradient.
Adam (Adaptive Moment Estimation), introdus de Kingma și Ba în 2014, combină două idei. În primul rând, impuls: menține o medie exponențială în descreștere a gradienților din trecut (primul moment), astfel încât actualizările cresc viteza în direcții consecvente. În al doilea rând, scalarea pe parametru: urmărește o medie a gradienților pătrați (al doilea moment) și împarte fiecare pas la rădăcina pătrată a acelei valori, astfel încât parametrii cu gradienți mari și zgomotoși fac pași mai mici, iar cei actualizați rar fac pași mai mari. Această adaptabilitate înseamnă că puteți utiliza adesea o rată de învățare într-o întreagă rețea. O variantă, AdamW, decuplează scăderea greutății de la actualizarea gradientului și a devenit implicită pentru antrenarea transformatoarelor mari și a modelelor de limbaj.
Instruirea modelelor de limbi mari precum GPT și Llama, care folosesc AdamW ca optimizator standard.
Reglarea fină a unui clasificator de imagini preantrenat (de exemplu, ResNet) pe un set de date personalizat doar cu o rată de învățare Adam implicită.
Instruirea modelelor de difuzie din spatele generatoarelor de imagini precum Stable Diffusion.
Rularea Adam pe 8 biți în biblioteci precum bitsandbytes pentru a se potrivi stărilor de optimizare în memoria GPU limitată.
Adam și AdamW rămân dominante, dar cercetările sporesc eficiența modelelor cu trilioane de parametri, unde stocarea a două valori suplimentare pe greutate este costisitoare. Variante ușoare de memorie, cum ar fi Adafactor, Adam pe 8 biți și optimizatori mai noi, cum ar fi Lion (care folosește doar impulsul bazat pe semne) și Sophia urmăresc să se potrivească cu calitatea lui Adam cu mai puțină memorie sau convergență mai rapidă. Așteptați-vă la optimizatori adaptivi reglați special pentru antrenamentul distribuit, cu precizie redusă, pentru a continua să evolueze.
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
Adam este calul de lucru pentru optimizarea celor mai multe rețele neuronale moderne, regland automat o rată de învățare separată pentru fiecare parametru. Contează pentru că face antrenamentul modelelor adânci mai rapid și mult mai puțin captivant decât coborârea simplă în gradient.
Adam păstrează o medie exponențială a gradienților (primul moment) și a gradienților la pătrat (al doilea moment) pentru fiecare parametru.
Atât m, cât și v sunt inițializate la zero, astfel încât estimările timpurii sunt părtinitoare; împărțirea la (1 - beta^t) corectează acest lucru.
AdamW aplică scăderea greutății direct greutăților, mai degrabă decât amestecarea acesteia în termenul de gradient adaptiv, ceea ce îmbunătățește generalizarea în transformatoare.
Epsilon (în jur de 1e-8) este adăugat la numitor, astfel încât parametrii cu medii de gradient pătrat aproape de zero să nu provoace o explozie.
Prin împărțirea la rădăcina pătrată a mediei gradientului pătrat a fiecărui parametru, Adam scalează dimensiunea pasului pe parametru, în loc să utilizeze o singură valoare globală.
Continuați să învățați
Mai multe ghiduri alese pentru acest subiect
Mai departeUrmătorul ghid
Optimizatoare ZeRO și Sharded
Tehnic