AI & Computational Science

Algebraic World Model: From Exploration to Proofs

How the science connects

Machine learningAutomated theorem …Formal verification

AI Insight

Researchers have developed the Algebraic World Model (AWM), a framework that enables AI systems to explore unfamiliar environments, construct mathematical models of their rules, and generate formal proofs in Lean 4 about what is possible or impossible in those environments. AWM successfully proved that certain game levels were unsolvable by constructing formal proofs that excluded all possible winning sequences, outperforming baseline models like Claude Opus 5.5 and GPT-6 Astra. When tested on 25 ARC-AGI-3 games comprising 183 levels, AWM achieved a score of 99.9% while using only 43% of the actions required by human players.


This work demonstrates AI's potential to reason rigorously about unknown systems without predefined specifications, which could enable automated verification of software, game design, robotics planning, and scientific hypothesis generation. The ability to formally prove impossibility results, not just find solutions, represents a significant advance in AI's capacity for mathematical reasoning about real-world environments.


⚠️ Preprint – Noch nicht peer-reviewed

Dieser Artikel wurde noch nicht von unabhängigen Experten begutachtet. Die Ergebnisse sind vorläufig und sollten mit Vorsicht interpretiert werden.

Abstract: AI has made remarkable progress in mathematics, solving challenging problems with machine-checkable proofs in Lean, yet typically relying on predefined mathematical definitions and formal specifications. This raises a natural question: can it reason rigorously about an unknown world? We introduce the Algebraic World Model (AWM), a framework that turns interaction with unfamiliar environments into explicit mathematical models supporting formal reasoning. AWM explores the environment and develops executable Python hypotheses, continually testing and revising them against observations. It then structures these hypotheses into typed objects and shared rules, simplifies their structure through category-theory-guided transformations, and formalizes the resulting model in Lean 4 to prove reachability and impossibility. AWM can formally prove that a goal is unreachable, even beyond explored trajectories. On two impossible game levels, it constructs closed, kernel-checked Lean proofs excluding every legal winning sequence in all six runs, while Claude Opus 5.5 and GPT-6 Astra, equipped with coding tools, produce none in twelve baseline runs. Four of AWM’s six initial bounds are correct, compared with one for Opus 5.5 and none for GPT-6 Astra; given only feedback that a cheaper win exists, AWM revises its rules and constructs new proofs, reaching six of six correct bounds. Moreover, AWM produces interpretable mathematical models whose rules can be inspected and tested step by step. Across all 25 public ARC-AGI-3 games, these models enable Lean to replay the winning trajectories of all 183 levels and generalize beyond observed play, predicting unseen transitions and detecting changes in game mechanics. With Claude Opus 5, AWM clears all levels with a score of 99.9, while its certified winning trajectories use only 43% of the human baseline’s actions.

Source: Algebraic World Model: From Exploration to Proofs