Agents can learn to imagine and reason about action consequences before acting by combining world modeling with planning in a single model, dramatically improving performance in complex environments with irreversible actions.
This paper introduces Agentic Imagination (Agima), a training framework that teaches agents to simulate action consequences before executing them.