← Search

Zichun Yu

6 accepted papers

2025

Group-Level Data Selection for Efficient Pretraining

NeurIPS 2025poster

The efficiency and quality of language model pretraining are largely determined by the way pretraining data are selected. In this paper, we introduce *Group-MATES*, an efficient group-level data selection approach to optimize the speed-quality frontier of language model pretraining. Specifically, Gr…

Cited by 0SourcecodeScholar
2025

Montessori-Instruct: Generate Influential Training Data Tailored for Student Learning

ICLR 2025poster

Synthetic data has been widely used to train large language models, but their generative nature inevitably introduces noisy, non-informative, and misleading learning signals. In this paper, we propose Montessori-Instruct, a novel data synthesis framework that tailors the data synthesis ability of th…

2024

MATES: Model-Aware Data Selection for Efficient Pretraining with Data Influence Models

NeurIPS 2024poster

Pretraining data selection has the potential to improve language model pretraining efficiency by utilizing higher-quality data from massive web data corpora. Current data selection methods, which rely on either hand-crafted rules or larger reference models, are conducted statically and do not captur…

2023

Augmentation-Adapted Retriever Improves Generalization of Language Models as Generic Plug-In

ACL 2023long

Retrieval augmentation can aid language models (LMs) in knowledge-intensive tasks by supplying them with external information. Prior works on retrieval augmentation usually jointly fine-tune the retriever and the LM, making them closely coupled. In this paper, we explore the scheme of generic retrie…

2022

Automatic Label Sequence Generation for Prompting Sequence-to-sequence Models

COLING 2022main

Prompting, which casts downstream applications as language modeling tasks, has shown to be sample efficient compared to standard fine-tuning with pre-trained models. However, one pitfall of prompting is the need of manually-designed patterns, whose outcome can be unintuitive and requires large valid…