← Search

Zhuohan Li

8 accepted papers

2024

LMSYS-Chat-1M: A Large-Scale Real-World LLM Conversation Dataset

ICLR 2024spotlight

Studying how people interact with large language models (LLMs) in real-world scenarios is increasingly important due to their widespread use in various applications. In this paper, we introduce LMSYS-Chat-1M, a large-scale dataset containing one million real-world conversations with 25 state-of-the-…

2023

FlexGen: High-Throughput Generative Inference of Large Language Models with a Single GPU

ICML 2023oral

The high computational and memory requirements of large language model (LLM) inference make it feasible only with multiple high-end accelerators. Motivated by the emerging demand for latency-insensitive tasks with batched processing, this paper initiates the study of high-throughput LLM inference us…

2023

Judging LLM-as-a-Judge with MT-Bench and Chatbot Arena

NeurIPS 2023poster

Evaluating large language model (LLM) based chat assistants is challenging due to their broad capabilities and the inadequacy of existing benchmarks in measuring human preferences. To address this, we explore using strong LLMs as judges to evaluate these models on more open-ended questions. We exami…

2021

TeraPipe: Token-Level Pipeline Parallelism for Training Large-Scale Language Models

ICML 2021spotlight

Model parallelism has become a necessity for training modern large-scale deep language models. In this work, we identify a new and orthogonal dimension from existing model parallel approaches: it is possible to perform pipeline parallelism within a single training sequence for Transformer-based lang…

2020

Train Big, Then Compress: Rethinking Model Size for Efficient Training and Inference of Transformers

ICML 2020poster

Since hardware resources are limited, the objective of training deep learning models is typically to maximize accuracy subject to the time and memory constraints of training and inference. We study the impact of model size in this setting, focusing on Transformer models for NLP tasks that are limite…

Cited by 360SourcePDFScholar
2019

Efficient Training of BERT by Progressively Stacking

ICML 2019oral

Unsupervised pre-training is popularly used in natural language processing. By designing proper unsupervised prediction tasks, a deep neural network can be trained and shown to be effective in many downstream tasks. As the data is usually adequate, the model for pre-training is generally huge and co…

2019

Fast Structured Decoding for Sequence Models

NeurIPS 2019poster

Autoregressive sequence models achieve state-of-the-art performance in domains like machine translation. However, due to the autoregressive factorization nature, these models suffer from heavy latency during inference. Recently, non-autoregressive sequence models were proposed to speed up the infere…

2018

Towards Binary-Valued Gates for Robust LSTM Training

ICML 2018oral

Long Short-Term Memory (LSTM) is one of the most widely used recurrent structures in sequence modeling. It aims to use gates to control information flow (e.g., whether to skip some information or not) in the recurrent computations, although its practical implementation based on soft gates only parti…