Learning Generative Interactive Environments By Trained Agent Exploration

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Kazemi, Naser, Savov, Nedko, Paudel, Danda, Van Gool, Luc
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866910655729106944
author Kazemi, Naser
Savov, Nedko
Paudel, Danda
Van Gool, Luc
author_facet Kazemi, Naser
Savov, Nedko
Paudel, Danda
Van Gool, Luc
contents World models are increasingly pivotal in interpreting and simulating the rules and actions of complex environments. Genie, a recent model, excels at learning from visually diverse environments but relies on costly human-collected data. We observe that their alternative method of using random agents is too limited to explore the environment. We propose to improve the model by employing reinforcement learning based agents for data generation. This approach produces diverse datasets that enhance the model's ability to adapt and perform well across various scenarios and realistic actions within the environment. In this paper, we first release the model GenieRedux - an implementation based on Genie. Additionally, we introduce GenieRedux-G, a variant that uses the agent's readily available actions to factor out action prediction uncertainty during validation. Our evaluation, including a replication of the Coinrun case study, shows that GenieRedux-G achieves superior visual fidelity and controllability using the trained agent exploration. The proposed approach is reproducable, scalable and adaptable to new types of environments. Our codebase is available at https://github.com/insait-institute/GenieRedux .
format Preprint
id arxiv_https___arxiv_org_abs_2409_06445
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Learning Generative Interactive Environments By Trained Agent Exploration
Kazemi, Naser
Savov, Nedko
Paudel, Danda
Van Gool, Luc
Computer Vision and Pattern Recognition
Artificial Intelligence
World models are increasingly pivotal in interpreting and simulating the rules and actions of complex environments. Genie, a recent model, excels at learning from visually diverse environments but relies on costly human-collected data. We observe that their alternative method of using random agents is too limited to explore the environment. We propose to improve the model by employing reinforcement learning based agents for data generation. This approach produces diverse datasets that enhance the model's ability to adapt and perform well across various scenarios and realistic actions within the environment. In this paper, we first release the model GenieRedux - an implementation based on Genie. Additionally, we introduce GenieRedux-G, a variant that uses the agent's readily available actions to factor out action prediction uncertainty during validation. Our evaluation, including a replication of the Coinrun case study, shows that GenieRedux-G achieves superior visual fidelity and controllability using the trained agent exploration. The proposed approach is reproducable, scalable and adaptable to new types of environments. Our codebase is available at https://github.com/insait-institute/GenieRedux .
title Learning Generative Interactive Environments By Trained Agent Exploration
topic Computer Vision and Pattern Recognition
Artificial Intelligence
url https://arxiv.org/abs/2409.06445