← Search

Yongchao Zhou

5 accepted papers

2024

DistillSpec: Improving Speculative Decoding via Knowledge Distillation

ICLR 2024poster

Speculative decoding~(SD) accelerates large language model inference by employing a faster {\em draft} model for generating multiple tokens, which are then verified in parallel by the larger {\em target} model, resulting in the text generated according to the target model distribution. However, iden…

Cited by 86SourcePDFScholar
2024

Identifying the Risks of LM Agents with an LM-Emulated Sandbox

ICLR 2024spotlight

Recent advances in Language Model (LM) agents and tool use, exemplified by applications like ChatGPT Plugins, enable a rich set of capabilities but also amplify potential risks—such as leaking private data or causing financial losses. Identifying these risks is labor-intensive, necessitating impleme…

2024

On-Policy Distillation of Language Models: Learning from Self-Generated Mistakes

ICLR 2024poster

Knowledge distillation (KD) is widely used for compressing a teacher model to reduce its inference cost and memory footprint, by training a smaller student model. However, current KD methods for auto-regressive sequence models suffer from distribution mismatch between output sequences seen during tr…

Cited by 109SourcePDFScholar
2023

Large Language Models are Human-Level Prompt Engineers

ICLR 2023poster

By conditioning on natural language instructions, large language models (LLMs) have displayed impressive capabilities as general-purpose computers. However, task performance depends significantly on the quality of the prompt used to steer the model, and most effective prompts have been handcrafted b…