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
Una Vita per Imparare: Inferenza di Modelli Mondiali Simbolici per Ambienti Stocastici da Esplorazione Non Guidata
La modellazione simbolica del mondo richiede l'inferenza e la rappresentazione della dinamica di transizione dell'ambiente come programmi eseguibili. I lavori precedenti si sono concentrati principalmente su ambienti deterministici con dati di interazione abbondanti, meccanismi semplici e guida umana. Questo articolo affronta un'impostazione più realistica e impegnativa: l'apprendimento in ambienti stocastici complessi, dove l'agente ha "una sola vita" per esplorare un ambiente ostile senza guida umana. L'articolo propone il framework OneLife, che modella la dinamica del mondo attraverso regole programmatiche attivate condizionatamente all'interno di un framework di programmazione probabilistica. Ogni regola opera attraverso una struttura premessa-effetto, attivandosi negli stati mondiali rilevanti. Questo crea un grafo computazionale dinamico che instrada il ragionamento e l'ottimizzazione solo attraverso le regole rilevanti, evitando le sfide di scalabilità quando tutte le regole predicono stati gerarchici complessi, e consentendo l'apprendimento della dinamica stocastica anche con attivazione di regole sparse.
La domanda di ricerca centrale è: Come può un agente invertire l'ingegneria delle regole di mondi stocastici complessi e pericolosi con un budget di interazione limitato e senza guida umana specifica dell'ambiente?
Framework OneLife: Propone un modello simbolico probabilistico del mondo che può apprendere da ambienti stocastici ostili con interazioni minime, senza accesso a ricompense definite da umani
Ambiente Crafter-OO: Reimplementazione dell'ambiente Crafter che espone uno stato simbolico strutturato orientato agli oggetti e funzioni di transizione pure
Protocollo di Valutazione: Introduce una nuova suite di valutazione della modellazione del mondo contenente 30+ scenari eseguibili e metriche di fedeltà dello stato/ordinamento dello stato
Miglioramento delle Prestazioni: Supera i metodi di base forti in 16/23 scenari di test e dimostra capacità di pianificazione
Data la funzione di transizione pura dell'ambiente T: S × A → Δ(S), dove:
S: spazio degli stati
A: spazio delle azioni
Δ(S): distribuzione di probabilità sullo spazio degli stati
L'obiettivo è apprendere un modello simbolico del mondo da una singola traiettoria di esplorazione non guidata, che possa prevedere la distribuzione di probabilità delle transizioni di stato.
Per una data transizione, solo l'insieme di regole che soddisfano la premessa I(s,a) = {i | c_i(s,a) è vero} viene attivato, creando un meccanismo di aggiornamento dei parametri sparse.
Meccanismo di Attivazione Condizionata: Realizza l'attivazione selettiva delle regole attraverso la struttura della premessa, evitando l'interferenza di regole non rilevanti
Aggiornamento dei Parametri Sparse: Aggiorna i gradienti solo per le regole attivate che predicono i cambiamenti osservati, fornendo un'assegnazione precisa del credito
Decomposizione di Regole Atomiche: Scompone eventi complessi in più regole semplici, migliorando la precisione dell'apprendimento
Framework di Programmazione Probabilistica: Supporta la modellazione e il ragionamento della dinamica stocastica
Random World Model: Assegna probabilità uniforme a tutti gli stati candidati
PoE-World: Modello simbolico del mondo all'avanguardia, utilizzato con la strategia di esplorazione e il sintetizzatore di regole dell'articolo per un confronto equo
Importanza del Problema: Affronta le sfide centrali nella modellazione simbolica del mondo — apprendimento in ambienti stocastici complessi con dati limitati
Innovazione Tecnica: Il meccanismo di attivazione condizionata e la strategia di aggiornamento sparse presentano innovazione significativa
Esperimenti Completi: Protocollo di valutazione completo e verifica sperimentale da molteplici angolazioni
Valore Pratico: Dimostra l'effetto pratico dell'applicazione di pianificazione
Contributo Ambientale: Crafter-OO fornisce una piattaforma di test preziosa per la modellazione simbolica del mondo
L'articolo cita lavori importanti da molteplici campi della modellazione simbolica del mondo, sintesi di programmi e apprendimento per rinforzo, fornendo una base bibliografica completa per la ricerca correlata. Le referenze chiave includono l'ambiente Crafter, il metodo PoE-World e vari lavori correlati sull'apprendimento di rappresentazioni programmatiche.
Valutazione Complessiva: Questo è un articolo di ricerca di alta qualità che fornisce contributi significativi nel campo importante ma impegnativo della modellazione simbolica del mondo. Il framework OneLife risolve problemi pratici attraverso una progettazione tecnica intelligente, la verifica sperimentale è completa e possiede importante valore accademico e potenziale pratico. Nonostante alcune limitazioni, fornisce direzioni chiare per la ricerca futura.