← Search

Shunyu Yao

33 accepted papers

2026

Hierarchical Representations for Cross-task Automated Heuristic Design using LLMs

ICML 2026poster

Designing heuristic algorithms for complex optimization problems is a time-consuming and expert-driven process. Recently, Automated Heuristic Design (AHD) using Large Language Models (LLMs) has shown significant promise for automating algorithm development. However, existing works mainly rely on pro…

Cited by 0SourceScholar
2026

LAMIC: Layout-Aware Multi-Image Composition via Scalability of Multimodal Diffusion Transformer

AAAI 2026technical

In controllable image synthesis, generating coherent and consistent images from multiple references with spatial layout awareness remains an open challenge. We propose LAMIC, a Layout-Aware Multi-Image Composition framework that, for the first time, extends single-reference diffusion models to multi

Cited by 0SourcePDFScholar
2026

Offline Multi-Objective Bandits: From Logged Data to Pareto-Optimal Policies

AAAI 2026technical

Offline policy learning from logged data is a critical paradigm for enabling effective decision-making without costly online exploration. However, its application has been largely confined to single-objective problems, a stark contrast to real-world scenarios where decision-making inherently involve

Cited by 0SourcePDFScholar
2026

URS: A Unified Neural Routing Solver for Cross-Problem Zero-Shot Generalization

ICML 2026poster

Multi-task neural routing solvers have emerged as a promising paradigm for their ability to solve multiple vehicle routing problems (VRPs) using a single model. However, existing neural solvers typically rely on predefined problem constraints or require per-problem fine-tuning, which substantially l…

Cited by 0SourceScholar
2025

Accelerating Adaptive Retrieval Augmented Generation via Instruction-Driven Representation Reduction of Retrieval Overlaps

ACL 2025finding

Retrieval-augmented generation (RAG) has emerged as a pivotal method for expanding the knowledge of large language models. To handle complex queries more effectively, researchers developed Adaptive-RAG (A-RAG) to enhance the generated quality through multiple interactions with external knowledge bas…

2025

Contextual Experience Replay for Self-Improvement of Language Agents

ACL 2025long

Large language model (LLM) agents have been applied to sequential decision-making tasks such as web navigation, but without any environment-specific experiences, they often fail in these complex tasks. Moreover, current LLM agents are not designed to continually learn from past experiences during in…

Cited by 0SourcePDFScholar
2025

HASH-RAG: Bridging Deep Hashing with Retriever for Efficient, Fine Retrieval and Augmented Generation

ACL 2025finding

Retrieval-Augmented Generation (RAG) encounters efficiency challenges when scaling to massive knowledge bases while preserving contextual relevance. We propose Hash-RAG, a framework that integrates deep hashing techniques with systematic optimizations to address these limitations. Our queries direct…

2025

Multi-Objective Evolution of Heuristic Using Large Language Model

AAAI 2025technical

Heuristics are commonly used to tackle various search and optimization problems. Design heuristics usually require tedious manual crafting with domain knowledge. Recent works have incorporated Large Language Models (LLMs) into automatic heuristic search, leveraging their powerful language and coding…

2025

Prompting Large Language Models to Tackle the Full Software Development Lifecycle: A Case Study

COLING 2025main

Recent advancements in large language models (LLMs) have significantly enhanced their coding capabilities. However, existing benchmarks predominantly focused on simplified or isolated aspects of coding, such as single-file code generation or repository issue debugging, falling short of measuring the…

2025

When Models Know More Than They Can Explain: Quantifying Knowledge Transfer in Human-AI Collaboration

NeurIPS 2025poster

As large language models (LLMs) increasingly serve as close collaborators for humans, it is crucial that they express their reasoning in ways that humans can understand and learn from. However, this capability remains relatively less understood and under-evaluated. To address this, we introduce a co…

Cited by 0SourceScholar
2025

{$\tau$}-bench: A Benchmark for \underline{T}ool-\underline{A}gent-\underline{U}ser Interaction in Real-World Domains

ICLR 2025poster

Existing benchmarks for language agents do not set them up to interact with human users or follow domain-specific rules, both of which are vital to safe and realistic deployment. We propose $\tau$-bench, a benchmark with two domains (retail and airline) emulating dynamic conversations between a user…

Cited by 2SourcePDFScholar
2024

COLLIE: Systematic Construction of Constrained Text Generation Tasks

ICLR 2024poster

Text generation under constraints have seen increasing interests in natural language processing, especially with the rapidly improving capabilities of large language models. However, existing benchmarks for constrained generation usually focus on fixed constraint types (e.g. generate a sentence cont…

2024

DPN: Decoupling Partition and Navigation for Neural Solvers of Min-max Vehicle Routing Problems

ICML 2024poster

The min-max vehicle routing problem (min-max VRP) traverses all given customers by assigning several routes and aims to minimize the length of the longest route. Recently, reinforcement learning (RL)-based sequential planning methods have exhibited advantages in solving efficiency and optimality. Ho…

2024

Referral Augmentation for Zero-Shot Information Retrieval

ACL 2024findings

We propose Referral-Augmented Retrieval (RAR), a simple technique that concatenates document indices with referrals: text from other documents that cite or link to the given document. We find that RAR provides significant performance gains for tasks across paper retrieval, entity retrieval, and open…

2024

SWE-agent: Agent-Computer Interfaces Enable Automated Software Engineering

NeurIPS 2024poster

Language model agents are increasingly being used to automate complicated tasks in digital environments. Just as humans benefit from powerful software applications, such as integrated development environments, for complex tasks like software engineering, we posit that language model agents represent…

Cited by 217SourcePDFScholar
2024

SWE-bench: Can Language Models Resolve Real-world Github Issues?

ICLR 2024oral

Language models have outpaced our ability to evaluate them effectively, but for their future development it is essential to study the frontier of their capabilities. We find real-world software engineering to be a rich, sustainable, and challenging testbed for evaluating the next generation of langu…

Cited by 500SourcePDFScholar
2023

InterCode: Standardizing and Benchmarking Interactive Coding with Execution Feedback

NeurIPS 2023poster

Humans write code in a fundamentally interactive manner and rely on constant execution feedback to correct errors, resolve ambiguities, and decompose tasks. While LLMs have recently exhibited promising coding capabilities, current coding benchmarks mostly consider a static instruction-to-code sequen…

2023

Personality Understanding of Fictional Characters during Book Reading

ACL 2023long

Comprehending characters’ personalities is a crucial aspect of story reading. As readers engage with a story, their understanding of a character evolves based on new events and information; and multiple fine-grained aspects of personalities can be perceived. This leads to a natural problem of situat…

2023

ReAct: Synergizing Reasoning and Acting in Language Models

ICLR 2023top-5%

While large language models (LLMs) have demonstrated impressive capabilities across tasks in language understanding and interactive decision making, their abilities for reasoning (e.g. chain-of-thought prompting) and acting (e.g. action plan generation) have primarily been studied as separate topics…

2023

Reflexion: language agents with verbal reinforcement learning

NeurIPS 2023poster

Large language models (LLMs) have been increasingly used to interact with external environments (e.g., games, compilers, APIs) as goal-driven agents. However, it remains challenging for these language agents to quickly and efficiently learn from trial-and-error as traditional reinforcement learning…

2023

Tree of Thoughts: Deliberate Problem Solving with Large Language Models

NeurIPS 2023oral

Language models are increasingly being deployed for general problem solving across a wide range of tasks, but are still confined to token-level, left-to-right decision-making processes during inference. This means they can fall short in tasks that require exploration, strategic lookahead, or where i…

2022

Linking Emergent and Natural Languages via Corpus Transfer

ICLR 2022spotlight

The study of language emergence aims to understand how human languages are shaped by perceptual grounding and communicative intent. Computational approaches to emergent communication (EC) predominantly consider referential games in limited domains and analyze the learned protocol within the game fra…

2022

Multi-Stage Episodic Control for Strategic Exploration in Text Games

ICLR 2022spotlight

Text adventure games present unique challenges to reinforcement learning methods due to their combinatorially large action spaces and sparse rewards. The interplay of these two factors is particularly demanding because large action spaces require extensive exploration, while sparse rewards provide l…

2022

TVShowGuess: Character Comprehension in Stories as Speaker Guessing

NAACL 2022long

We propose a new task for assessing machines’ skills of understanding fictional characters in narrative stories. The task, TVShowGuess, builds on the scripts of TV series and takes the form of guessing the anonymous main characters based on the backgrounds of the scenes and the dialogues. Our human…

2022

WebShop: Towards Scalable Real-World Web Interaction with Grounded Language Agents

NeurIPS 2022accept

Most existing benchmarks for grounding language in interactive environments either lack realistic linguistic elements, or prove difficult to scale up due to substantial human involvement in the collection of data or feedback signals. We develop WebShop – a simulated e-commerce website environment wi…

2021

Dynamic Context-Sensitive Filtering Network for Video Salient Object Detection

ICCV 2021poster

The ability to capture inter-frame dynamics has been critical to the development of video salient object detection (VSOD). While many works have achieved great success in this field, a deeper insight into its dynamic nature should be developed. In this work, we aim to answer the following questions:…

Cited by 128PDFcodeScholar
2021

Reading and Acting while Blindfolded: The Need for Semantics in Text Game Agents

NAACL 2021long

Text-based games simulate worlds and interact with players using natural language. Recent work has used them as a testbed for autonomous language-understanding agents, with the motivation being that understanding the meanings of words or semantics is a key component of how humans understand, reason,…

Cited by 28SourcePDFScholar
2021

Self-Attention Networks Can Process Bounded Hierarchical Languages

ACL 2021long

Despite their impressive performance in NLP, self-attention networks were recently proved to be limited for processing formal languages with hierarchical structure, such as Dyck-k, the language consisting of well-nested parentheses of k types. This suggested that natural language can be approximated…

2019

Modeling Expectation Violation in Intuitive Physics with Coarse Probabilistic Object Representations

NeurIPS 2019poster

From infancy, humans have expectations about how objects will move and interact. Even young children expect objects not to move through one another, teleport, or disappear. They are surprised by mismatches between physical expectations and perceptual observations, even in unfamiliar scenes with comp…

2018

3D-Aware Scene Manipulation via Inverse Graphics

NeurIPS 2018poster

We aim to obtain an interpretable, expressive, and disentangled scene representation that contains comprehensive structural and textural information for each object. Previous scene representations learned by neural networks are often uninterpretable, limited to a single object, or lacking 3D knowled…