← Search

Yiheng Shu

9 accepted papers

2026

REMem: Reasoning with Episodic Memory in Language Agent

ICLR 2026poster

Humans excel at remembering concrete experiences along spatiotemporal contexts and performing reasoning across those events, i.e., the capacity for episodic memory. In contrast, memory in language agents remains mainly semantic, and current agents are not yet capable of effectively recollecting and…

Cited by 0SourcecodeScholar
2025

From RAG to Memory: Non-Parametric Continual Learning for Large Language Models

ICML 2025poster

Our ability to continuously acquire, organize, and leverage knowledge is a key feature of human intelligence that AI systems must approximate to unlock their full potential. Given the challenges in continual learning with large language models (LLMs), retrieval-augmented generation (RAG) has become…

2025

Mind2Web 2: Evaluating Agentic Search with Agent-as-a-Judge

NeurIPS 2025poster

Agentic search such as Deep Research systems-where agents autonomously browse the web, synthesize information, and return comprehensive citation-backed answers-represents a major shift in how users interact with web-scale information. While promising greater efficiency and cognitive offloading, the…

Cited by 0SourceScholar
2025

Navigating the Digital World as Humans Do: Universal Visual Grounding for GUI Agents

ICLR 2025oral

Multimodal large language models (MLLMs) are transforming the capabilities of graphical user interface (GUI) agents, facilitating their transition from controlled simulations to complex, real-world applications across various platforms. However, the effectiveness of these agents hinges on the robust…

2024

HippoRAG: Neurobiologically Inspired Long-Term Memory for Large Language Models

NeurIPS 2024poster

In order to thrive in hostile and ever-changing natural environments, mammalian brains evolved to store large amounts of knowledge about the world and continually integrate new information while avoiding catastrophic forgetting. Despite the impressive accomplishments, large language models (LLMs), e…

2024

Middleware for LLMs: Tools Are Instrumental for Language Agents in Complex Environments

EMNLP 2024main

The applications of large language models (LLMs) have expanded well beyond the confines of text processing, signaling a new era where LLMs are envisioned as generalist agents capable of operating within complex environments. These environments are often highly expansive, making it impossible for the…

Cited by 31SourcePDFScholar
2023

Question Decomposition Tree for Answering Complex Questions over Knowledge Bases

AAAI 2023technical

Knowledge base question answering (KBQA) has attracted a lot of interest in recent years, especially for complex questions which require multiple facts to answer. Question decomposition is a promising way to answer complex questions. Existing decomposition methods split the question into sub-questio…

2022

Logical Form Generation via Multi-task Learning for Complex Question Answering over Knowledge Bases

COLING 2022main

Question answering over knowledge bases (KBQA) for complex questions is a challenging task in natural language processing. Recently, generation-based methods that translate natural language questions to executable logical forms have achieved promising performance. These methods use auxiliary informa…

2022

TIARA: Multi-grained Retrieval for Robust Question Answering over Large Knowledge Base

EMNLP 2022main

Pre-trained language models (PLMs) have shown their effectiveness in multiple scenarios. However, KBQA remains challenging, especially regarding coverage and generalization settings. This is due to two main factors: i) understanding the semantics of both questions and relevant knowledge from the KB;…