Improving Weak World Models Behind Strong Agents in Atari Pong

Yukuan Lu, Zaishuo Xia, Weyl Lu +1

Strong world-model agents frequently contain weak world models. We study this agent-world-model gap by reproducing five visual world-model agents in Atari Pong: DreamerV3, DIAMOND,…