Oossa

Fireworks: расхождение вычислений может сорвать обучение RL

Эксперимент Fireworks показал: несогласованные вычисления тренера и движка генерации приводят к падению награды, а модели MoE требуют дополнительной синхронизации.

ЗаметкаАвтор: Опубликовано Oossa: 1 мин чтения

Fireworks AI сообщает, что в ходе обучения с подкреплением модели GLM‑5.2 награда обрушилась после 20 шагов, когда тренер и движок генерации по-разному вычисляли вероятности токенов. После согласования вычислений расхождение KL оставалось нулевым, а награда — стабильной на протяжении всех 25 шагов. Команда также выяснила, что в модели Qwen‑3.5‑MoE небольшие различия в суммировании выходов экспертов привели к значительному расхождению k3. Это показывает, что для маршрутизации MoE нужно согласовывать и выбор экспертов, и точность вычислений.

Почему это важно

Согласование вычислений тренера и движка генерации помогает ML-командам избежать дорогостоящих неудачных запусков RL и надёжнее масштабировать обучение с подкреплением.

Эта статья была полезной?
Поделиться

Читайте также

Oossa · Рассылка

Неделя ИИ — простыми словами

Каждый понедельник: главное за неделю понятным языком. Бесплатно, без спама.