← Search

Yunfan Shao

10 accepted papers

2025

Case2Code: Scalable Synthetic Data for Code Generation

COLING 2025main

Large Language Models (LLMs) have shown outstanding breakthroughs in code generation. Recent work improves code LLMs by training on synthetic data generated by some powerful LLMs, which can be challenging to scale due to the dependence on a teacher model and high generation costs. In this paper, we…

2025

FastMCTS: A Simple Sampling Strategy for Data Synthesis

ACL 2025long

Synthetic high-quality multi-step reasoning data can significantly enhance the performance of large language models on various tasks. However, most existing methods rely on rejection sampling, which generates trajectories independently and suffers from inefficiency and imbalanced sampling across pro…

2025

UnitCoder: Scalable Code Synthesis from Pre-training Corpora

EMNLP 2025

Large Language Models (LLMs) have demonstrated remarkable capabilities in various tasks, yet code generation remains a major challenge. Despite the abundant sources of code data, constructing high-quality training datasets at scale poses a significant challenge. Pre-training code data typically suff

2024

Balanced Data Sampling for Language Model Training with Clustering

ACL 2024findings

Data plays a fundamental role in the training of Large Language Models (LLMs). While attention has been paid to the collection and composition of datasets, determining the data sampling strategy in training remains an open question. Most LLMs are trained with a simple strategy, random sampling. Howe…

2024

Unified Active Retrieval for Retrieval Augmented Generation

EMNLP 2024finding

In Retrieval-Augmented Generation (RAG), retrieval is not always helpful and applying it to every instruction is sub-optimal. Therefore, determining whether to retrieve is crucial for RAG, which is usually referred to as Active Retrieval. However, existing active retrieval methods face two challenge…

2023

PerturbScore: Connecting Discrete and Continuous Perturbations in NLP

EMNLP 2023long findings

With the rapid development of neural network applications in NLP, model robustness problem is gaining more attention. Different from computer vision, the discrete nature of texts makes it more challenging to explore robustness in NLP. Therefore, in this paper, we aim to connect discrete perturbation…

Cited by 0SourcecodeScholar
2022

Black-Box Tuning for Language-Model-as-a-Service

ICML 2022spotlight

Extremely large pre-trained language models (PTMs) such as GPT-3 are usually released as a service. It allows users to design task-specific prompts to query the PTMs through some black-box APIs. In such a scenario, which we call Language-Model-as-a-Service (LMaaS), the gradients of PTMs are usually…

2021

Accelerating BERT Inference for Sequence Labeling via Early-Exit

ACL 2021long

Both performance and efficiency are crucial factors for sequence labeling tasks in many real-world scenarios. Although the pre-trained models (PTMs) have significantly improved the performance of various sequence labeling tasks, their computational cost is expensive. To alleviate this problem, we ex…

2020

CoLAKE: Contextualized Language and Knowledge Embedding

COLING 2020main

With the emerging branch of incorporating factual knowledge into pre-trained language models such as BERT, most existing models consider shallow, static, and separately pre-trained entity embeddings, which limits the performance gains of these models. Few works explore the potential of deep contextu…