Identifiable Token Correspondence for World Models

Youngin Kim, Ray Sun, Inho Kim +2

Token-based transformer world models have shown strong performance in visual reinforcement learning, but often suffer from temporal inconsistency in long-horizon rollouts, includin…