← Search

Shaokun Zhang

10 accepted papers

2026

Golden Goose: A Simple Trick to Synthesize Unlimited RLVR Tasks from Unverifiable Internet Text

ICML 2026poster

Reinforcement Learning with Verifiable Rewards (RLVR) has become a cornerstone for unlocking complex reasoning in Large Language Models (LLMs). Yet, scaling up RL is bottlenecked by limited existing verifiable data, where improvements increasingly saturate over prolonged training. To overcome this, …

Cited by 0SourceScholar
2026

Nemotron-Research-Tool-N1: Exploring Tool-Using Language Models with Reinforced Reasoning

ICLR 2026poster

Enabling large language models with external tools has become a pivotal strategy for extending their functionality beyond text space. To enhance LLMs' tool-calling abilities, previous approaches primarily rely on supervised fine-tuning (SFT) with trajectories distilled from stronger models, often re…

Cited by 0SourcecodeScholar
2025

BEST-Route: Adaptive LLM Routing with Test-Time Optimal Compute

ICML 2025poster

Large language models (LLMs) are powerful tools but are often expensive to deploy at scale. LLM query routing mitigates this by dynamically assigning queries to models of varying cost and quality to obtain a desired tradeoff. Prior query routing approaches generate only one response from the select…

Cited by 0SourcePDFScholar
2025

Divide, Optimize, Merge: Scalable Fine-Grained Generative Optimization for LLM Agents

EMNLP 2025

LLM-based optimization has shown remarkable potential in improving agentic systems. However, the conventional approach of prompting LLM-based generative optimizer with the trajectories on the whole training dataset in a single pass becomes untenable as datasets grow, leading to context window overfl

Cited by 0SourcePDFScholar
2025

Which Agent Causes Task Failures and When? On Automated Failure Attribution of LLM Multi-Agent Systems

ICML 2025spotlight

Failure attribution in LLM multi-agent systems—identifying the agent and step responsible for task failures—provides crucial clues for systems debugging but remains underexplored and labor-intensive. In this paper, we propose and formulate a new research area: automated failure attribution for LLM…

2024

A Novel Framework for Structure Descriptors-Guided Hand-drawn Floor Plan Reconstruction

IROS 2024poster

In the absence of a pre-built indoor map, robot navigation suffers from the limitations of sensors and environments, resulting in decreased efficiency in performing ad-hoc tasks. Given that blueprints are difficult to obtain, an intuitive method is to provide robots with prior knowledge via hand-dra…

Cited by 0SourceScholar
2024

IDEAL: Influence-Driven Selective Annotations Empower In-Context Learners in Large Language Models

ICLR 2024poster

In-context learning is a promising paradigm that utilizes in-context examples as prompts for the predictions of large language models. These prompts are crucial for achieving strong performance. However, since the prompts need to be sampled from a large volume of annotated examples, finding the righ…

Cited by 28SourcePDFScholar
2024

Offline Training of Language Model Agents with Functions as Learnable Weights

ICML 2024poster

Researchers and practitioners have recently reframed powerful Large Language Models (LLMs) as *agents*, enabling them to automate complex tasks largely via the use of specialized functions. To facilitate the development of LLM agents, we present a novel paradigm of training LLM agents without modify…

Cited by 16SourcePDFScholar
2024

Refined Coreset Selection: Towards Minimal Coreset Size under Model Performance Constraints

ICML 2024spotlight

Coreset selection is powerful in reducing computational costs and accelerating data processing for deep learning algorithms. It strives to identify a small subset from large-scale data, so that training only on the subset practically performs on par with full data. Practitioners regularly desire to…

2023

Targeted Hyperparameter Optimization with Lexicographic Preferences Over Multiple Objectives

ICLR 2023top-5%

Motivated by various practical applications, we propose a novel and general formulation of targeted multi-objective hyperparameter optimization. Our formulation allows a clear specification of an automatable optimization goal using lexicographic preference over multiple objectives. We then propose a…

Cited by 33SourcePDFScholar