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. Aniket Bera and Prof. Lin Tan. My research focuses on physical AI and robot learning, particularly foundation models for robotics and reliable robotic systems. Previously, I worked on coding agents and code generation.

Research Interests

  • Physical AI & Robotics

    World Action Models, Vision-Language-Action Models, Robot Learning, and Safe Autonomy.

  • Generative AI

    LLM Planners, Agentic AI, Reinforcement Learning, and Preference Optimization.

Selected Publications

COCOA real-robot rollouts: clear the table, stack the cubes, plate the banana
Coming Soon | 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 corrective VLA policy that refines the action chunks proposed by a frozen flow-based VLA before execution, trained with SFT followed by flow-based correction preference optimization. Consistent performance gains across three VLA backbones π0.5, FLOWER VLA, and Xiaomi-Robotics-0.

SELP versus prior work on a UAV drone, robot dog, and robot arm: safety and time cost

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

NL-to-LTL Translation, LTL-constrained decoding, and domain-specific fine-tuning make LLM planners follow complex and conditional user instructions, improving safety rate by 10.8% and plan efficiency by 19.8% on robot manipulation and drone navigation tasks.

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

Analyzing and Mitigating Cost-Inefficient Behaviors in Coding Agents

Yiran Hu, Nan Jiang, Shanchao Liang, Anik Dey, Yi Wu, Lin Tan

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.

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.

Work Experience

Amazon Web Services

, Applied Scientist Intern
Summer 2025

JPMorgan Chase & Co.

, AI Research Intern
Summer 2024

Education

Ph.D. in Computer Science, Purdue University

Advisors: Prof. Aniket Bera and Prof. Lin Tan

West Lafayette, IN, USA

2021 – Present

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

Shenzhen, China

2017 – 2021

Recognition & Press

  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.