Xuezhou Xu

I’m currently a research assistant 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
    (*equal contribution)
    arXiv preprint arXiv:2610.25600, 2026
    Accepted by IROS 2026 Workshop on Search Algorithms for Robot Learning

    VQ-WM vector-quantizes a visual world model’s latent space by controller reachability, so a discrete planner can search far ahead over reachable states while a continuous controller tracks each planned state with visual feedback.

  2. In Submission
    Xuezhou Xu*, Yutong Li*, Jinpeng Liu, and Xingyu Liu
    (*equal contribution)
    2026

    Toss2Reorient reorients objects in the air by tossing and catching them with a dexterous hand-arm system, cutting the average reorientation time by 50.6% across 30 objects in simulation.

  3. NeurIPS 2024
    Chengyang Ying, Hao Zhongkai, Xinning Zhou, Xuezhou Xu, Hang Su, Xingxing Zhang, and Jun Zhu
    Advances in Neural Information Processing Systems, 2024

    PEAC pre-trains agents across embodiments through reward-free interaction, learning embodiment-aware, task-agnostic knowledge that transfers to new embodiments.

  4. arXiv
    Hengkai Tan*, Xuezhou Xu*, Chengyang Ying, Xinyi Mao, Songming Liu, Xingxing Zhang, Hang Su, and Jun Zhu
    (*equal contribution)
    arXiv preprint arXiv:2411.01850, 2024

    ManiBox is a bounding-box-guided framework that scales up simulation data generation to teach manipulation policies spatial generalization.


Academic Service

  • Reviewer: ICRA 2027

Blogs

Coming soon.