Agima from Berkeley trains agents to imagine before acting, generalizing to harder tasks

WeijiaShi2 · x · 2026-10-24

Researchers from UC Berkeley, Impossible Research, UW and Duke present Agima (Agentic Imagination): a single omni model that generates imagined frames of where its actions lead and plans over them, with actions and imagined frames produced in one sequence. Three planning modes: linear rollout (imagine each next frame), counterfactual branching (reject actions whose imagined outcomes fail), and forward thinking (imagine full candidate plans). Trained on easy splits of Maze2D, Sokoban, Pacman and PushT, it transfers to harder unseen splits (e.g. larger mazes, more boxes). Paper, code and project page are public.

Related event: Agima: Berkeley-led agent plans by imagining the future(2 posts)→

Original post →

More from coding & agent

coding & agent channel →