One Life to Learn: Inferring Symbolic World Models for Stochastic Environments from Unguided Exploration
Khan, Prasad, Stengel-Eskin et al.
Symbolic world modeling requires inferring and representing an environment's transitional dynamics as an executable program. Prior work has focused on largely deterministic environments with abundant interaction data, simple mechanics, and human guidance. We address a more realistic and challenging setting, learning in a complex, stochastic environment where the agent has only "one life" to explore a hostile environment without human guidance. We introduce OneLife, a framework that models world dynamics through conditionally-activated programmatic laws within a probabilistic programming framework. Each law operates through a precondition-effect structure, activating in relevant world states. This creates a dynamic computation graph that routes inference and optimization only through relevant laws, avoiding scaling challenges when all laws contribute to predictions about a complex, hierarchical state, and enabling the learning of stochastic dynamics even with sparse rule activation. To evaluate our approach under these demanding constraints, we introduce a new evaluation protocol that measures (a) state ranking, the ability to distinguish plausible future states from implausible ones, and (b) state fidelity, the ability to generate future states that closely resemble reality. We develop and evaluate our framework on Crafter-OO, our reimplementation of the Crafter environment that exposes a structured, object-oriented symbolic state and a pure transition function that operates on that state alone. OneLife can successfully learn key environment dynamics from minimal, unguided interaction, outperforming a strong baseline on 16 out of 23 scenarios tested. We also test OneLife's planning ability, with simulated rollouts successfully identifying superior strategies. Our work establishes a foundation for autonomously constructing programmatic world models of unknown, complex environments.
academic
Une Vie pour Apprendre : Inférence de Modèles Mondiaux Symboliques pour Environnements Stochastiques à partir d'Exploration Non Guidée
La modélisation symbolique du monde nécessite d'inférer et de représenter la dynamique de transition d'un environnement sous forme de programmes exécutables. Les travaux antérieurs se sont principalement concentrés sur des environnements déterministes disposant de données d'interaction abondantes, de mécanismes simples et d'une guidance humaine. Cet article aborde un cadre plus réaliste et plus difficile : l'apprentissage dans des environnements stochastiques complexes, où l'agent ne dispose que d'« une seule vie » pour explorer un environnement hostile sans guidance humaine. Cet article propose le cadre OneLife, qui modélise la dynamique du monde par le biais de règles programmatiques conditionnellement activées dans un cadre de programmation probabiliste. Chaque règle fonctionne selon une structure prémisse-effet, s'activant dans les états mondiaux pertinents. Cela crée un graphe de calcul dynamique qui achemine le raisonnement et l'optimisation uniquement par les règles pertinentes, évitant les défis d'extensibilité lorsque toutes les règles prédisent sur des états hiérarchiques complexes, et permettant l'apprentissage de dynamiques stochastiques même avec une activation de règles clairsemée.
Les approches traditionnelles de modélisation symbolique du monde font face aux défis clés suivants :
Limitation des données : Dans le monde réel, les agents ne peuvent souvent effectuer que des interactions limitées, particulièrement dans les environnements dangereux
Gestion de la stochasticité : Les environnements réels possèdent une stochasticité irréductible, comme le comportement imprévisible des PNJ
Absence de guidance externe : Manque de récompenses spécifiques à l'environnement ou d'objectifs fournis par l'humain
Complexité d'extensibilité : Les méthodes existantes peinent à s'adapter lorsque l'environnement contient de nombreux mécanismes d'interaction
La question centrale de cette recherche est : Comment un agent peut-il rétro-ingénier les règles de mondes stochastiques complexes et hostiles avec un budget d'interaction limité et sans guidance humaine spécifique à l'environnement ?
Cadre OneLife : Propose un modèle probabiliste symbolique du monde capable d'apprendre à partir d'environnements stochastiques hostiles avec une interaction minimale, sans accès à des récompenses définies par l'humain
Environnement Crafter-OO : Réimplémentation de l'environnement Crafter exposant un état symbolique structuré orienté objet et des fonctions de transition pures
Protocole d'Évaluation : Introduit une nouvelle suite d'évaluation de modélisation du monde contenant plus de 30 scénarios exécutables et des métriques de fidélité d'état/classement d'état
Amélioration des Performances : Surpasse les méthodes de base fortes dans 16/23 scénarios de test et démontre des capacités de planification
Étant donné la fonction de transition pure d'un environnement T: S × A → Δ(S), où :
S : espace d'état
A : espace d'action
Δ(S) : distribution de probabilité sur l'espace d'état
L'objectif est d'apprendre un modèle symbolique du monde à partir d'une seule trajectoire d'exploration non guidée, capable de prédire la distribution de probabilité des transitions d'état.
Pour une transition donnée, seul l'ensemble des règles satisfaisant les prémisses I(s,a) = {i | c_i(s,a) is true} est activé, créant un mécanisme de mise à jour des paramètres clairsemé.
Mécanisme d'Activation Conditionnelle : Réalise l'activation sélective des règles par le biais de structures de prémisse, évitant l'interférence des règles non pertinentes
Mise à Jour des Paramètres Clairsemée : Effectue les mises à jour de gradient uniquement sur les règles activées prédisant les changements observés, fournissant une attribution de crédit précise
Décomposition en Règles Atomiques : Décompose les événements complexes en plusieurs règles simples, améliorant la précision d'apprentissage
Cadre de Programmation Probabiliste : Soutient la modélisation et le raisonnement sur la dynamique stochastique
Random World Model : Attribue une probabilité uniforme à tous les états candidats
PoE-World : Modèle de monde symbolique de pointe, utilisant la stratégie d'exploration et le synthétiseur de règles de cet article pour une comparaison équitable
OneLife résout avec succès le défi d'apprendre des modèles symboliques du monde à partir d'interactions limitées et non guidées dans des environnements stochastiques complexes
Le mécanisme d'activation conditionnelle des règles programmatiques et la stratégie de mise à jour des paramètres clairsemée sont des innovations clés
Le modèle du monde appris soutient une planification et une prise de décision efficaces
Goulot d'Étranglement de l'Exploration : La stratégie d'exploration pilotée par LLM peine toujours à découvrir complètement les arbres technologiques complexes
Problèmes de Mémoire : L'agent d'exploration oublie facilement les informations apprises précédemment
Spécificité à l'Environnement : L'implémentation actuelle cible principalement l'environnement Crafter-OO
Complexité Computationnelle : Les frais généraux computationnels de la synthèse de règles et de l'inférence des paramètres sont importants
Importance du Problème : Résout les défis centraux de la modélisation symbolique du monde — apprentissage dans des environnements stochastiques complexes avec données limitées
Innovation Technique : Le mécanisme d'activation conditionnelle et la stratégie de mise à jour clairsemée présentent une innovation significative
Expérimentation Complète : Protocole d'évaluation complet et vérification expérimentale multifacette
Valeur Pratique : Démontre l'efficacité réelle des applications de planification
Contribution Environnementale : Crafter-OO fournit une plateforme de test précieuse pour la modélisation symbolique du monde
L'article cite des travaux importants dans plusieurs domaines : modélisation symbolique du monde, synthèse de programmes, apprentissage par renforcement, fournissant une base bibliographique complète pour les recherches connexes. Les références clés incluent l'environnement Crafter, la méthode PoE-World, et divers travaux connexes sur l'apprentissage de représentations programmatiques.
Évaluation Générale : Ceci est un article de recherche de haute qualité qui apporte des contributions significatives dans le domaine important mais difficile de la modélisation symbolique du monde. Le cadre OneLife résout les problèmes pratiques par une conception technique ingénieuse, la vérification expérimentale est complète, et il possède une valeur académique et un potentiel pratique importants. Malgré certaines limitations, il indique clairement les directions pour les recherches futures.