Yi Wu

Yi Wu

PhD Student · Computer Science · Purdue University

Yi Wu

About

I am a Ph.D. student in the Department of Computer Science at Purdue University, co-advised by Prof. Lin Tan and Prof. Aniket Bera. I received my B.E. in Computer Science from the Southern University of Science and Technology, where I worked with Prof. Shin Hwei Tan.

Research Interests

  • Physical AI & Robotics

    Vision-Language-Action Policies, Action Correction, and Safe Robot Task Planning.

  • Generative AI

    LLM Planners, Coding Agents, and Preference Optimization.

Selected Publications

COCOA real-robot rollouts: clear the table, stack the cubes, plate the banana
Preprint 2026

COCOA: Action-Conditioned Correction for Vision-Language-Action Policies

Yi Wu, Manav Kulshrestha, Junyi Wu, Kai Cheng, Nan Jiang, Aniket Bera, Lin Tan

A separate corrective policy that revises the action chunks proposed by a frozen flow-based VLA before execution, trained with supervised flow matching followed by correction preference optimization. Consistent gains on SimplerEnv, CALVIN, and real-robot tasks across π0.5, FLOWER VLA, and Xiaomi-Robotics-0.

Cost-inefficient behaviors found in a Claude Code trajectory
Preprint 2026

Analyzing and Mitigating Cost-Inefficient Behaviors in Coding Agents

Yiran Hu, Yi Wu, et al.

A study of 1,200 Claude Code and Mini-SWE-Agent trajectories on SWE-bench that identifies three recurring cost-inefficient behaviors, then shows developer-designed skills cut task cost by up to 41.73%, roughly twice the gain of agent-synthesized skills.

SELP motivating example: LTL-constrained drone navigation plan

SELP: Generating Safe and Efficient Task Plans for Robot Agents with Large Language Models

Yi Wu, Zikang Xiong, Yiran Hu, Shreyash S. Iyengar, Nan Jiang, Aniket Bera, Lin Tan, Suresh Jagannathan

Equivalence voting over LTL formulas, LTL-constrained decoding, and domain-specific fine-tuning make LLM planners follow complex user constraints, improving safety rate by 10.8% and plan efficiency by 19.8% on drone navigation.

Earlier Work

  • ISSTA 2023

    How Effective Are Neural Networks for Fixing Security Vulnerabilities. Yi Wu, Nan Jiang, Hung Viet Pham, Thibaud Lutellier, Jordan Davis, Lin Tan, Petr Babkin, Sameena Shah.

  • ASE SRC 2020

    Anti-patterns for Java Automated Program Repair Tools. Yi Wu. Won 1st Place.

Experience

Applied Scientist Intern, Amazon Web Services

Summer 2025

AI Research Associate Intern, JPMorgan Chase & Co.

Summer 2024

Education

Ph.D. in Computer Science, Purdue University

Advisors: Prof. Lin Tan and Prof. Aniket Bera

2021 – Present

B.E. in Computer Science, Southern University of Science and Technology

Advisor: Prof. Shin Hwei Tan

2017 – 2021

Academic Services

Conference Reviewer
ICRA 2026
Journal Reviewer
IEEE TSE 2025

Press Coverage

  1. 2025

    Purdue CS News: SELP named a Best Paper Award Finalist at ICRA 2025.

  2. 2020

    SUSTech CS News: First place in the ACM Student Research Competition at ASE 2020.