Исследователи представили метод Predictive Divergence Masks, оптимизирующий процесс обучения LLM с подкреплением (RL). Новый подход заменяет стандартные маски доверительной области, используемые в PPO, более точными критериями близости и направления. Это позволяет эффективнее контролировать обновления политики, предотвращая деградацию модели при работе с off-policy данными и повышая общую стабильность обучения.

Традиционные методы обучения с подкреплением, такие как PPO, часто полагаются на коэффициенты важности сэмплированных токенов для ограничения шагов обновления. Однако такой подход нередко приводит к избыточным ограничениям или нестабильности, когда модель слишком сильно отклоняется от исходной стратегии поведения. Авторы работы предлагают использовать предиктивные маски, которые анализируют динамику расхождения вероятностей токенов до применения обновления.

Метод позволяет точнее оценивать, насколько текущее обновление политики соответствует заданным целям оптимизации, не допуская «катастрофических» сдвигов в распределении вероятностей. Это решение особенно актуально для задач Alignment и дообучения моделей на специфических наборах данных, где сохранение стабильности при интенсивном обновлении весов является критическим фактором для качества итоговых ответов.

Ключевые факты

  • Метод Predictive Divergence Masks заменяет стандартные критерии PPO для более точного контроля off-policy обновлений.
  • Предложенный подход использует два критерия: близости (proximity) и направления (direction) для оценки допустимости изменений в политике.
  • Технология направлена на решение проблемы нестабильности при обучении LLM с подкреплением, минимизируя риск деградации модели.
  • Исследование сфокусировано на улучшении математического аппарата, лежащего в основе алгоритмов оптимизации RL для больших языковых моделей.