Jiaxu Zhou

About me

I am an undergraduate student in Math and Information Engineering at The Chinese University of Hong Kong. During the summer of 2026, I was a visiting student intern at UC Berkeley, where I was honored to work with Prof. Koushil Sreenath.

I am interested in (1) Learning Structured Dynamics and World Models, (2) Planning and Control with Learned Models, and (3) Autonomous Exploration and Skill Discovery.

Publications

2026

ICRA 2027 In Submission WMGuide teaser animation

WMGuide: Guiding Diffusion Action Prior with Learned World Model for Humanoid Planning

Jiaxu Zhou, Yiyang Shao, Yaokun Han, Yufeng Chi, Yakun Sophia Shao, Koushil Sreenath

Submitted to IEEE International Conference on Robotics and Automation (ICRA), 2027.

Video

ICRA 2027 In Submission BiFlow Policy teaser animation

BiFlow Policy: Bidirectional Normalizing-Flow Policies for Robotic Manipulation with One Network Evaluation

Yaokun Han, Jiaxu Zhou, Sattvik Sapuram, Koushil Sreenath

Submitted to IEEE International Conference on Robotics and Automation (ICRA), 2027.

NHB Major Revision Overview of biases examined by the PIMMUR Principles

The PIMMUR Principles: Ensuring Validity in Collective Behavior of LLM Societies

Jiaxu Zhou*, Jen-Tse Huang*, Xuhui Zhou, Man Ho Lam, Xintao Wang, Hao Zhu, Wenxuan Wang, Maarten Sap

Major revision at Nature Human Behaviour, 2026.

Paper Code

2025

ICML 2025

Linear, flat, and hierarchical multi-agent collaboration structures

On the Resilience of LLM-Based Multi-Agent Collaboration with Faulty Agents

Jen-Tse Huang, Jiaxu Zhou, Tailin Jin, Xuhui Zhou, Zixi Chen, Wenxuan Wang, Youliang Yuan, Michael Lyu, Maarten Sap

Proceedings of the 42nd International Conference on Machine Learning (ICML), 2025.

Paper Code

NAACL 2025 SOTOPIA-S4 system architecture and web interface

SOTOPIA-S4: A User-Friendly System for Flexible, Customizable, and Large-Scale Social Simulation

Xuhui Zhou*, Zhe Su*, Sophie Feng, Jiaxu Zhou, Jen-Tse Huang, Hsien-Te Kao, Spencer Lynch, Svitlana Volkova, Tongshuang Wu, Anita Woolley, et al.

Proceedings of NAACL-HLT, System Demonstrations, 2025.

Paper Project Code

Awards

Professor Charles K. Kao Research Exchange Scholarship2026
Talent Development Scholarship from Hong Kong Government2025
Dean's Honours List2023, 2024
Mr. & Mrs. Chin F. Foin Memorial Scholarship2023, 2024
First Prize, National Olympiad in Informatics in Provinces (NOIp)2021