Skip to content

Author

Zaishuo Xia

We have 3 of 7 papers

We haven’t gathered this author’s papers yet. Follow them and we’ll fetch their work.

Not the right person? Other researchers publish under this name.

Preprint Jul 2026

Improving Weak World Models Behind Strong Agents in Atari Pong

Ball-region CGSReg improves pixel-space zero-shot MBRL in DreamerV3, DIAMOND, TWISTER, and Simulus, and also improves closed-loop rollouts in the first three; STORM shows no clear improvement.

Yukuang Lu, Zaishuo Xia, Weyl Lu et al. · 1 citation

Concept-Guided Spatial Regularization for World Models in Atari Pong

World models are usually evaluated as components of model-based reinforcement learning (MBRL) systems, leaving their standalone reliability understudied. We reproduce five visual world-model agents in Atari Pong -- DreamerV3, DIAMOND, TWISTER, Simulus, and STORM -- and match their reported agent performance. We then fr...

Yukuang Lu, Zaishuo Xia, Weyl Lu et al. · 0 citations
Preprint Aug 2026

Beyond Myopic World Models: Long-Horizon End-to-End Training for Direct Future Prediction

The Direct Prediction World Model is introduced, a non-recursive architecture that compresses an action sequence of arbitrary length into a single embedding and predicts the endpoint observation in a single forward pass, making long-horizon end-to-end training practical at horizons where unrolled autoregressive trainin...

Xinyi Li, Zaishuo Xia, Chenjie Hao et al. · 0 citations

We use cookies to run the site and, with your consent, for analytics and to show ads. See our Cookie Policy.