← Search

Yujia Bao

12 accepted papers

2026

Adaptive Data Augmentation with Multi-armed Bandit: Sample-Efficient Embedding Calibration for Implicit Pattern Recognition

CVPR 2026

Recognizing visual and textual patterns is essential in many real-world applications of modern AI. However, tackling long-tail pattern recognition tasks remains challenging for current pre-trained foundation models such as LLMs and VLMs. While finetuning pre-trained models can improve accuracy in re

Cited by 0SourceScholar
2026

DRAGON: Guard LLM Unlearning in Context via Negative Detection and Reasoning

ICLR 2026poster

Unlearning in Large Language Models (LLMs) is crucial for protecting private data and removing harmful knowledge. Most existing approaches rely on fine-tuning to balance unlearning efficiency with general language capabilities. However, these methods typically require training or access to retain da…

Cited by 0SourcecodeScholar
2026

MCP-Bench: Benchmarking Tool-Using LLM Agents with Complex Real-World Tasks via MCP Servers

ICLR 2026poster

We introduce MCP-Bench, a benchmark for evaluating large language models (LLMs) on realistic, multi-step tasks that demand tool use, cross-tool coordination, precise parameter control, and planning/reasoning for solving tasks. Built on the Model Context Protocol (MCP), MCP-Bench connects LLMs to 28…

Cited by 0SourcecodeScholar
2025

From Isolated Conversations to Hierarchical Schemas: Dynamic Tree Memory Representation for LLMs

ICLR 2025poster

Recent advancements in large language models have significantly improved their context windows, yet challenges in effective long-term memory management remain. We introduce MemTree, an algorithm that leverages a dynamic, tree-structured memory representation to optimize the organization, retrieval,…

Cited by 3SourcePDFScholar
2025

Improving Data Efficiency via Curating LLM-Driven Rating Systems

ICLR 2025poster

Instruction tuning is critical for adapting large language models (LLMs) to downstream tasks, and recent studies have demonstrated that small amounts of human-curated data can outperform larger datasets, challenging traditional data scaling laws. While LLM-based data quality rating systems offer a c…

Cited by 3SourcePDFScholar
2025

KVLink: Accelerating Large Language Models via Efficient KV Cache Reuse

NeurIPS 2025poster

We describe KVLink, an approach for efficient key-value (KV) cache reuse in large language models (LLMs). In many LLM applications, different inputs can share overlapping context, such as the same retrieved document appearing in multiple queries. However, the LLMs still need to encode the entire con…

Cited by 0SourcecodeScholar
2025

LLM Unlearning via Loss Adjustment with Only Forget Data

ICLR 2025poster

Unlearning in Large Language Models (LLMs) is essential for ensuring ethical and responsible AI use, especially in addressing privacy leak, bias, safety, and evolving regulations. Existing approaches to LLM unlearning often rely on retain data or a reference LLM, yet they struggle to adequately bala…

Cited by 2SourcePDFScholar
2024

Channel Vision Transformers: An Image Is Worth 1 x 16 x 16 Words

ICLR 2024poster

Vision Transformer (ViT) has emerged as a powerful architecture in the realm of modern computer vision. However, its application in certain imaging fields, such as microscopy and satellite imaging, presents unique challenges. In these domains, images often contain multiple channels, each carrying se…

2022

Learning Stable Classifiers by Transferring Unstable Features

ICML 2022spotlight

While unbiased machine learning models are essential for many applications, bias is a human-defined concept that can vary across tasks. Given only input-label pairs, algorithms may lack sufficient information to distinguish stable (causal) features from unstable (spurious) features. However, related…

2021

Predict then Interpolate: A Simple Algorithm to Learn Stable Classifiers

ICML 2021spotlight

We propose Predict then Interpolate (PI), a simple algorithm for learning correlations that are stable across environments. The algorithm follows from the intuition that when using a classifier trained on one environment to make predictions on examples from another environment, its mistakes are info…

2020

Few-shot Text Classification with Distributional Signatures

ICLR 2020poster

In this paper, we explore meta-learning for few-shot text classification. Meta-learning has shown strong performance in computer vision, where low-level patterns are transferable across learning tasks. However, directly applying this approach to text is challenging--lexical features highly informati…

Cited by 233SourcecodeScholar