Skip to content
Portrait of Ziyang Rao (饶子扬)

饶子扬 / Ziyang Rao

PhD candidate @ HKUST (Guangzhou) Guangzhou, China

I study embodied AI, flow matching and uncertainty in vision-language-action models at HKUST (Guangzhou).

Flow Matching VLAUncertainty QuantificationMechanistic Interpretability

About Me

I am a PhD candidate in Artificial Intelligence at HKUST (Guangzhou), advised by Prof. Hui Xiong, and a research intern at the X-Lab of AI² Robotics.

My work centres on embodied AI: I study the geometry of flow-matching policies and ask when their actions can be trusted, so that vision-language-action models fail loudly rather than silently. This grows out of a wider interest in uncertainty quantification and mechanistic interpretability — what representations a model actually forms, and what its internals reveal about the reliability of its outputs. My earlier research also spans information retrieval and multimodal LLMs.

I earned my MPhil in AI at the same lab, co-advised by Prof. Hui Xiong and Dr. Xuming Hu, where alongside research I was actively engaged in AI startups and knowledge transfer.

Education

Hong Kong University of Science and Technology (Guangzhou)

PhD. candidate in Artificial Intelligence

2026 - present

Hong Kong University of Science and Technology (Guangzhou)

MPhil. in Artificial Intelligence

2023 - 2025

Renmin University of China

BSc. in Economic Statistics

2019 - 2023

Selected Publications

PreprintFlow MatchingUncertainty QuantificationEmbodied AI

The Geometry of Flow-Matching Uncertainty: A Cost-free Uncertainty Proxy and Its Application in Flow-based VLA Failure Detection

Ziyang Rao, Yiren Zhao, Weiyu Guo, Ben Fei, Yandong Guo, Hui Xiong

KDD 2026LLM AgentsElectronic Health RecordsMemory

U-EHR: A Self-Evolving EHR Agent with Step-Level Credit Assignment and UCB-Guided Memory Retrieval

Hao Wu, Zihan Wang, Ziyang Rao, Heyi Lin, Jinjing Zhu, Qianyi Cai, Yi Zhou, An Lin, Hao Wang, Hui Xiong

SIGIR 2026Information RetrievalLLM

LLM-Oriented Information Retrieval: A Denoising-First Perspective

Lu Dai, Liang Sun, Fanpu Cao, Ziyang Rao, Cehao Yang, Hao Liu, Hui Xiong

ICLR 2025Event CameraDataset

RGB-Event ISP: The Dataset and Benchmark

Yunfan Lu, Yanlin Qian, Ziyang Rao, Junren Xiao, Liming Chen, Hui Xiong

PreprintVision-Language-ActionProprioceptionEmbodied AI

How Should Vision-Language-Action Models Use Proprioceptive State?

Yiren Zhao, Ziyang Chen, Ziyang Rao, Pengteng Li, He Zhang, Weiyu Guo, Yandong Guo, Rushi Dai

PreprintFlow MatchingImitation Learning

Source-Lifted Flow Matching for Intervenable Multimodal Imitation

He Zhang, Ying Sun, Pengteng Li, Ziyang Chen, Yiren Zhao, Ziyang Rao, Weiyu Guo, Yandong Guo, Hui Xiong

PreprintMechanistic InterpretabilityLLM

Towards Mechanistically Understanding Why Memorized Knowledge Fails to Generalize in Large Language Model Finetuning

Lu Dai, Ziyang Rao, Yili Wang, Hanqing Wang, Hao Liu, Hui Xiong