An AI learns to look before it leaps
Agima solved 40% of harder box-pushing puzzles against 2% for an action-only counterpart. Both trained on two-box Sokoban, then faced four boxes with real feedback every five moves.
A push can block another box. Agima makes that changed state explicit: it predicts the next board, then uses the picture to choose another move.
During training, expert moves are paired with the boards they actually produced. At test time, Agima supplies those pictures itself. Our hypothesis: explicit state updates reduce bookkeeping mistakes.
This score uses one imagined sequence, before the later experiments with branching and reward training.
The interventions suggest visual state is useful, but not sufficient. Better predictions alone leave most puzzles unsolved; selecting a workable plan remains another plausible bottleneck.