STANFORD UNIVERSITY · COMPUTER SCIENCE

Zhengxu (Jason) Yan

M.S. student · Robot learning & AI agents

I am a master’s student in Computer Science at Stanford University, working with Chelsea Finn in the IRIS Lab. I study what makes experience useful for learning, and how robots and other AI agents can identify and use the information they need to make decisions.

My current work focuses on observability in robot demonstrations. I also work with Diyi Yang and Alexander Spangher in the SALT Lab on understanding and evaluating the contributions of NLP datasets.

Previously, I studied Computer Science and Data Science at UC Berkeley, where I worked on verifiable code generation, reinforcement learning, and ML systems.

Zhengxu (Jason) Yan at his UC Berkeley graduation
Berkeley, 2025

RESEARCH

What makes experience useful for learning?

A successful demonstration can still hide the information a robot needs. A failed proof can expose a mistake in the model or in the task itself. These questions motivate my work on informative demonstrations, feedback-guided agents, and evidence-grounded evaluation.

Research & publications

* Equal contribution

FOCUS overview: physical ambiguity, action ambiguity, and observability-based curationLearning from experience
Manuscript / 2026

What Your Policy Cannot See: Robot Data Curation via Observability Estimation

T. Gao*, Z. Yan*, A. Mahajan, R. T. Bhowmik, J. Hejna, D. Sadigh, C. Finn

Successful demonstrations can still hide the information a policy needs. FOCUS estimates visual observability using auxiliary observations to identify ambiguous supervision and curate robot training data.

Vero workflow: benchmark curation, agent generation, independent evaluation, and formal auditVerifiable agents
NeurIPS 2026 · Evaluations and Datasets / 2026

Vero: Can AI Agents Build Formally Verified Software Repositories?

Zhe Ye*, Hantao Lou*, Yuechun Sun*, Peiyang Song, Zhengxu Yan, Timothe Kasriel, Qingyang Zhang, Kaiyu Yang, Soonho Kong, Jingxuan He, Dawn Song

A benchmark for generating implementations and machine-checked proofs across multi-module Lean repositories, with a formal audit mechanism for identifying flawed specifications and reference code.

VERINA tasks: specification generation, code generation, and proof generationVerifiable agents
ICLR 2026 / 2026

VERINA: Benchmarking Verifiable Code Generation

Zhe Ye, Zhengxu Yan, Jingxuan He, Timothe Kasriel, Kaiyu Yang, Dawn Song

A modular benchmark for code, specification, and proof generation in Lean. Evaluating each component and their combinations helps reveal where verifiable code generation fails.

Learning from experience
Journal of Building Engineering / 2026

A Deep Reinforcement Learning Framework for Multi-Dimensional Adaptability in Building Operations

Davide Schaumann, Zhengxu Yan, Eduard Haiman, Yehuda E. Kalay

An action-masked PPO framework coordinates space, staffing, and activity scheduling in a cardiac catheterization lab simulation under operational constraints and competing objectives.

Scientific discovery
Work in progress / 2026

Auditing Added Information in NLP Dataset Papers with Dataset Contribution Units

Z. Yan, J. Fang, A. Spangher, J. Baumann, D. Yang

What does a new dataset add? Dataset Contribution Units represent claims about tasks, domains, annotation, scale, evaluation, and availability, and connect those claims with evidence in prior work.

Scientific discovery
Preprint / 2025

NoveltyRank: A Retrieval-Augmented Framework for Conceptual Novelty Estimation in AI Research

Zhengxu Yan, Han Li, Yuming Feng

A retrieval-based framework for estimating conceptual novelty by comparing research papers with prior literature, using both classification and pairwise ranking.

ML systems
Manuscript / 2025

DiT-Serve: An Efficient Serving Engine for Diffusion Transformers

M. Luo, A. Hao, Z. Yan, C. Cao, Q. L. N. Nguyen

A serving engine for diffusion transformers with distributed attention, continuous batching, and scheduling for mixed-length, mixed-resolution workloads.

Earlier publications 2 papers · 2024

Research experience

Open a role for details
Intelligence through Robotic Interaction at Scale Lab (IRIS), Stanford AI Lab (SAIL)Graduate Research AssistantNovember 2025 - Present

Supervised by Dr. Chelsea Finn

  • Information-Theoretic Data Curation for Vision-Language-Action Models
  • Co-led the design and development of observability-aware data curation methods for vision-language-action models, from research formulation and method design to implementation and evaluation.
  • Co-designed the scoring framework and trajectory-level aggregation methods to identify demonstrations with insufficient task-relevant visual information.
  • Implemented scoring models and trained and evaluated policies on curated demonstration subsets against standard baselines.
  • Extensions to Feedback Descent
  • Proposed and implemented variants of Feedback Descent that improved performance over the original method.
Stanford Social and Language Technologies Lab (SALT), Stanford AI Lab (SAIL)Graduate Research AssistantNovember 2025 - Present

Supervised by Dr. Diyi Yang and Dr. Alexander Spangher

  • Independently developed Dataset Contribution Units (DCUs), a typed representation of how a dataset extends prior work.
  • Defined contribution dimensions spanning task, domain, annotation, scale, evaluation, and availability.
  • Developed and evaluated type-routed retrieval and support-aware reranking to connect contribution claims with supporting or conflicting prior evidence.
  • Built the paper bank, extraction pipeline, retrieval benchmark, and longitudinal-analysis tools for auditing NLP dataset contributions and adoption.
UC Berkeley Sky Computing LabResearch AssistantFebruary 2024 - October 2025

Supervised by Dr. Ion Stoica

  • Led a five-person team developing DiT-Serve, an efficient serving system for diffusion transformers.
  • Integrated Ring Attention and Brick Attention for distributed long-sequence video generation.
  • Designed continuous batching and request scheduling for mixed-length, mixed-resolution workloads.
  • Built shared model interfaces and multi-GPU benchmarks for memory, latency, and throughput.
  • Contributed to DiT-Serve's system design, implementation, and experimental evaluation.
UC Berkeley College of Environmental DesignResearch AssistantFebruary 2024 - September 2025

Supervised by Dr. Yehuda Kalay and Dr. Davide Schaumann

  • Led a three-person team developing a multi-objective reinforcement-learning framework for adaptive building operations.
  • Formulated cardiac-lab control as joint optimization over environmental and operational constraints.
  • Built the simulation and implemented PPO and Q-learning controllers for multi-dimensional adaptation.
  • Directed experiments and analysis for the resulting Journal of Building Engineering publication.
UC Berkeley School of Public HealthMachine Learning Engineer LeadJanuary 2024 - September 2025

Supervised by Dr. Stefano Bertozzi

  • Led ML engineering for Rapid Reviews: Infectious Diseases, an automated preprint-triage system.
  • Developed SciBERT- and LLM-based models for scope classification and conflict-of-interest screening.
  • Designed human-in-the-loop workflows for domain experts to inspect and correct model outputs.
  • Directed teams building ingestion, deduplication, and metadata normalization across preprint repositories.
  • Collaborated with Berkeley Public Health and UCSF researchers on deployment in review workflows.
Berkeley RDI, UC Berkeley Department of Electrical Engineering and Computer SciencesResearch Assistant2024 - 2026

Supervised by Dr. Dawn Song

  • Co-developed VERINA, an ICLR 2026 benchmark for machine-verifiable Lean code, specifications, and proofs.
  • Built compiler-guided generation and repair pipelines with executable evaluation for each output type.
  • Extended the work to Vero, the first repository-level benchmark for joint implementation and proof synthesis.
  • Contributed benchmark curation, dependency-aware evaluation, and audits for flawed specifications or reference code.
University of Nebraska-LincolnResearch AssistantApril 2023 - September 2023

Supervised by Dr. Valerie Jones

  • Developed a human-in-the-loop NLP method for analyzing older adults' voice-assistant interactions.
  • Designed annotation and error-review workflows that let domain researchers refine model rules.
  • Contributed to empirical analysis and two peer-reviewed journal publications.

Education

Stanford University

M.S. in Computer Science

Expected April 2027
University of California, Berkeley

B.A. in Computer Science and Data Science

2021–2025 · High Distinction

Teaching & service

Large Language Model Agents

Teaching Assistant · UC Berkeley

CS 194-196 · August 2024–May 2025

Assignments and technical materials on tools, planning, memory, and multi-agent coordination; student project advising; Berkeley RDI LLM Agents Hackathon.