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 In Submission [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