Исследовательская группа Reactor представила Open Dreamer — полноценную открытую реализацию конвейера мировых моделей Dreamer 4. Проект, написанный на JAX и Flax NNX, включает в себя все ключевые компоненты для обучения агентов, работающих с видеотокенизацией и латентной динамикой, что позволяет сообществу воспроизводить сложные архитектуры обучения с подкреплением в контролируемой среде.
Релиз включает два репозитория, охватывающих полный цикл обучения: от каузального видеотокенизатора до модели латентной динамики, обусловленной действиями (action-conditioned). Использование стека JAX обеспечивает высокую производительность при вычислениях, что критически важно для обучения моделей, предсказывающих развитие состояний среды в долгосрочной перспективе. Публикация рецепта обучения дает исследователям инструмент для глубокого анализа того, как агенты строят внутренние репрезентации мира.
Данная реализация упрощает доступ к передовым методам обучения моделей мира, которые ранее были ограничены закрытыми проприетарными разработками. Наличие готовых инструментов для генерации роллаутов и оценки качества через FVD (Frechet Video Distance) позволяет стандартизировать процесс тестирования новых алгоритмов принятия решений и планирования в агентных системах.
Ключевые факты
- Проект реализован на базе фреймворков JAX и Flax NNX.
- В состав репозитория входят каузальный видеотокенизатор и модель латентной динамики.
- Реализованы инструменты для генерации роллаутов и метрики FVD для оценки качества генерации.
- Релиз включает полный рецепт обучения, обеспечивающий воспроизводимость результатов.
- Разработчиком выступила исследовательская группа Reactor.
