Yiheng Li 李一恒

Ph.D. Candidate

Dept. of Mechanical Engineering
University of California, Berkeley
Berkeley, CA

Email: yhli AT berkeley DOT edu

Biography

I am a third-year (2021-now) Ph.D. student 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 generative 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 2024