Stabilus mokymas PPO ir nestabilių politikos gradientų metoduose
„Proximal Policy Optimization“ (PPO) sustiprinto mokymosi srityje taiko apribotas tikslo funkcijas ir pasitikėjimo srities mąstymą, taip smarkiai sumažindamas nepastovumą, kuris kamuoja paprastus politikos gradiento metodus. Nors tradiciniai metodai, tokie kaip „REINFORCE“ ir standartiniai veikėjų-kritikų algoritmai, gali nukrypti arba sugriūti mokymo metu, PPO dizainas išlaiko atnaujinimus ribotus ir atkartojamus tarp paleidimų.
Akcentai
Apribotas PPO tikslas užkerta kelią katastrofiškam politikos žlugimui, nuo kurio kenčia įprasti politikos gradientai.
Vanilinių politikos gradientų atveju reikia atidžiai derinti mokymosi tempus ir bazinius lygius, kad būtų išvengta skirtumų.
PPO pakartotinai naudoja diegimus keliose epochose, todėl yra efektyvesnis imties atžvilgiu nei vien tik pagal politiką taikomi metodai.
PPO tapo standartiniu algoritmu, naudojamu RLHF sistemoms, naudojamoms šiuolaikiniams dideliems kalbų modeliams mokyti.
Kas yra Stabilus mokymas PPO?
Apribotas pakaitinis tikslas, kuris palaiko politikos atnaujinimus saugiose ribose, užkertant kelią destruktyviems mokymosi etapams.
PPO 2017 m. „OpenAI“ platformoje pristatė Johno Schulmano komanda kaip TRPO patobulinimą.
Pagrindinis mechanizmas naudoja apribotą tikimybės santykį maždaug nuo 0,8 iki 1,2, kad apribotų, kiek naujoji politika gali nukrypti nuo senosios.
PPO efektyviai keičia mastelį per milijonus aplinkos žingsnių ir veikia viename GPU arba CPU klasteryje.
Jis tapo numatytuoju algoritmu, naudojamu daugeliui žinomų RLHF sistemų, naudojamų dideliems kalbos modeliams mokyti.
Empiriniai lyginamosios analizės rodo, kad PPO atsigauna po netinkamos iniciacijos daug grakščiau nei įprastos politikos gradiento bazinės versijos.
Kas yra Nestabilūs politikos gradiento metodai?
Klasikiniai sustiprinimo mokymosi algoritmai, kurie atnaujina politiką tiesiai pagal laukiamos grąžos gradientą, dažnai sukurdami nepastovias mokymosi kreives.
REINFORCE, pagrindinis politikos gradiento algoritmas, buvo paskelbtas Ronaldo Williamso 1992 m.
Vanilinių politikos gradientų kintamumas yra didelis, nes jie remiasi Monte Karlo metodu pagrįstomis grąžomis iš visų epizodų.
Be patikimų regionų, vienas didelis atnaujinimas gali sugriauti politiką į išsigimusį deterministinį veiksmą.
Šiems metodams konverguoti dažnai reikalingas platus hiperparametrų derinimas, įskaitant mokymosi greičio mažėjimą ir atlygio formavimą.
Veikėjų-kritikų variantai, tokie kaip A2C, sumažina dispersiją, tačiau vis tiek trūksta griežtų atnaujinimo apribojimų, kuriuos taiko PPO.
Palyginimo lentelė
Funkcija
Stabilus mokymas PPO
Nestabilūs politikos gradiento metodai
Atnaujinimo mechanizmas
Apribotas surogatinis tikslas, kurio tikimybės santykis yra artimas 1,0
Neapdorotas gradiento kilimas pagal numatomą grąžą be griežto atnaujinimo apribojimo
Treniruočių stabilumas
Aukštas – atsigauna po blogų žingsnių ir retai nukrypsta
Žemas – jautrus mokymosi greičiui ir atlygio skalei, linkęs žlugti
Imties efektyvumas
Vidutinis; vienam išleidimui naudoja kelias mini partijos SGD epochas
Dažnai prastas, nebent derinamas su baziniais rodikliais ar dispersijos mažinimo gudrybėmis
Įgyvendinimo sudėtingumas
Paprasta – maždaug toks pat kodo pėdsakas kaip ir įprasto politikos gradiento
Paprasta iš esmės, bet jos stabilizavimas reikalauja papildomos inžinerijos
Hiperparametrų jautrumas
Santykinai atlaidus esant įvairiems įrašų santykiams ir mokymosi greičiams
Labai jautrus; maži pokyčiai gali visiškai nutraukti mokymą
Nuokrypių tvarkymas
Integruotas iškirpimas veikia kaip numanomas dispersijos reduktorius
Reikalingi atskiri metodai, tokie kaip baziniai rodikliai, GAE arba pranašumų normalizavimas
Sieninio laikrodžio našumas
Greitas veikimas naudojant šiuolaikinę įrangą dėl pirmos eilės optimizavimo
Palyginama pagal žingsnį, tačiau nestabilumas dažnai švaisto sieninio laikrodžio laiką nepavykusiems vykdymams
Dažniausiai naudojami atvejai
RLHF kalbos modeliams, robotikai, žaidimams, nuolatiniam valdymui
Teorinė analizė, paprastos aplinkos, sustiprinto mokymosi mokymas
Išsamus palyginimas
Pagrindinė algoritminė filosofija
Pagrindinė PPO idėja yra ta, kad politikos atnaujinimai turėtų būti maži ir grįžtami. Apkarpydamas tikimybės santykį tarp naujos ir senos politikos, algoritmas neleidžia optimizavimo įrankiui žengti žingsnio, kuris per vieną iteraciją pernelyg drastiškai pakeistų elgseną. Nestabilūs politikos gradiento metodai taiko priešingą požiūrį: jie seka neapdorotu laukiamos grąžos gradientu, pasitikėdami, kad gerai suderintas mokymosi greitis kontroliuos situaciją. Praktiškai šis pasitikėjimas dažnai yra nepagrįstas.
Stabilumas ir konvergencijos elgsena
PPO vykdymas paprastai rodo triukšmingą, bet monotoniškai gerėjančią mokymosi kreivę, su retkarčiais pasitaikančiais nuosmukiais, kurie atsistato per kelias iteracijas. Tuo tarpu įprasti politikos gradientai gali išlikti stabilūs tūkstančius žingsnių, o tada staiga žlugti, kai reta didelio atlygio trajektorija nustumia parametrus į blogą zoną. Apribotas tikslas PPO veikia kaip saugos stabdys, ribojantis bet kurios vienos patirties partijos įtaką.
Inžinerijos ir derinimo pridėtinės išlaidos
Norint, kad įprasti politikos gradientai veiktų patikimai, dažnai reikia rankiniu būdu derinti mokymosi rodiklius, nuolaidų koeficientus, entropijos premijas ir gradiento iškirpimo slenksčius. PPO sujungia didžiąją dalį šios inžinerijos į vieną iškirpimo hiperparametrą, paprastai nustatomą nuo 0,1 iki 0,3, kuris yra patikimas atliekant įvairias užduotis. Komandoms, tiekiančioms gamybines RL sistemas, šis derinimo naštos sumažinimas tiesiogiai reiškia greitesnius iteracijos ciklus.
Efektyvumo kompromisų pavyzdys
PPO pakartotinai naudoja kiekvieną diegimą kelioms mini paketinių atnaujinimų epochoms, o tai pagerina imties efektyvumą, palyginti su grynai politiką atitinkančiais metodais, tokiais kaip REINFORCE. Tačiau šis pakartotinis naudojimas taip pat yra priežastis, kodėl iškirpimas yra svarbus: be jo algoritmas per daug prisitaikytų prie pasenusių trajektorijų. Nestabilūs politikos gradiento metodai paprastai yra vieno bandymo kiekvienam diegimui, todėl jie yra mažiau imties efektyvūs, bet ir mažiau linkę į tą konkretų gedimo režimą.
Įvaikinimas realiame pasaulyje
PPO tapo faktiniu taikomojo pastiprinimo mokymosi pasirinkimu, maitinančiu sistemas nuo „OpenAI“ „Dota 5v5“ agentų iki RLHF kanalų, sukurtų už „ChatGPT“ ir kitų šiuolaikinių pokalbių robotų. Paprasti politikos gradiento metodai išlieka vertingi kaip mokymo priemonės ir kaip atskaitos taškai mokslinių tyrimų darbuose, tačiau jie retai naudojami gamybinėse sistemose, kur patikimumas yra svarbus. Poslinkis link PPO atspindi platesnę mašininio mokymosi tendenciją link metodų, kurie veikia iš karto.
Privalumai ir trūkumai
Stabilus mokymas PPO
Privalumai
+Labai stabilūs atnaujinimai
+Atleidžiantys hiperparametrai
+Paprasta įgyvendinti
+Stiprūs empiriniai rezultatai
Pasirinkta
−Šiek tiek šališki atnaujinimai
−Galima per daug pritaikyti išvyniojimus
−Reikalingas klipų derinimas
−Mažiau teorinio elegancijos
Nestabilūs politikos gradiento metodai
Privalumai
+Teoriškai švaru
+Lengva išgauti
+Puikiai tinka mokymui
+Mažas skaičiavimas vienam žingsniui
Pasirinkta
−Didelės dispersijos įvertinimai
−Linkęs į skirtumus
−Reikalingas stiprus derinimas
−Prastas mėginių našumas
Dažni klaidingi įsitikinimai
Mitas
PPO tėra išgalvota REINFORCE versija be jokio realaus teorinio pagrindimo.
Realybė
PPO remiasi TRPO patikimo regiono idėja, tačiau suvaržytas optimizavimas pakeičiamas pirmos eilės iškirptu surogatiniu metodu. Iškirpimas pateikia praktinį patikimo regiono apribojimo aproksimavimą, todėl jis taip gerai veikia empiriškai, nepaisant to, kad yra paprasčiau įgyvendinamas.
Mitas
Vanilinės politikos gradientai visada konverguoja, jei naudojate pakankamai mažą mokymosi greitį.
Realybė
Mažas mokymosi greitis sulėtina divergenciją, bet jos nepašalina. Blogos trajektorijos vis tiek gali nustumti politiką į išsigimusius regionus, o didelis Monte Karlo grąžos dispersijos lygis reiškia, kad retkarčiais dideli efektyvūs atnaujinimai iš esmės yra neišvengiami be aiškių apribojimų.
Mitas
PPO negalima naudoti nuolatinio valdymo užduotims.
Realybė
PPO puikiai veikia nuolatinio valdymo etalonuose, tokiuose kaip „MuJoCo“ judėjimas ir robotų manipuliavimas. Apkirptas objektas nepriklauso nuo veiksmo erdvės, o PPO su Gauso politika išlieka tvirtu pagrindu problemoms spręsti – nuo keturkojų ėjimo iki miklių rankų manipuliavimo.
Mitas
Nestabilūs politikos gradientai yra pasenę ir nebenaudojami tyrimuose.
Realybė
Natūralūs politikos gradientai išlieka pagrindiniais sustiprinto mokymosi tyrimuose. Jie pateikiami kaip atskaitos taškai beveik kiekviename naujame algoritmų straipsnyje, o tokie variantai kaip natūralūs politikos gradientai vis dar naudojami šiuolaikiniame darbe, susijusiame su pasitikėjimo regionais ir suvaržytu optimizavimu.
Mitas
PPO garantuoja monotonišką tobulėjimą kiekvieno treniruotės metu.
Realybė
PPO smarkiai pagerina stabilumą, bet negarantuoja monotoniškos pažangos. Mokymosi kreivėse vis dar yra triukšmo, o patologinės atlygio funkcijos arba itin reti signalai vis tiek gali sukelti gedimus. Stabilumas reiškia mažiau katastrofiškų griūčių, o ne nulinį gedimų skaičių.
Dažnai užduodami klausimai
Kas daro PPO stabilesnį nei įprasti politikos gradientai?
Apribotas tikimybės santykis PPO tiksle neleidžia politikai per daug pasikeisti per vieną atnaujinimą. Įprasti politikos gradientai neturi tokio apsauginio barjero, todėl didelės dispersijos patirties paketas gali nustumti parametrus į regioną, kuriame politika žlunga. PPO iš esmės pakeičia nedidelį šališkumo kiekį į didelį dispersijos sumažėjimą.
Ar PPO atitinka politiką, ar ne?
PPO techniškai yra pagal politiką pagrįstas algoritmas, nes atnaujinimams jis naudoja duomenis iš dabartinės politikos. Tačiau kiekvieną išleidimą jis pakartotinai naudoja kelioms mini paketinių atnaujinimų epochoms, todėl jis turi kai kuriuos pagal politiką nesusijusių metodų efektyvumo pranašumus, tačiau nesudėtingai naudoja patirties pakartojimo buferio.
Kodėl vanilinių politikos gradientai pasižymi dideliu dispersijos lygiu?
Monte Karlo metodo grąža iš visų epizodų gali labai skirtis priklausomai nuo to, kurios trajektorijos yra tiriamos. Neturint bazinio lygio ar pranašumo įverčio, gradiento įvertis iš esmės yra atlygio ir veiksmo rodiklių sandauga, kuri pasižymi didele dispersija, ypač aplinkoje, kurioje yra ilgi horizontai arba mažai atlygio.
Ar PPO galima derinti su kitais stabilumo triukais, tokiais kaip gradiento iškirpimas?
Taip, ir dažnai taip ir yra. Daugelis praktikų taiko gradientinį kirpimą ant PPO objektyvaus kirpimo, naudoja apibendrintą pranašumo įvertinimą dispersijai mažinti ir normalizuoja pranašumus tarp mini partijų. Šie papildymai papildo, o ne pakeičia pagrindinį PPO kirpimo mechanizmą.
Koks yra tipiškas PPO naudojamas kirpimo santykis?
Numatytasis iškarpų santykis yra 0,2, o tai reiškia, kad tikimybės santykis yra apribotas maždaug nuo 0,8 iki 1,2. Vertės nuo 0,1 iki 0,3 paprastai gerai veikia atliekant įvairias užduotis, nors kai kuriose aplinkose, priklausomai nuo atlygio struktūros, naudingesnis yra griežtesnis arba laisvesnis iškarpų skaičius.
Ar PPO veikia diskrečiose ir tolydžiose veiksmo erdvėse?
PPO natūraliai apdoroja abiejų tipų veiksmų erdves. Diskretiesiems veiksmams politika išveda kategorinį skirstinį. Nuolatiniams veiksmams paprastai išveda Gauso skirstinį su išmoktu vidurkiu ir fiksuota arba išmokta dispersija. Apkarpymo mechanizmas veikia pagal tikimybės santykį, neatsižvelgiant į veiksmų erdvę.
Kuo PPO skiriasi nuo TRPO?
PPO iš esmės yra pirmos eilės TRPO aproksimacija, kurią įgyvendinti daug paprasčiau. TRPO naudoja KL divergencijos apribojimą, išspręstą naudojant konjuguotuosius gradientus ir linijų paiešką, o PPO visa tai pakeičia viena iškirpimo operacija. PPO yra greitesnis kiekvienoje iteracijoje ir lengviau derinamas, nors TRPO siūlo šiek tiek stipresnes teorines garantijas.
Kodėl PPO naudojamas RLHF kalbos modelių mokymui?
PPO stabilumas ir gebėjimas apdoroti didelius modelius paskirstytoje įrangoje pavertė jį natūraliu pasirinkimu, kai „OpenAI“ reikėjo tiksliai suderinti GPT modelius su žmonių pageidavimų duomenimis. Apribotas tikslas neleidžia politikai per daug nukrypti nuo prižiūrimo tiksliai suderinto modelio, todėl išsaugomas sklandumas ir įtraukiami atlygio signalai.
Ar įprastos politikos gradientai vis tiek gali pranokti PPO bet kokiomis aplinkybėmis?
Siauruose tyrimų kontekstuose su kruopščiai suderintais hiperparametrais ir paprastomis aplinkomis paprasti politikos gradientai gali atitikti PPO galutinį našumą. Tačiau jiems paprastai reikia daug daugiau derinimo pastangų ir jie duoda mažiau nuoseklius rezultatus esant atsitiktiniams pradiniams dydžiams. PPO laimi dėl patikimumo, o ne būtinai dėl asimptotinio našumo.
Kokį vaidmenį bazinė linija atlieka politikos gradiento metoduose?
Bazinė linija atima įvertintą vertę iš grąžos prieš apskaičiuojant gradientą, taip sumažinant dispersiją neįvedant šališkumo. Įprasti pasirinkimai apima kritiko išmoktą vertės funkciją, slenkantį grąžos vidurkį arba tiesiog vidutinį atlygį partijoje. PPO paprastai naudoja išmoktą vertės funkciją kaip bazinę liniją.
Nuosprendis
Rinkitės PPO, kai jums reikia patikimo, bendrosios paskirties sustiprinimo mokymosi algoritmo, kuris veiktų įvairiose aplinkose be didelio derinimo. Paprastus politikos gradiento metodus naudokite pirmiausia švietimo tikslais, teorinei analizei arba kai norite konkrečiai ištirti gedimų režimus, kuriuos PPO buvo sukurtas pašalinti.