← Search

Ziniu Zhang

4 accepted papers

2025

Efficient Ensemble for Fine-tuning Language Models on Multiple Datasets

ACL 2025long

This paper develops an ensemble method for fine-tuning a language model to multiple datasets. Existing methods, such as quantized LoRA (QLoRA), are efficient when adapting to a single dataset. When training on multiple datasets of different tasks, a common setup in practice, it remains unclear how t…

2025

Linear-Time Demonstration Selection for In-Context Learning via Gradient Estimation

EMNLP 2025

This paper introduces an algorithm to select demonstration examples for in-context learning of a query set. Given a set of n examples, how can we quickly select k out of n to best serve as the conditioning for downstream inference? This problem has broad applications in prompt tuning and chain-of-th

2025

MMInA: Benchmarking Multihop Multimodal Internet Agents

ACL 2025finding

Autonomous embodied agents live on an Internet of multimedia websites. Can they hop around multimodal websites to complete complex user tasks? Existing benchmarks fail to assess them in a realistic, evolving environment for their embodiment across websites. To answer this question, we present MMInA,…

2024

Scalable Fine-tuning from Multiple Data Sources: A First-Order Approximation Approach

EMNLP 2024finding

We study the problem of fine-tuning a language model (LM) for a target task by optimally using the information from n auxiliary tasks. This problem has broad applications in NLP, such as targeted instruction tuning and data selection in chain-of-thought fine-tuning. The key challenge of this problem…