CV

Education

• M.S. in Data Science, Harvard University, 2023-2025

• B.S. in Computer Science and Technology, Peking University, 2019-2023

Work experience

• Jun. 2024-Sep. 2024: Visiting Graduate Researcher
  • UCLA
  • Supervisor: Prof. Kai-Wei Chang
• Oct. 2021-Jun. 2023: Research Assistant
  • Peking University
  • Supervisor: Prof. Baobao Chang
• Mar. 2022-Nov 2022: Research Assistant
  • The University of Hong Kong, XLANG Lab
  • Supervisor: Prof. Tao Yu

Publications

AgentGrow: LLMs as Scalable, Customizable General-Purpose Simulators For Language Agent Training

SoCal NLP 2024

[abstract]

Dialogue Action Tokens: Steering Language Models in Goal-Directed Dialogue with a Multi-Turn Planner

Kenneth Li*, Yiming Wang*, Fernanda Viégas, Martin Wattenberg

ICLR 2025 (Under review)

[paper][code]

DS-1000: A Natural and Reliable Benchmark for Data Science Code Generation

Yuhang Lai*, Chengxi Li*, Yiming Wang*, Tianyi Zhang*, Ruiqi Zhong*, Luke Zettlemoyer, Scott Wen-tau Yih, Daniel Fried, Sida Wang, Tao Yu

ICML 2023

[paper][code][website]

Query Your Model with Definitions in FrameNet: An Effective Method for Frame Semantic Role Labeling

Ce Zheng, Yiming Wang, Baobao Chang

AAAI-2023

[paper][code]

Award

• Spring 2023: PKU Outstanding Graduate

• Fall 2022: Ubiquant Scholarship

• Fall 2022: PKU Merit Student

Teaching Assistant

• Spring 2023: Data Structures and Algorithms(B) Lecturer: Prof. Leye Wang

• Spring 2022: Design of Digital Logic Lecturer: Prof. Dong Tong

Service and leadership

• PKU 2022 Winter/ 2021 Summer School for Excellent Middle School Students. Class Advisor.
• Sep. 2019 - Jun. 2020: Operation Dept, PKU EECS Student Union. Member.

Skills

• C/C++/Python/OCaml/SQL/Verilog.