Yiheng Li 李一恒

Ph.D. Candidate

Mechanical Engineering
University of California, Berkeley
Berkeley, CA

Email: yhli AT berkeley DOT edu

Biography

I am a 4th-year (2021-now) Ph.D. student at MSC Lab in the Department of Mechanical Engineering, University of California at Berkeley, under the co-supervision of Prof. Masayoshi Tomizuka and Prof. David Bogy.

My research interest includes diffusion models and autonomous driving.

Publications [Google Scholar]

	Immiscible Diffusion: Accelerating Diffusion Training with Noise Assignment, Yiheng Li, Heyang Jiang, Akio Kodaira, Masayoshi Tomizuka, Kurt Keutzer, Chenfeng Xu Annual Conference on Neural Information Processing Systems (NeurIPS 2024) [Paper\|Github]

	WOMD-Reasoning: A Large-Scale Language Dataset for Interaction and Driving Intentions Reasoning, Yiheng Li, Chongjian Ge, Chenran Li, Chenfeng Xu, Masayoshi Tomizuka, Chen Tang, Mingyu Ding, Wei Zhan International Conference on Machine Learning (ICML 2025) [Paper\|Dataset\|Github]

	Pre-training on Synthetic Driving Data for Trajectory Prediction, Yiheng Li, Seth Z Zhao, Chenfeng Xu, Chen Tang, Chenran Li, Mingyu Ding, Masayoshi Tomizuka, Wei Zhan International Conference on Intelligent Robots and Systems (IROS 2024) [Paper\|Github]

Honors and Awards

J. K. Zee Fellowship, UC Berkeley, 2024
Graduate Division Block Grant, UC Berkeley, 2022-2023

Academic Service

Conference Review:
ICRA 2023, 2024
ICLR 2025
CVPR 2025