← Search

Boshi Wang

13 accepted papers

2026

Improving Code Localization with Repository Memory

ICLR 2026poster

Code localization is a fundamental challenge in repository-level software engineering tasks such as bug fixing. While existing methods equip language agents with comprehensive tools/interfaces to fetch information from the repository, they overlook the critical aspect of *memory*, where each instanc…

Cited by 0SourceScholar
2026

Is the Reversal Curse a Binding Problem? Uncovering Limitations of Transformers from a Basic Generalization Failure

ICLR 2026poster

Despite their impressive capabilities, LLMs exhibit a basic generalization failure known as the *Reversal Curse*, where they struggle to learn reversible factual associations. Understanding why this occurs could help identify weaknesses in current models and advance their generalization and robustne…

Cited by 0SourcecodeScholar
2025

ScienceAgentBench: Toward Rigorous Assessment of Language Agents for Data-Driven Scientific Discovery

ICLR 2025poster

The advancements of language language models (LLMs) have piqued growing interest in developing LLM-based language agents to automate scientific discovery end-to-end, which has sparked both excitement and skepticism about the true capabilities of such agents. In this work, we argue that for an agent…

Cited by 21SourcePDFScholar
2024

Grokking of Implicit Reasoning in Transformers: A Mechanistic Journey to the Edge of Generalization

NeurIPS 2024poster

We study whether transformers can learn to *implicitly* reason over parametric knowledge, a skill that even the most capable language models struggle with. Focusing on two representative reasoning types, composition and comparison, we consistently find that transformers *can* learn implicit reasonin…

Cited by 3SourcePDFScholar
2024

How Trustworthy are Open-Source LLMs? An Assessment under Malicious Demonstrations Shows their Vulnerabilities

NAACL 2024long

The rapid progress in open-source Large Language Models (LLMs) is significantly driving AI development forward. However, there is still a limited understanding of their trustworthiness. Deploying these models at scale without sufficient trustworthiness can pose significant risks, highlighting the ne…

2024

LLMs in the Imaginarium: Tool Learning through Simulated Trial and Error

ACL 2024long

Tools are essential for large language models (LLMs) to acquire up-to-date information and take consequential actions in external environments. Existing work on tool-augmented LLMs primarily focuses on the broad coverage of tools and the flexibility of adding new tools. However, a critical aspect th…

2023

Automatic Evaluation of Attribution by Large Language Models

EMNLP 2023long findings

A recent focus of large language model (LLM) development, as exemplified by generative search engines, is to incorporate external references to generate and support its claims. However, evaluating the attribution, i.e., verifying whether the generated statement is fully supported by the cited refere…

Cited by 0SourcecodeScholar
2023

Mind2Web: Towards a Generalist Agent for the Web

NeurIPS 2023spotlight

We introduce Mind2Web, the first dataset for developing and evaluating generalist agents for the web that can follow language instructions to complete complex tasks on any website. Existing datasets for web agents either use simulated websites or only cover a limited set of websites and tasks, thus…

2023

Towards Understanding Chain-of-Thought Prompting: An Empirical Study of What Matters

ACL 2023long

Chain-of-Thought (CoT) prompting can dramatically improve the multi-step reasoning abilities of large language models (LLMs). CoT explicitly encourages the LLM to generate intermediate rationales for solving a problem, by providing a series of reasoning steps in the demonstrations. Despite its succe…

2022

Automatic Loss Function Search for Predict-Then-Optimize Problems with Strong Ranking Property

ICLR 2022poster

Combinatorial optimization problems with parameters to be predicted from side information are commonly seen in a variety of problems during the paradigm shift from reactive decision making to proactive decision making. Due to the misalignment between the continuous prediction results and the discret…

Cited by 2SourcePDFScholar
2021

Predictive Job Scheduling under Uncertain Constraints in Cloud Computing

IJCAI 2021poster

Capacity management has always been a great challenge for cloud platforms due to massive, heterogeneous on-demand instances running at different times. To better plan the capacity for the whole platform, a class of cloud computing instances have been released to collect computing demands beforehand.…

Cited by 7SourcePDFScholar