MemWM: Memory-Augmented Text-Based World Model

Yujun Wang, Tao Zhang, Jinhe Bi +9

World models are increasingly used to support planning in agents by predicting how environment states evolve in response to agent actions. Yet fluent next-state predictions can sti…