← Search

Zhiruo Wang

8 accepted papers

2026

TOM-SWE: User Mental Modeling For Software Engineering Agents

ICML 2026poster

Recent advances in coding agents have made them capable of planning, editing, running, and testing complex code bases. Despite their growing ability in coding tasks, these systems still struggle to infer and track user intent, especially when instructions are underspecified or context-dependent. To …

Cited by 0SourceScholar
2024

ECCO: Can We Improve Model-Generated Code Efficiency Without Sacrificing Functional Correctness?

EMNLP 2024main

Although large language models (LLMs) have been largely successful in generating functionally correct programs, conditioning models to produce efficient solutions while ensuring correctness remains a challenge. Further, unreliability in benchmarking code efficiency is a hurdle across varying hardwar…

2024

TroVE: Inducing Verifiable and Efficient Toolboxes for Solving Programmatic Tasks

ICML 2024poster

Language models (LMs) can solve tasks such as answering questions about tables or images by writing programs. However, using primitive functions often leads to verbose and error-prone programs, and higher-level functions require expert design. To enable better solutions without human labor, we ask c…

2023

API-Assisted Code Generation for Question Answering on Varied Table Structures

EMNLP 2023long main

A persistent challenge to table question answering (TableQA) by generating executable programs has been adapting to varied table structures, typically requiring domain-specific logical forms. In response, this paper introduces a unified TableQA framework that: (1) provides a unified representation f…

Cited by 0SourceScholar
2023

Execution-Based Evaluation for Open-Domain Code Generation

EMNLP 2023long findings

To extend the scope of coding queries to more realistic settings, we propose ODEX, the first Open-Domain EXecution-based natural language (NL) to Python code generation dataset. ODEX has 945 NL-Code pairs spanning 79 diverse libraries, along with 1,707 human-written test cases for execution. Our NL-…

Cited by 0SourcecodeScholar
2023

SPAE: Semantic Pyramid AutoEncoder for Multimodal Generation with Frozen LLMs

NeurIPS 2023spotlight

In this work, we introduce Semantic Pyramid AutoEncoder (SPAE) for enabling frozen LLMs to perform both understanding and generation tasks involving non-linguistic modalities such as images or videos. SPAE converts between raw pixels and interpretable lexical tokens (or words) extracted from the LLM…

Cited by 59SourcePDFScholar
2022

HiTab: A Hierarchical Table Dataset for Question Answering and Natural Language Generation

ACL 2022long

Tables are often created with hierarchies, but existing works on table reasoning mainly focus on flat tables and neglect hierarchical tables. Hierarchical tables challenge numerical reasoning by complex hierarchical indexing, as well as implicit relationships of calculation and semantics. We present…

2022

Retrieval as Attention: End-to-end Learning of Retrieval and Reading within a Single Transformer

EMNLP 2022main

Systems for knowledge-intensive tasks such as open-domain question answering (QA) usually consist of two stages: efficient retrieval of relevant documents from a large corpus and detailed reading of the selected documents. This is usually done through two separate models, a retriever that encodes th…