Xuezhou Xu

I’m currently a research assistant and incoming PhD student at MARS Lab, working with Prof. Jianfei Yang and Dr. Wenyu Liang from A*STAR.

I completed my B.Sc. in Mathematics (Highest Distinction) with a minor in Computer Science at NUS. I was fortunate to work with Prof. Benjamin Riviere, Prof. Xingyu Liu, Prof. Jun Zhu and Prof. Hang Su.

More about me β†’

Research Interests

My current research focuses on world models in robotics: helping robots learn how the world works, so they can predict, plan and act more intelligently.

I enjoy mathematics-driven solutions. When tackling a real-world problem, I start by modeling simple cases, then build a more general theoretical framework to solve it, and finally refine that framework until it is elegant and intuitive.

My long-term interest is to study the brain 🧠. I want to understand the fundamental logic of cognition from the perspectives of cognitive science, psychology and biology, and use those insights to build the next generation of robot brains.


News

  • [10/2026] πŸŽ‰ Our paper Vector-Quantized World Model is out on arXiv! Check out the project page.
  • [05/2026] πŸ—½ Started as a Research Associate at RRL, New York University.
  • [04/2026] πŸ† Honored to receive the A*STAR Graduate Scholarship (Computing)! Grateful for the support from A*STAR!
  • [08/2025] πŸ”¬ Started as a Visiting Scholar at the Department of Electrical and Computer Engineering, National University of Singapore.
  • [06/2025] πŸŽ“ I complete my undergraduate in Mathematics with Highest Distinct honour from National University of Singapore!

Selected Publications

  1. In Submission
    Xuezhou Xu*, Chung-Pang Wang*, Zhenzhen Li, and Benjamin Riviere
    arXiv preprint arXiv:2610.25600, 2026
  2. In Submission
    Xuezhou Xu*, Yutong Li*, Jinpeng Liu, and Xingyu Liu
    2026
  3. NeurIPS 2024
    Chengyang Ying, Hao Zhongkai, Xinning Zhou, Xuezhou Xu, and 3 more authors
    Advances in Neural Information Processing Systems, 2024
  4. arXiv
    Hengkai Tan*, Xuezhou Xu*, Chengyang Ying, Xinyi Mao, and 4 more authors
    arXiv preprint arXiv:2411.01850, 2024

Blogs

Coming soon.