← Search

Yunzhe Qi

9 accepted papers

2026

AutoTool: Dynamic Tool Selection and Integration for Agentic Reasoning

ICML 2026poster

Agentic reinforcement learning has advanced large language models (LLMs) to reason through long chain-of-thought trajectories while interleaving external tool use. Existing approaches assume a fixed inventory of tools, which limits the adaptability of LLM agents to new or evolving toolsets. We prese…

Cited by 0SourceScholar
2026

Influence-Preserving Proxies for Gradient-Based Data Selection in LLM FineTuning

ICLR 2026poster

Supervised fine-tuning (SFT) relies critically on selecting training data that most benefits model's downstream performance. Gradient-based data selection methods such as TracIn and Influence Functions leverage influence to identify useful samples, but their computational cost scales poorly, making…

Cited by 0SourcecodeScholar
2025

Learning to Instruct: Fine-Tuning a Task-Aware Instruction Optimizer for Black-Box LLMs

EMNLP 2025

The performance of Large Language Models (LLMs) critically depends on designing effective instructions, which is particularly challenging for black-box LLMs with inaccessible internal states. To this end, we introduce Learning to Instruct , a novel paradigm that formulates instruction optimization a

Cited by 0SourcePDFScholar
2025

Transformer Copilot: Learning from The Mistake Log in LLM Fine-tuning

NeurIPS 2025spotlight

Large language models are typically adapted to downstream tasks through supervised fine-tuning on domain-specific data. While standard fine-tuning focuses on minimizing generation loss to optimize model parameters, we take a deeper step by retaining and leveraging the model’s own learning signals, a…

Cited by 0SourcecodeScholar
2025

Unlocking Efficient, Scalable, and Continual Knowledge Editing with Basis-Level Representation Fine-Tuning

ICLR 2025poster

Large language models (LLMs) have achieved remarkable performance on vari- ous natural language tasks. However, they are trained on static corpora and their knowledge can become outdated quickly in the fast-changing world. This moti- vates the development of knowledge editing methods designed to upd…

Cited by 1SourcePDFScholar
2024

PageRank Bandits for Link Prediction

NeurIPS 2024poster

Link prediction is a critical problem in graph learning with broad applications such as recommender systems and knowledge graph completion. Numerous research efforts have been directed at solving this problem, including approaches based on similarity metrics and Graph Neural Networks (GNN). However,…

2024

Robust Neural Contextual Bandit against Adversarial Corruptions

NeurIPS 2024poster

Contextual bandit algorithms aim to identify the optimal arm with the highest reward among a set of candidates, based on the accessible contextual information. Among these algorithms, neural contextual bandit methods have shown generally superior performances against linear and kernel ones, due to t…

Cited by 0SourcePDFScholar
2023

Meta-Learning with Neural Bandit Scheduler

NeurIPS 2023poster

Meta-learning has been proven an effective learning paradigm for training machine learning models with good generalization ability. Apart from the common practice of uniformly sampling the meta-training tasks, existing methods working on task scheduling strategies are mainly based on pre-defined sam…