← Search

Jiaxuan You

42 accepted papers

2026

ExpWeaver: LLM Agents Learn from Experience via Latent RAG

ICML 2026poster

Experience learning has achieved promising results in enhancing LLM agent planning and reasoning by integrating past interactions as reusable knowledge. However, existing methods remain confined to explicit text space---retrieving experiences via semantic similarity and concatenating them into the c…

Cited by 0SourceScholar
2026

GraphPlanner: Graph-Based Agentic Routing for LLMs

ICLR 2026poster

LLM routing has achieved promising results in integrating the strengths of di- verse models while balancing efficiency and performance. However, to support more realistic and challenging applications, routing must extend into agentic LLM settings—where task planning, multi-round cooperation among he…

Cited by 0SourcecodeScholar
2026

Learning Query-Aware Budget-Tier Routing for Runtime Agent Memory

ICML 2026poster

Memory is increasingly central to Large Language Model (LLM) agents operating beyond a single context window, yet most existing systems rely on offline, query-agnostic memory construction that can be inefficient and may discard query-critical information. Although runtime memory utilization is a nat…

Cited by 0SourceScholar
2026

Probing the Knowledge Boundary: An Interactive Agentic Framework for Deep Knowledge Extraction

ICML 2026poster

Large Language Models (LLMs) can be seen as compressed knowledge bases, but it remains unclear what knowledge they truly contain and how far their knowledge boundaries extend. Existing benchmarks are mostly static and provide limited support for systematic knowledge probing. In this paper, we propos…

Cited by 0SourceScholar
2026

Relational In-Context Learning via Synthetic Pre-training with Structural Prior

ICML 2026poster

Relational Databases (RDBs) are the backbone of modern business, yet they have missed the Foundation Model revolution. Unlike text or images, high-quality RDB data is private and scarce, rendering the standard approach of ``pre-training on the internet'' infeasible. Consequently, existing solutions …

Cited by 0SourceScholar
2026

Self-Aligned Reward: Towards Effective and Efficient Reasoners

ICLR 2026poster

Reinforcement learning with verifiable rewards has significantly advanced reasoning with large language models (LLMs) in domains such as mathematics and logic. However, verifiable signals provide only coarse-grained or binary correctness feedback. This limitation results in inefficiencies like overl…

Cited by 0SourceScholar
2026

h-MINT: Modeling Pocket-Ligand Binding with Hierarchical Molecular Interaction Network

ICLR 2026poster

Accurate molecular representations are critical for drug discovery, and a central challenge lies in capturing the chemical environment of molecular fragments, as key interactions, such as H-bond and π stacking—occur only under specific local conditions. Most existing approaches represent molecules a…

Cited by 0SourcecodeScholar
2025

ConsistencyChecker: Tree-based Evaluation of LLM Generalization Capabilities

ACL 2025long

Evaluating Large Language Models (LLMs) requires effective methods to assess semantic consistency across multiple reversible transformations. Traditional self-consistency methods often fail to capture subtle semantic errors in multi-step tasks. We introduce ConsistencyChecker, a tree-based evaluatio…

2025

DeFT: Decoding with Flash Tree-attention for Efficient Tree-structured LLM Inference

ICLR 2025spotlight

Large language models (LLMs) are increasingly employed for complex tasks that process multiple generation calls in a tree structure with shared prefixes of tokens, including few-shot prompting, multi-step reasoning, speculative decoding, etc. However, existing inference systems for tree-based applic…

2025

Graph of Records: Boosting Retrieval Augmented Generation for Long-context Summarization with Graphs

ACL 2025long

Retrieval-augmented generation (RAG) has revitalized Large Language Models (LLMs) by injecting non-parametric factual knowledge. Compared with long-context LLMs, RAG is considered an effective summarization tool in a more concise and lightweight manner, which can interact with LLMs multiple times us…

2025

MultiAgentBench : Evaluating the Collaboration and Competition of LLM agents

ACL 2025long

Large Language Models (LLMs) have shown remarkable capabilities as autonomous agents; yet existing benchmarks either focus on single-agent tasks or are confined to narrow domains, failing to capture the dynamics of multi-agent coordination and competition. In this paper, we introduce MultiAgentBench…

2025

ResearchTown: Simulator of Human Research Community

ICML 2025poster

Large Language Models (LLMs) have demonstrated remarkable potential in scientific domains, yet a fundamental question remains unanswered: Can we simulate human research communities with LLMs? Addressing this question can deepen our understanding of the processes behind idea brainstorming and inspire…

2025

Router-R1: Teaching LLMs Multi-Round Routing and Aggregation via Reinforcement Learning

NeurIPS 2025poster

The rapid emergence of diverse large language models (LLMs) has spurred the development of LLM routers that assign user queries to the most suitable model. However, existing LLM routers typically perform a single-round, one-to-one mapping (\textit{i.e.}, assigning each query to a single model in iso…

Cited by 0SourcecodeScholar
2025

SafeScientist: Enhancing AI Scientist Safety for Risk-Aware Scientific Discovery

EMNLP 2025

Recent advancements in large language model (LLM) agents have significantly accelerated scientific discovery automation, yet concurrently raised critical ethical and safety concerns. To systematically address these challenges, we introduce **SafeScientist**, an innovative AI scientist framework expl

2024

Arxiv Copilot: A Self-Evolving and Efficient LLM System for Personalized Academic Assistance

EMNLP 2024system demonstrations

As scientific research proliferates, researchers face the daunting task of navigating and reading vast amounts of literature. Existing solutions, such as document QA, fail to provide personalized and up-to-date information efficiently. We present Arxiv Copilot, a self-evolving, efficient LLM system…

Cited by 0SourcePDFScholar
2024

Deep Reinforcement Learning for Modelling Protein Complexes

ICLR 2024poster

Structure prediction of large protein complexes (a.k.a., protein multimer mod- elling, PMM) can be achieved through the one-by-one assembly using provided dimer structures and predicted docking paths. However, existing PMM methods struggle with vast search spaces and generalization challenges: (1) T…

Cited by 1SourcePDFScholar
2024

In-Context Learning May Not Elicit Trustworthy Reasoning: A-Not-B Errors in Pretrained Language Models

EMNLP 2024finding

Recent advancements in artificial intelligence have led to the creation of highly capable large language models (LLMs) that can perform tasks in a human-like manner. However, LLMs exhibit only infant-level cognitive abilities in certain areas. One such area is the A-Not-B error, a phenomenon seen in…

2024

InstructG2I: Synthesizing Images from Multimodal Attributed Graphs

NeurIPS 2024poster

In this paper, we approach an overlooked yet critical task Graph2Image: generating images from multimodal attributed graphs (MMAGs). This task poses significant challenges due to the explosion in graph size, dependencies among graph entities, and the need for controllability in graph conditions. To…

2024

LLM-Evolve: Evaluation for LLM’s Evolving Capability on Benchmarks

EMNLP 2024main

The advancement of large language models (LLMs) has extended their use to dynamic and interactive real-world applications, where models engage continuously with their environment and potentially enhance their performance over time. Most existing LLM benchmarks evaluate LLMs on i.i.d. tasks, overlook…

Cited by 1SourcePDFScholar
2024

Position: Relational Deep Learning - Graph Representation Learning on Relational Databases

ICML 2024poster

Much of the world's most valued data is stored in relational databases and data warehouses, where the data is organized into tables connected by primary-foreign key relations. However, building machine learning models using this data is both challenging and time consuming because no ML algorithm can…

Cited by 12SourcePDFScholar
2024

RankRAG: Unifying Context Ranking with Retrieval-Augmented Generation in LLMs

NeurIPS 2024poster

Large language models (LLMs) typically utilize the top-k contexts from a retriever in retrieval-augmented generation (RAG). In this work, we propose a novel method called RankRAG, which instruction-tunes a single LLM for both context ranking and answer generation in RAG. In particular, the instruc…

Cited by 18SourcePDFScholar
2023

AutoTransfer: AutoML with Knowledge Transfer - An Application to Graph Neural Networks

ICLR 2023poster

AutoML has demonstrated remarkable success in finding an effective neural architecture for a given machine learning task defined by a specific dataset and an evaluation metric. However, most present AutoML techniques consider each task independently from scratch, which requires exploring many archit…

2022

Relational Multi-Task Learning: Modeling Relations between Data and Tasks

ICLR 2022spotlight

A key assumption in multi-task learning is that at the inference time the multi-task model only has access to a given data point but not to the data point’s labels from other tasks. This presents an opportunity to extend multi-task learning to utilize data point’s labels from other auxiliary tasks,…

2020

Handling Missing Data with Graph Representation Learning

NeurIPS 2020poster

Machine learning with missing data has been approached in many different ways, including feature imputation where missing feature values are estimated based on observed values and label prediction where downstream labels are learned directly from incomplete data. However, existing imputation models…

2019

G2SAT: Learning to Generate SAT Formulas

NeurIPS 2019poster

The Boolean Satisfiability (SAT) problem is the canonical NP-complete problem and is fundamental to computer science, with a wide array of applications in planning, verification, and theorem proving. Developing and evaluating practical SAT solvers relies on extensive empirical testing on a set of re…

Cited by 54SourcePDFScholar
2019

GNNExplainer: Generating Explanations for Graph Neural Networks

NeurIPS 2019poster

Graph Neural Networks (GNNs) are a powerful tool for machine learning on graphs.GNNs combine node feature information with the graph structure by recursively passing neural messages along edges of the input graph. However, incorporating both graph structure and feature information leads to complex…

2018

Graph Convolutional Policy Network for Goal-Directed Molecular Graph Generation

NeurIPS 2018spotlight

Generating novel graph structures that optimize given objectives while obeying some given underlying rules is fundamental for chemistry, biology and social science research. This is especially important in the task of molecular graph generation, whose goal is to discover novel molecules with desired…

2018

GraphRNN: Generating Realistic Graphs with Deep Auto-regressive Models

ICML 2018oral

Modeling and generating graphs is fundamental for studying networks in biology, engineering, and social sciences. However, modeling complex distributions over graphs and then efficiently sampling from these distributions is challenging due to the non-unique, high-dimensional nature of graphs and the…

2018

Hierarchical Graph Representation Learning with Differentiable Pooling

NeurIPS 2018spotlight

Recently, graph neural networks (GNNs) have revolutionized the field of graph representation learning through effectively learned node embeddings, and achieved state-of-the-art results in tasks such as node classification and link prediction. However, current GNN methods are inherently flat and do n…

Cited by 2077SourcePDFScholar