навчання з підкріпленнямППОградієнт політикимашинне навчанняштучний інтелект
Стабільне навчання в PPO проти методів градієнта нестабільної політики
Проксимальна оптимізація політики (PPO) впроваджує обрізані цільові функції та мислення на основі довірчих областей у навчання з підкріпленням, що значно зменшує волатильність, яка є проблемою для звичайних градієнтних підходів до політики. У той час як традиційні методи, такі як REINFORCE та стандартні алгоритми «актор-критик», можуть розходитися або руйнуватися під час навчання, конструкція PPO забезпечує обмеженість оновлень та їх відтворюваність протягом усіх прогонів.
Найважливіше
Обмежена мета PPO запобігає катастрофічному колапсу політики, від якого страждають звичайні градієнти політики.
Градієнти ванільної політики вимагають ретельного налаштування темпів навчання та базових рівнів, щоб уникнути розбіжностей.
PPO повторно використовує розгортання протягом кількох епох, що забезпечує кращу ефективність вибірки, ніж чисті методи, що базуються на політиці.
PPO став стандартним алгоритмом, що лежить в основі систем RLHF, що використовуються для навчання сучасних моделей великих мов.
Що таке Стабільне навчання в PPO?
Обрізана сурогатна мета, яка утримує оновлення політики в безпечному діапазоні, запобігаючи деструктивним крокам навчання.
PPO був представлений командою Джона Шульмана в OpenAI у 2017 році як удосконалення TRPO.
Основний механізм використовує скорочений коефіцієнт ймовірності приблизно від 0,8 до 1,2, щоб обмежити те, наскільки нова політика може відхилятися від старої.
PPO ефективно масштабується на мільйонах кроків середовища та працює на одному кластері графічного процесора або процесора.
Він став алгоритмом за замовчуванням для багатьох відомих систем RLHF, що використовуються для навчання великих мовних моделей.
Емпіричні тести показують, що PPO відновлюється після поганої ініціалізації набагато плавніше, ніж базові лінії градієнта політики Vanilla.
Що таке Методи градієнта нестабільної політики?
Класичні алгоритми навчання з підкріпленням, які оновлюють політики безпосередньо вздовж градієнта очікуваної прибутковості, часто створюючи нестабільні криві навчання.
REINFORCE, базовий алгоритм градієнта політики, був опублікований Рональдом Вільямсом у 1992 році.
Градієнти ванільної політики страждають від високої дисперсії, оскільки вони залежать від результатів Монте-Карло з повних епізодів.
Без довірчих регіонів одне велике оновлення може перетворити політику на вироджену детерміністичну дію.
Ці методи часто вимагають значного налаштування гіперпараметрів, включаючи спад швидкості навчання та формування винагороди, для збіжності.
Варіанти актор-критика, такі як A2C, зменшують дисперсію, але все ще не мають жорстких обмежень на оновлення, які застосовує PPO.
Таблиця порівняння
Функція
Стабільне навчання в PPO
Методи градієнта нестабільної політики
Механізм оновлення
Обрізана сурогатна мета з коефіцієнтом ймовірності, обмеженим поблизу 1,0
Необроблений градієнтний підйом при очікуваному поверненні без жорсткого обмеження оновлення
Стабільність тренувань
Високий — відновлюється після невдалих кроків і рідко відхиляється від норми
Низький — чутливий до швидкості навчання та шкали винагород, схильний до колапсу
Ефективність вибірки
Помірний; використовує кілька епох міні-пакету SGD на розгортання
Часто погано, якщо не поєднувати з базовими рівнями або хитрощами щодо зменшення дисперсії
Складність впровадження
Простий — приблизно такий самий розмір коду, як і звичайний градієнт політики
Простий у базовій формі, але його стабілізація вимагає додаткової інженерії
Чутливість гіперпараметрів
Відносно прощає недоліки в широкому діапазоні коефіцієнтів кліпів та швидкостей навчання
Висока чутливість; невеликі зміни можуть повністю перервати тренування
Обробка відхилень
Вбудоване відсікання діє як неявний засіб зменшення дисперсії
Потрібні окремі методи, такі як базові лінії, GAE або нормалізація переваг
Виконання настінного годинника
Швидка робота на сучасному обладнанні завдяки оптимізації першого порядку
Порівнянно за кроком, але нестабільність часто витрачає час настінних годинників на невдалі прогони
Типові випадки використання
RLHF для мовних моделей, робототехніки, ігор, безперервного керування
Теоретичний аналіз, прості середовища, навчання з підкріпленням
Детальне порівняння
Основна алгоритмічна філософія
Визначальна ідея PPO полягає в тому, що оновлення політик повинні бути невеликими та оборотними. Зменшуючи коефіцієнт ймовірності між новими та старими політиками, алгоритм запобігає кроку оптимізатора, який би занадто різко змінив поведінку за одну ітерацію. Методи градієнта нестабільної політики використовують протилежний підхід: вони дотримуються необробленого градієнта очікуваної прибутковості, сподіваючись, що добре налаштована швидкість навчання дозволить контролювати ситуацію. На практиці ця довіра часто виявляється хибною.
Стабільність та конвергенційна поведінка
Проведення PPO зазвичай демонструє шумну, але монотонно покращувальну криву навчання з випадковими спадами, які відновлюються протягом кількох ітерацій. Градієнти ванільної політики, навпаки, можуть стабілізуватися протягом тисяч кроків, а потім раптово руйнуватися, коли рідкісна траєкторія високої винагороди зміщує параметри в погану область. Обмежена ціль у PPO діє як запобіжне гальмо, обмежуючи вплив будь-якої окремої партії досвіду.
Накладні витрати на інженерію та налаштування
Надійна робота градієнтів стандартної політики часто означає ручне налаштування швидкості навчання, коефіцієнтів дисконтування, бонусів ентропії та порогів відсікання градієнтів. PPO консолідує значну частину цієї інженерії в один гіперпараметр відсікання, зазвичай встановлений на значення від 0,1 до 0,3, який є стійким для широкого кола завдань. Для команд, що розробляють RL-системи у виробничому середовищі, таке зменшення навантаження на налаштування безпосередньо призводить до швидших циклів ітерацій.
Компроміси ефективності зразків
PPO повторно використовує кожне розгортання протягом кількох епох міні-пакетних оновлень, що покращує ефективність вибірки порівняно з чисто політиковими методами, такими як REINFORCE. Однак, саме це повторне використання також є причиною важливості відсікання: без нього алгоритм перенавчився б до застарілих траєкторій. Методи нестабільного градієнта політики зазвичай є однопрохідними за розгортання, що робить їх менш ефективними щодо вибірки, але також менш схильними до цього конкретного режиму відмови.
Впровадження в реальному світі
PPO став фактичним вибором для прикладного навчання з підкріпленням, що забезпечує роботу систем від агентів Dota 5v5 від OpenAI до конвеєрів RLHF, що стоять за ChatGPT та іншими сучасними чат-ботами. Методи градієнта політики Vanilla залишаються цінними як навчальні інструменти та базові показники в дослідницьких роботах, але вони рідко зустрічаються у виробничих системах, де важлива надійність. Перехід до PPO відображає ширшу тенденцію в машинному навчанні до методів, які працюють «нестандартно».
Переваги та недоліки
Стабільне навчання в PPO
Переваги
+Високостабільні оновлення
+Прощаючі гіперпараметри
+Простий у впровадженні
+Переконливі емпіричні результати
Збережено
−Трохи упереджені оновлення
−Може переобладнати розгортання
−Потрібне налаштування кліпси
−Менше теоретичної елегантності
Методи градієнта нестабільної політики
Переваги
+Теоретично чистий
+Легко вивести
+Чудово підходить для навчання
+Низький рівень обчислень на крок
Збережено
−Оцінки з високою дисперсією
−Схильний до розбіжностей
−Потрібне серйозне налаштування
−Низька ефективність вибірки
Поширені помилкові уявлення
Міф
PPO — це просто вигадлива версія REINFORCE без реального теоретичного обґрунтування.
Реальність
PPO базується на ідеї довірчої області з TRPO, але замінює обмежену оптимізацію на відсіканий сурогат першого порядку. Відсікання забезпечує практичне наближення обмеження довірчої області, тому воно так добре працює емпірично, незважаючи на простішу реалізацію.
Міф
Градієнти ванільної політики завжди сходяться, якщо використовувати достатньо малий коефіцієнт навчання.
Реальність
Невеликий коефіцієнт навчання уповільнює дивергенцію, але не усуває її. Негативні траєкторії все ще можуть призвести до виродження політики, а висока дисперсія результатів Монте-Карло означає, що періодичні великі ефективні оновлення практично неминучі без явних обмежень.
Міф
PPO не може використовуватися для завдань безперервного контролю.
Реальність
PPO надзвичайно добре працює на тестах безперервного керування, таких як локомоція MuJoCo та роботизована маніпуляція. Відсічена ціль не залежить від простору дії, а PPO з гауссовими політиками залишається сильною основою для задач, починаючи від ходьби на четвероногих і закінчуючи спритними маніпуляціями руками.
Міф
Нестабільні градієнти політики застаріли та більше не використовуються в дослідженнях.
Реальність
Ванільні градієнти політики залишаються основоположними в дослідженнях навчання з підкріпленням. Вони фігурують як базові показники майже в кожній новій роботі про алгоритми, а такі варіанти, як природні градієнти політики, досі впливають на сучасні дослідження довірчих регіонів та обмеженої оптимізації.
Міф
PPO гарантує монотонне покращення на кожному тренувальному запуску.
Реальність
PPO значно покращує стабільність, але не гарантує монотонного прогресу. Криві навчання все ще містять шум, а патологічні функції винагороди або надзвичайно розріджені сигнали все ще можуть спричиняти збої. Стабільність означає менше катастрофічних збоїв, а не нульову кількість збоїв.
Часті запитання
Що робить PPO стабільнішим, ніж звичайні градієнти політики?
Обрізаний коефіцієнт ймовірності в цілі PPO запобігає надмірній зміні політики за одне оновлення. Градієнти ванільної політики не мають такого захисту, тому високодисперсійна партія досвіду може змістити параметри в область, де політика руйнується. PPO по суті обмінює невелику кількість зміщення на значне зменшення дисперсії.
Чи є PPO політикою чи ні?
PPO технічно є алгоритмом, що відповідає політиці, оскільки він використовує дані з поточної політики для оновлень. Однак, він повторно використовує кожне розгортання протягом кількох епох міні-пакетних оновлень, що дає йому деякі переваги ефективності зразків методів поза політикою без складності буфера повторення досвіду.
Чому градієнти ванільної політики мають високу дисперсію?
Результати Монте-Карло з повних епізодів можуть суттєво відрізнятися залежно від того, які траєкторії вибірково досліджуються. Без базової лінії або оцінювача переваг оцінка градієнта по суті є сумою винагород, помножених на показники дії, що має високу дисперсію, особливо в середовищах з довгими горизонтами або розрідженими винагородами.
Чи можна поєднувати PPO з іншими трюками стабільності, такими як градієнтне відсікання?
Так, і часто так і є. Багато фахівців застосовують градієнтне відсікання поверх об'єктивного відсікання PPO, використовують узагальнену оцінку переваг для зменшення дисперсії та нормалізують переваги між міні-партіями. Ці доповнення доповнюють, а не замінюють основний механізм відсікання PPO.
Яке типове співвідношення кліпси використовується в PPO?
Коефіцієнт кліпінгу за замовчуванням становить 0,2, тобто коефіцієнт ймовірності обмежений приблизно значенням від 0,8 до 1,2. Значення від 0,1 до 0,3 зазвичай добре працюють для широкого кола завдань, хоча деякі середовища виграють від жорсткішого або слабшого кліпінгу залежно від структури винагороди.
Чи працює PPO для просторів дискретних та безперервних дій?
PPO природним чином обробляє обидва типи простору дій. Для дискретних дій політика видає категоріальний розподіл. Для безперервних дій вона зазвичай видає гаусів розподіл із вивченим середнім значенням та фіксованою або вивченою дисперсією. Механізм відсікання працює на відношенні ймовірностей незалежно від простору дій.
Чим PPO відрізняється від TRPO?
PPO, по суті, є наближенням першого порядку TRPO, яке набагато простіше реалізувати. TRPO використовує обмеження дивергенції KL, яке розв'язується за допомогою спряжених градієнтів та пошуку по лініях, тоді як PPO замінює все це однією операцією відсікання. PPO швидший на ітерацію та легший у налаштуванні, хоча TRPO пропонує дещо сильніші теоретичні гарантії.
Чому PPO використовується для RLHF у навчанні мовних моделей?
Стабільність PPO та здатність обробляти великі моделі на розподіленому обладнанні зробили його природним вибором, коли OpenAI потрібно було точно налаштувати моделі GPT з даними про людські вподобання. Обрізана ціль запобігає надмірному відхиленню політики від контрольованої точно налаштованої моделі, що зберігає плавність, враховуючи сигнали винагороди.
Чи можуть градієнти ванільної політики все ще перевершувати PPO в будь-яких умовах?
У вузьких дослідницьких умовах з ретельно налаштованими гіперпараметрами та простими середовищами, градієнти ванільних політик можуть відповідати кінцевій продуктивності PPO. Однак, вони зазвичай вимагають набагато більше зусиль з налаштування та дають менш узгоджені результати для випадкових початкових значень. PPO виграє за стійкістю, а не обов'язково за асимптотичною продуктивністю.
Яку роль відіграє базова лінія в методах градієнта політики?
Базова лінія віднімає оціночне значення від дохідності перед обчисленням градієнта, що зменшує дисперсію без внесення зміщення. Звичайні варіанти включають функцію цінності, вивчену критиком, ковзне середнє дохідності або просто середню винагороду в партії. PPO зазвичай використовує вивчену функцію цінності як свою базову лінію.
Висновок
Оберіть PPO, коли вам потрібен надійний універсальний алгоритм навчання з підкріпленням, який працює в різноманітних середовищах без значного налаштування. Звертайтеся до методів градієнта політики звичайних знань переважно для освітніх цілей, теоретичного аналізу або коли ви спеціально хочете вивчити режими відмов, для усунення яких був розроблений PPO.